sqlalchemy从现有的构造新的声明性类

时间:2014-07-22 16:47:51

标签: python python-3.x sqlalchemy

我有一个用例,已经使用declarative_base映射了190多个表。

偶尔我需要为其中一些表创建'staging'表。登台表的结构与源表的不同之处在于它们在源表名前加上“stage_”,并且它们没有某些字段。

这是我在经过一些试验和错误后想出来的。

from sqlalchemy import Table, Column

def MutateDeclarative(source):
    columns = []
    omit_columns = ['created_at', 'updated_at']
    for c in source.__table__.c:
        if c.name not in omit_columns:
            columns.append(((c.name, 
            c.type), 
            {'primary_key': c.primary_key,
            'nullable': c.nullable,
            'doc': c.doc,
            'default': c.default,
            'unique': c.unique,
            'autoincrement': c.autoincrement}))
    class Stage(Base):
        __tablename__ = 'stage_' + source.__tablename__    
        __table__ = Table('stage_' + source.__tablename__, 
            Base.metadata, *[Column(*c[0], **c[1]) for c in columns])
    return Stage

此代码有效。我可以称之为

from models import Customer
MutatedCustomer = MutateDeclarative(Customer)

但我真的不知道我在那里做了什么,只是它有效。这种用法是否会产生任何意想不到的后果,或者是否存在从另一个声明性类生成类的标准方法,唯一的变化是表名的更改和某些列的省略。

非常感谢社区解释这一点。

1 个答案:

答案 0 :(得分:0)

我不确切地知道你的意图是什么,所以我无法判断是否有意想不到的后果。然而,这对我来说完全没问题。您只需为MutateDeclarative()的每次调用创建一个新类。

如果您遇到任何问题("意外后果"),您也必须延长MutateDeclarative()来处理它。

感觉有点像python中基于原型的继承。