Question

我在Python中使用SQLAlchemy，并声明我的类继承自声明基类，如下所示：

from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class SomeClass(Base):
    __tablename__ = 'some_table'
    id = Column(Integer, primary_key=True)
    name =  Column(String(50))

作为用户，我想将__tablename__定义为参数，而不是硬编码值，如下所示：

class SomeClass(Base):
    __tablename__ = f'{environment}_some_table'
    id = Column(Integer, primary_key=True)
    name =  Column(String(50))

我的理解是，在导入此软件包时将评估f'{environment}_some_table'，因此我无法在稍后阶段（即在交互式笔记本中）修改它。我有一段破碎的代码试图通过嵌套类和封装来解决这个问题，但是我没有设法引用外部类的实例变量。

class Outer:
    def __init__(self, env):
        self.environment = env

    class SomeClass(Base):
        __tablename__ = f'{self.environment}_some_table'
        id = Column(Integer, primary_key=True)
        name =  Column(String(50))

我已经阅读了几个SO问题，最好不要使用嵌套类，因为这些类之间没有建立特殊关系。那么我应该用什么样的设计来解决这个问题？

提前致谢！

Answer 1

您可以在函数范围内创建所有模型定义，以便依赖于外部参数：

def create_models(environment):
    class SomeClass(Base):
        __tablename__ = f'{environment}_some_table'
        id = Column(Integer, primary_key=True)
        name =  Column(String(50))
    ...
    globals().update(locals()) # update outer scope if needed

... # some time later
create_models('cheese')
... # now create the db session/engine/etc ...

另一个选择是内置重新加载方法。检查出来:)）

Python Sqlalchemy - tablename作为变量

1 个答案: