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

时间:2014-08-04 12:46:27

标签: python sqlalchemy

我需要在for循环中的不同位置创建许多类似的数据库。在循环开始时,我将引擎创建为磁盘上的新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()

问题是,在for循环的下一次迭代中,我无法创建我的表,因为它们仍然存在于某个地方?

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

还来自基地;

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

这与这个问题实际上是一样的; Dynamic Python Class Definition in SQLAlchemy。这不可能吗?

1 个答案:

答案 0 :(得分:3)

作为一个最小的例子:

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

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

for name in 'asfd', 'jkl', 'xyz':
    engine = create_engine('sqlite:///{}'.format(name))
    Base.metadata.create_all(engine)

运行良好。 drop_all表示丢弃表,可能不是您想要的。一旦您使用bind,您就可以将元数据对象绑定到该特定引擎。虽然:

Base.metadata.bind = engine
Base.metadata.create_all()

也适用于最小的例子。

修改

根据示例案例,您将收到错误,因为您尝试使用相同的Base子类来定义具有相同表名的类(例如vector_0),该子类具有附加到它的MetaData对象,每个表名只能有一个。

  • 在新的简单情况下,每个数据库的表之间没有区别,因此您应该将调用从{@ 1}}移出主循环,只进行一次。

    < / LI>
  • 如果您有每个数据库的行为,则每次都可以使用新的generate_variable_table_class(即将Base移动到函数中!)(另外,调用Foo }应该高于generate_variable_table_class而不是在最后)

  • 即便如此,sqlalchemy并不喜欢他们都被命名为create_all。声明性ORM评估您的类定义,因此在您之后设置NewTable之前它会看到NewTable。解决方案是不使用声明性系统(参见&#34;经典映射&#34;在文档中)。

  • 但另一种方法是扩展声明性元类以处理名称更改。 __name__函数采用显式declarative_base参数,因此这似乎符合框架规范。要使用下面的那个,您可以在metaclass定义中设置__name__ = "vector_ORM_{}".format(name)。如果你想要真正干净更新NewTable,虽然sqlalchemy不会在任何地方使用它。

__qualname__