如何在SQLAlchemy中删除多个表

Question

受到这个问题的启发：如何在 SQLAlchemy 中删除表？，我最终提出了一个问题：如何删除多个表。

假设我有 3 个表，如下所示，我想删除 2 个表（想象一下还有更多表，所以不需要手动删除表）。

表格

import sqlalchemy as sqla
import sqlalchemy.ext.declarative as sqld
import sqlalchemy.orm as sqlo

sqla_base = sqld.declarative_base()

class name(sqla_base):
    __tablename__ = 'name'
    id = sqla.Column(sqla.Integer, primary_key=True)
    name = sqla.Column(sqla.String)

class job(sqla_base):
    __tablename__ = 'job'
    id = sqla.Column(sqla.Integer, primary_key=True)
    group = sqla.Column(sqla.String)

class company(sqla_base):
    __tablename__ = 'company'
    id = sqla.Column(sqla.Integer, primary_key=True)
    company = sqla.Column(sqla.String)

engine = sqla.create_engine("sqlite:///test.db", echo=True)
sqla_base.metadata.bind = engine

# Tables I want to delete
to_delete = ['job', 'function']

# Get all tables in the database
for table in engine.table_names():
    # Delete only the tables in the delete list
    if table in to_delete:
        sql = sqla.text("DROP TABLE IF EXISTS {}".format(table))
        engine.execute(sql)

# Making new tables now the old ones are deleted
sqla_base.metadata.create_all(engine)

SQLAlchemy 中如何？编辑

这是可行的，但是我想知道我是否可以在 SQLAlchemy 风格中执行相同的操作，而不是sqla.text("DROP TABLE IF EXISTS {}".format(table))使用（不使用sqla_base.metadata.drop_all()，因为这会删除所有表）执行原始 SQL 代码。

我知道该函数tablename.__table__.drop()或tablename.__table__.drop(engine)存在，但我不想为每个表手动键入它。

根据 @daveoncode 给出的答案，以下代码执行我想要的操作（编辑 2：添加checkfirst=True，以防它在 db 和 str() 中不存在）：

for table in sqla_base.metadata.sorted_tables:
    if str(table) in self.to_delete:
        table.drop(checkfirst=True)

问题

如何以 SQLAlchemy 风格删除多个表，实现与上面的原始 SQL 代码相同的效果？

Answer 1

您得到的错误非常清楚：

AttributeError: 'str' object has no attribute '__table__'

您不是在 Table 对象上迭代，而是在表名称（又名字符串！）上迭代，所以字符串当然没有 attribute __table__，所以您的语句：

tablename.__table__.drop() or tablename.__table__.drop(engine)

是错的！它应该是：

table_instance.__table__.drop() or table_instance.__table__.drop(engine)

您可以从元数据访问表实例。看看这里：

http://docs.sqlalchemy.org/en/latest/core/metadata.html#sqlalchemy.schema.MetaData.sorted_tables

更新：

无论如何，drop_all()这是用一个简单命令删除所有表的方法：http://docs.sqlalchemy.org/en/latest/core/metadata.html#sqlalchemy.schema.MetaData.drop_all