tablename作为变量

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))

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

提前感谢！

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 ... 

另一个要考虑的选择是内置的重载方法。(请看:)