python - SQLAlchemy 中的动态 Python 类定义

Question

我正在使用声明性基础通过 SQLAlchemy 创建后端应用程序。 ORM 需要大约 15 个表，每个表都映射到 SQLAlchemy 中的一个类对象。因为这些类对象的定义都是相同的，所以我认为工厂模式可以更简洁地生成类。但是，不仅必须定义这些类，还必须将它们分配给唯一的变量名，以便在整个项目中导入和使用它们。

(对不起，如果这个问题有点长，我更新了它，因为我更好地理解了这个问题。)

因为我们有很多列(~1000)，所以我们在外部文本文件中定义它们的名称和类型以保持可读性。完成后，声明我们的模型的一种方式是这样的:

class Foo1(Base):
    __tablename___ = 'foo1'

class Foo2(Base):
    __tablename___ = 'foo2'

... etc

然后我可以通过遍历外部文本文件的内容并在每个类定义上使用 setattr() 来添加列。

这没问题，但感觉太重复了，因为我们有大约 15 张 table 。因此，我尝试编写一个可以动态定义类的工厂函数。

def orm_factory(class_name):
    class NewClass(Base):
        __tablename__ = class_name.lower()
    NewClass.__name__ = class_name.upper()
    return NewClass

同样，我可以遍历列并使用 setattr()。当我把它放在一起时，它看起来像这样:

for class_name in class_name_list:
    ORMClass = orm_factory(class_name)
    header_keyword_list = get_header_keyword_list(class_name)
    define_columns(ORMClass, header_keyword_list)

其中get_header_keyword_list获取列信息，define_columns执行setattr()赋值。当我使用它并运行 Base.metadata.create_all() 时，SQL 模式生成得很好。

但是，当我尝试将这些类定义导入另一个模型时，我得到了这样的错误:

SAWarning: The classname 'NewClass' is already in the registry of this declarative base, mapped to <class 'ql_database_interface.IR_FLT_0'>

根据我昨天学到的知识，我现在意识到这是完全有道理的:Python class variable name vs __name__ .

您可以通过在工厂函数中使用 type 作为类生成器来解决这个问题(如下面的两个答案所做的那样)。但是，这并不能解决能够导入类的问题，因为虽然类是在工厂函数中动态构造的，但该函数的输出分配给的变量是静态的。即使它是动态的，例如字典键，它也必须位于模块 namespace 中才能从另一个模块导入。有关更多详细信息，请参阅我的答案。

Answer 1

这听起来像是一个粗略的想法。但解决这个问题很有趣，所以这里是你如何让它发挥作用的。

据我了解，您的问题是您想将动态创建的类添加到模块中。我使用模块和 init.py 文件创建了一个 hack。

动态模块/__init__.py:

import dynamic

class_names = ["One", "Two", "Three"]

for new_name in class_names:
     dynamic.__dict__['Class%s' % new_name] = type("Class%s" % (new_name), (object,), {'attribute_one': 'blah'})

dynamicModule/dynamic.py:

"""Empty file"""

测试.py:

import dynamicModule
from dynamicModule import dynamic
from dynamicModule.dynamic import ClassOne

dynamic.ClassOne
"""This all seems evil but it works for me on python 2.6.5"""

__init__.py:

"""Empty file"""