Python SQLalchemy - 表名作为变量

Question

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

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' 将在我导入此包时进行评估，因此我将无法在稍后阶段（即在交互式笔记本中）对其进行修改。我有一段代码试图通过嵌套 classes 和封装来解决这个问题，但我没有设法引用外部 class.

的实例变量

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 问题，最好不要使用嵌套的 classes，因为这些 classes 之间没有建立特殊关系。那么我应该用什么样的设计来解决这个问题呢？

提前致谢！

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 方法。检查一下:)

Python SQLalchemy - 表名作为变量

Python Sqlalchemy - tablename as a variable

python

nested

sqlalchemy

runtime

class-variables