Python Sqlalchemy - 表名作为变量

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__为参数，而不是硬编码的 value ，如下所示：

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

另一个可以查看的选择是内置的reload方法。一探究竟 ：）