sqlalchemy - 反映带空格的表格和列

时间:2013-10-07 01:09:05

标签: python sqlalchemy override flask-sqlalchemy

如何在列名(和表名)中包含空格的数据库中使用sqlalchemy?

db.auth_stuff.filter("db.auth_stuff.first name"=='Joe')显然无法奏效。我不想在进行反射时手动定义所有内容,而是希望在从数据库读取的现有表名之间添加类似lambda x: x.replace(' ','_')的内容,并在我的模型中使用。 (创建一个通用函数来重命名所有不能与python一起使用的表名也可能很有用 - 保留字等。)

这样做有简单/干净的方法吗?

我想我需要定义自己的mapper类?

https://groups.google.com/forum/#!msg/sqlalchemy/pE1ZfBlq56w/ErPcn1YYSJgJ

或者使用某种__mapper_args__参数 - http://docs.sqlalchemy.org/en/rel_0_8/orm/mapper_config.html#naming-all-columns-with-a-prefix

理想地:

class NewBase(Base):
    __mapper_args__ = {
        'column_rename_function' : lambda x: x.replace(' ','_')
    }

class User(NewBase):
    __table__ = "user table"
    }

1 个答案:

答案 0 :(得分:5)

你可以使用reflection event来为列提供.key,但是当涉及主键列时,完整的配方有一个错误,这个错误在尚未发布的0.8.3版本中得到修复(如以及大师)。如果您在https://bitbucket.org/zzzeek/sqlalchemy/get/rel_0_8.zip处查看0.8.3,则此配方即使使用主键列也可以使用:

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

Base = declarative_base(cls=DeferredReflection)


e = create_engine("sqlite://", echo=True)
e.execute("""
    create table "user table" (
            "id col" integer primary key,
            "data col" varchar(30)
    )
""")

from sqlalchemy import event

@event.listens_for(Table, "column_reflect")
def reflect_col(inspector, table, column_info):
    column_info['key'] = column_info['name'].replace(' ', '_')

class User(Base):
    __tablename__ = "user table"

Base.prepare(e)

s = Session(e)
print s.query(User).filter(User.data_col == "some data")

DeferredReflection是一个可选的助手,可用于声明+反射。