完全使用动态表生成刷新SQLAlchemy

Question

我需要在一个循环里创建很多相似的数据库，位置各不相同。在循环开始时，我会为新的 path_sql_db 在磁盘上创建一个引擎。

    engine = sa.create_engine("sqlite:///{}".format(path_sql_db), echo=0, listeners=[util_sa.ForeignKeysListener()])
    Session = sa.orm.sessionmaker(bind=engine)
    session = Session()

然后，我的表在几个模块中都继承自一个叫 DB_Base 的东西，这个东西是在外部模块里定义的；

from sqlalchemy.ext.declarative import declarative_base
DB_Base = declarative_base()

问题是，在下一个循环迭代时，我无法创建我的表，因为它们似乎还存在于某个地方？

InvalidRequestError: Table 'vector_var01' is already defined for this MetaData instance.  
Specify 'extend_existing=True' to redefine options and columns on an existing Table object.

我试过从引擎上调用 MetaData.drop_all()；

meta = sa.MetaData(bind = engine)
meta.reflect()
meta.drop_all()
session.close()

也试过从 Base 上调用；

DB_Base.metadata.bind = engine
DB_Base.metadata.reflect()
DB_Base.metadata.drop_all()

但都没有成功，我还是在黑暗中摸索。

这个错误提到的 MetaData 实例到底指的是哪个？我该如何完全重置我的数据库代码的状态？

编辑

好的，我找到了问题所在。我正在尝试动态生成 ORM 表。我在研究优化程序，并把设计空间变量存储在自己的表中，每个可能的变量值一行。

导致错误的最小示例；

from sqlalchemy import *
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
Base = declarative_base()

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

def generate_variable_table_class(name):
    """This is a helper function which dynamically creates a new ORM enabled class
    The table will hold the individual values of each variable
    Individual values are stored as a string
    """

    class NewTable( Base ):
        __tablename__ = "vector_{}".format(name)
        id = Column(Integer, primary_key=True)
        value = Column(String(16), nullable=False, unique=True)
        def __init__(self,value):
            self.value = str(value)

        def __str__(self):
            return self.value

        def __repr__(self):
            return self.value    


    NewTable.__name__ = "vector_ORM_{}".format(name)

    return NewTable


if __name__ == "__main__":

    for name in 'asfd', 'jkl', 'xyz':
        print("For loop: ",name)
        engine = create_engine(r'sqlite:///c:\testdelete\{}.sql'.format(name))
        Base.metadata.create_all(engine)
        Session = sessionmaker(bind=engine)
        session = Session()

        bunch_o_foos = [Foo(name = i) for i in range(10)]
        session.add_all(bunch_o_foos)
        session.commit()
        for foo in bunch_o_foos:
            print(foo.id)


        variables = [generate_variable_table_class(i) for i in range(10)]

这实际上和这个问题是一样的； SQLAlchemy 中的动态 Python 类定义。这真的不可能吗？

orm sqlalchemy 循环迭代 metadata 数据库重置数据库引擎动态表设计空间变量

完全使用动态表生成刷新SQLAlchemy

1 个回答

撰写回答