如何使用sqlalchemy IntegrityError查找违规属性

时间:2012-07-03 14:49:30

标签: python sqlalchemy integrity

我有一个非常简单的SqlAlchemy模型

class User(Base):
    """ The SQLAlchemy declarative model class for a User object. """
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    phone = Column(String, unique=True)
    email = Column(String, unique=True)

插入新用户时,如果电子邮件或电话是重复的,则可能会出现IntegrityError

有没有办法检测哪些列违反了完整性错误?或者是进行单独查询以查看或存在值的唯一方法吗?

4 个答案:

答案 0 :(得分:3)

不幸的是,没有干净的方法可以做到这一点,但我在IntegrityError和parse模块上使用了orig属性:

try:
    db.session.add(user)
    db.session.commit()
except IntegrityError, e:
    dupe_field = parse('duplicate key value violates unique constraint "{constraint}"\nDETAIL:  Key ({field})=({input}) already exists.\n', str(e.orig))["field"]

这可能不是IntegrityError抛出的唯一错误字符串,它可能会在SQLAlchemy的未来更新中发生变化,因此它不理想

答案 1 :(得分:0)

您可以使用以下方法获取基础代码,消息并相应地格式化消息。

except exc.IntegrityError as e:
       errorInfo = e.orig.args
       print(errorInfo[0])  #This will give you error code
       print(errorInfo[1])  #This will give you error message

顺便说一句,您必须从sqlalchemy导入exc:from sqlalchemy import exc 让我知道您是否需要其他信息。我可以尝试一下。

有关sqlalchemy exc的更多信息,请找到代码:https://github.com/zzzeek/sqlalchemy/blob/master/lib/sqlalchemy/exc.py

答案 2 :(得分:0)

try:
    ....
except IntegrityError as e:
    print(e.orig.diag.message_detail)

这对我有用。

答案 3 :(得分:-3)

我通常会使用try catch。

try:
    session.commit()
catch:   
str(sys.exc_info()[0]) + " \nDESCRIPTION:  "+ str(sys.exc_info()[1]) + "\n" + str(sys.exc_info()[2])

当我遇到完整性错误时,我收到以下消息,我跳过了特定的传输并继续其余的传输

DESCRIPTION:  (IntegrityError) duplicate key value violates unique constraint "test_code"
DETAIL:  Key (test_code)=(5342) already exists.
'INSERT INTO test_table (pk, test_code, test_name) VALUES (%(pk)s, %(test_code)s, %(test_name)s)' { 'pk': '1', 'test_code': '5342', 'test_name': 'test' }