在SQLAlchemy中映射非ascii,长列名称失败,NoSuchColumnError

时间:2014-07-24 22:09:23

标签: python sql-server sqlalchemy pyodbc

可能是RowProxy对象'键被截断到一定长度?

我的旧版MSSQL数据库中有一些带有重音字母的列名。

我使用以下代码将其列映射到python属性:

@event.listens_for(Table, "column_reflect")
def column_reflect(inspector, table, column_info):
    if table.name == 'D_Allomanylista_Komplex_V':
        # set column.key = "attr_<lower_case_name>"
        columns_to_change = {
            'Allapot': 'allapot',
            'BIZTNEVE': 'biztosito_neve',
            u'Levelez\xe9siC\xedmUtca': 'LevelezesiCimUtca',
            u'Szerz\u0151d\xe9sSorsz\xe1m': 'SzerzodesSzam',
        }
        column_info['key'] = columns_to_change.get(column_info['name'], column_info['name'])

komplex_table = inspect_komplex_table()

class BiztositasokModel(Base):
    __table__ = komplex_table
    __mapper_args__ = {
        'primary_key': [
            komplex_table.c[u'KTVSZAM'],
            komplex_table.c[u'NyugtaSzam'],
            komplex_table.c[u'ajanlatszam'],
            komplex_table.c['SzerzodesSzam'],
            ],
        'include_properties': [
            'biztosito_neve',
            komplex_table.c['allapot'],
            komplex_table.c['LevelezesiCimUtca'],
            komplex_table.c['SzerzodesSzam'],
            'KTVSZAM', 'NyugtaSzam', 'ajanlatszam'],
    }

不幸的是,当我尝试查询时,此代码会出错,说明SzerzodesSzam列不存在。

/Users/viktornagy/.virtualenvs/dosszie/lib/python2.7/site-packages/sqlalchemy/orm/loading.pyc in _instance(row, result)
    361             identitykey = (
    362                 identity_class,
--> 363                 tuple([row[column] for column in pk_cols])
    364             )
    365

/Users/viktornagy/.virtualenvs/dosszie/lib/python2.7/site-packages/sqlalchemy/engine/result.pyc in _key_fallback(self, key, raiseerr)
    329                 raise exc.NoSuchColumnError(
    330                     "Could not locate column in row for column '%s'" %
--> 331                     expression._string_or_unprintable(key))
    332             else:
    333                 return None

NoSuchColumnError: "Could not locate column in row for column 'D_Allomanylista_Komplex_V.Szerz\\u0151d\\xe9sSorsz\\xe1m'"

我进入调试模式以找出问题的根源,并发现列上方的第363行为Column('Szerz\u0151d\xe9sSorsz\xe1m', INTEGER(), table=<D_Allomanylista_Komplex_V>, key='SzerzodesSzam', nullable=False),而相应的row.key()u'D_Allomanylista_Komplex_V_Szerz\u0151d\xe9sSorsz'。好像密钥会被截断(u'Szerz\u0151d\xe9sSorsz'一次,u'Szerz\u0151d\xe9sSorsz\xe1m'在另一点。)

查询row[u'D_Allomanylista_Komplex_V_Szerz\u0151d\xe9sSorsz']确实有效,并给出了预期的结果。

有没有办法避免这种截断?

2 个答案:

答案 0 :(得分:0)

以上似乎是pyodbc驱动程序的限制。切换到pymssql解决了这个问题。

答案 1 :(得分:0)

适用于我(即使在OSX上使用freetds 0.86,通常这是unicode DDL严重崩溃的地方):

from sqlalchemy import *
from sqlalchemy.orm import *
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class Foo(Base):
    __tablename__ = 'D_Allomanylista_Komplex_V'

    bar = Column(u'Szerz\u0151d\xe9sSorsz\xe1m', INTEGER(), autoincrement=False, primary_key=True)

e = create_engine("mssql+pyodbc://scott:tiger@ms_2008", echo=True)

c = e.connect()
t = c.begin()

Base.metadata.create_all(c)

s = Session(c)
s.add(Foo(bar=5))
s.commit()

row = s.query(Foo).first()

assert row.bar == 5

输出:

CREATE TABLE [D_Allomanylista_Komplex_V] (
    [SzerződésSorszám] INTEGER NOT NULL, 
    PRIMARY KEY ([SzerződésSorszám])
)

2014-07-24 19:15:22,504 INFO sqlalchemy.engine.base.Engine ()
2014-07-24 19:15:22,507 INFO sqlalchemy.engine.base.Engine INSERT INTO [D_Allomanylista_Komplex_V] ([SzerződésSorszám]) VALUES (?)
2014-07-24 19:15:22,507 INFO sqlalchemy.engine.base.Engine (5,)
2014-07-24 19:15:22,510 INFO sqlalchemy.engine.base.Engine SELECT TOP 1 [D_Allomanylista_Komplex_V].[SzerződésSorszám] AS [D_Allomanylista_Komplex_V_SzerződésSorszám] 
FROM [D_Allomanylista_Komplex_V]