web-dev-qa-db-ja.com

SQLAlchemy IntegrityErrorをキャッチできません

試してみてください、sqlalchemy IntegrityErrorを正しくキャッチできないようです。

_from sqlalchemy import exc

try:
    insert_record()
except exc.IntegrityError, exc:
    print exc # this is never called
    handle_elegantly() # this is never called
_

予想されるとおり：

_IntegrityError: (IntegrityError) insert or update on table "my_table" 
                violates foreign key constraint "my_table_some_column_fkey"
_

私は明示的にしようとしました：

_from sqlalchemy.exc import IntegrityError
_

UPDATE：

セッションがデータベースにフラッシュされるまで、およびtry/exceptblocksが実行された後、Integrity Errorがスローされない、ここで何が起こっているかに合うと思われる何かを見つけました： SQLAlchemyで整合性エラーをキャッチしよう

ただし、tryブロックにsession.flush()を追加すると、InvalidRequestErrorが生成されます。

_ERROR:root:This Session's transaction has been rolled back due to a previous 
           exception during flush. To begin a new transaction with this Session, 
           first issue Session.rollback(). 
           Original exception was: (IntegrityError)
_
pythonexceptionexception-handlingerror-handlingsqlalchemy
27
Chrispy

Flaskアプリケーションで同じニーズがあります。以下のように処理し、動作します：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import exc

db = SQLAlchemy(Flask(__name__))

try:
     db.session.add(resource)
     return db.session.commit()
except exc.IntegrityError as e:
     db.session().rollback()
23
JsonBruce

エラーをキャッチしたかどうかに関係なく、IntegrityErrorが発生するとすぐに、作業していたセッションは無効になります。 2番目のエラーメッセージはTo begin a new transaction with this Session, first issue Session.rollback().を指示しているため、セッションを引き続き使用するにはsession.rollback()を発行する必要があります。

確かに言うことはできませんが、あなたまたはあなたのWebフレームワークが、何らかの方法でIntegrityErrorを発生させたセッションを使い続けようとしていると推測しています。例外をキャッチした後、または_handle_elegantly_関数でsession.rollback()を発行することをお勧めします。

以下を実行すると、私の意味がわかります：

_from sqlalchemy import types
from sqlalchemy import exc
from sqlalchemy import create_engine
from sqlalchemy.schema import Column
from zope.sqlalchemy import ZopeTransactionExtension
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker, scoped_session

Base = declarative_base()


class User(Base):
    __tablename__ = 'user'
    name = Column(types.String, primary_key=True)


def handle_elegantly(name):
    session = DBSession()
    session.add(User(name=name))
    session.flush()
    print 'Exception elegantly handled!!\n'


def pretend_view(request):
    """Pretend view in a Pyramid application using pyramid_tm"""
    session = DBSession()
    user = User()
    print '\n-------Here we rollback before continuing -------'
    try:
        session.add(user)
        session.flush()
    except exc.IntegrityError:
        session.rollback()
        handle_elegantly('This will run fine')

    print '\n------- Here we do not, and this will error -------'
    try:
        session.add(user)
        session.flush()
    except exc.IntegrityError:
        handle_elegantly('Exception will be raised')


if __name__ == '__main__':
    engine = create_engine('sqlite://')
    global DBSession
    DBSession = scoped_session(
        sessionmaker(extension=ZopeTransactionExtension()))
    DBSession.configure(bind=engine)
    Base.metadata.bind = engine
    Base.metadata.create_all()
    pretend_view("dummy request")
_
22
Jason

SQLALCHEMY_COMMIT_ON_TEARDOWN = False

0
ivan