有没有办法在SQLAlchemy中访问PostgreSQL SQLSTATE代码？

Question

我正在寻找一种从SQLAlchemy获取PostgreSQL异常中的错误代码的方法。

有没有办法做到这一点。感谢

Answer 1

是的，可以。它在orig.pgcode成员上可用。

例如

import sqlalchemy as sa

try:
   exec_database_script(...)
except sa.exc.IntegrityError as e:
   pgcode = int(e.orig.pgcode)

   if pgcode == 23514:
      print("Check constraint violation!")