主页 / coding / 问题 / 79188446
Accepted
BaruchLi
Asked: 2024-11-14 18:57:31 +0800 CST

为什么当“DROP TABLE IF EXISTS”找不到表时 Flask-SQLAlchemy 不会引发异常？

  • 772

我正在使用 Flask-SQLAlchemy（版本 2.5.1）和 Python 3.11 连接到 PostgreSQL 15 数据库。

如果表存在，我想使用 DROP TABLE IF EXISTS ... CASCADE 删除该表。这是我的代码：

from sqlalchemy import text
from sqlalchemy.exc import SQLAlchemyError

try:
    sql = text("DROP TABLE IF EXISTS {table_name} CASCADE")
    db.session.execute(sql)
    db.session.commit()

except Exception as e:
    logger.error(f"Failed to drop {table_name} table: {str(e)}")

except SQLAlchemyError as e:
    db.session.rollback()
    logger.error(f"An error occurred: {str(e)}")

但是，当表不存在时，我没有看到 Python 中引发任何异常，即使 PostgreSQL 日志显示以下内容：

LOG:  statement: DROP TABLE IF EXISTS devices CASCADE
NOTICE:  table "devices" does not exist, skipping
LOG:  statement: COMMIT

由于该表不存在，PostgreSQL 会记录一个通知，但我的 Python 代码中不会传播任何异常。

我的问题是：

  1. 当表不存在时，传递“原始”SQL 语句是否会引发异常？
  2. 有没有办法将数据库的状态消息绑定到我的 Python 代码，以便我可以更明确地处理这种情况？
  3. execute()我是否应该在和commit()命令之后手动检查数据库的错误代码？
sqlalchemy

1 个回答

  1. Best Answer

    1. 当表不存在时，传递“原始”SQL 语句是否会引发异常？

      • 否：如果 PostgreSQL 出现异常，SQLAlchemy 也会引发异常，并且由于IF EXISTS已指定，如果表不存在，PostgreSQL 不会出错。

    2. 有没有办法将数据库的状态消息绑定到我的 Python 代码，以便我可以更明确地处理这种情况？

      • 您可以根据需要记录NOTICE消息，但如果不检查日志，则无法直接访问它们（或者至少没有记录这样做的方法）。请参阅NOTICE 日志记录的文档。

    3. 我是否应该在 execute() 和 commit() 命令之后手动检查数据库中的错误代码？

      • 没有理由这样做，除非由于某种原因您需要知道数据库已跳过删除不存在的表。

    这是一个简单的例子：

    import logging

import sqlalchemy as sa

logging.basicConfig()
logging.getLogger('sqlalchemy')
logging.getLogger('sqlalchemy.dialects.postgresql').setLevel(logging.INFO)

engine = sa.create_engine('postgresql+psycopg2:///so')

with engine.begin() as conn:
    conn.execute(sa.text("""DROP TABLE IF EXISTS t"""))

    这将产生以下输出：

    INFO:sqlalchemy.engine.Engine:select pg_catalog.version()
INFO:sqlalchemy.engine.Engine:[raw sql] {}
INFO:sqlalchemy.engine.Engine:select current_schema()
INFO:sqlalchemy.engine.Engine:[raw sql] {}
INFO:sqlalchemy.engine.Engine:show standard_conforming_strings
INFO:sqlalchemy.engine.Engine:[raw sql] {}
INFO:sqlalchemy.engine.Engine:BEGIN (implicit)
INFO:sqlalchemy.engine.Engine:DROP TABLE IF EXISTS t
INFO:sqlalchemy.engine.Engine:[generated in 0.00009s] {}
INFO:sqlalchemy.dialects.postgresql:NOTICE:  table "t" does not exist, skipping
INFO:sqlalchemy.engine.Engine:COMMIT
    • 0

相关问题