在Python中插入之前检查SQLAlchemy

时间:2015-09-08 15:33:25

标签: python exception sqlalchemy race-condition

我有一个关于如何检查条目是否已存在于数据库表中的更正确/有效方法的问题。

假设我有一个名为Foo的表,其中包含单个属性,Name也是 唯一

如果已经存在具有此类名称的条目,那么在插入之前检查更好的方法是什么?我可以想到两个选择:

from sqlalchemy import MetaData, Table, Column, String, \
create_engine
from sqlalchemy.orm import mapper, relationship, sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.exc import IntegrityError

Base = declarative_base()

class Foo(Base):
    __tablename__ = "foo"

    name = Column(String(), primary_key=True)

    @staticmethod
    def silent_insert1(name):
            if not session.query(Foo).filter(Foo.name == name).first():
                    foo = Foo()
                    foo.name = name
                    session.add(foo)
                    session.commit()
            else:
                    print("already exists 1")

    @staticmethod
    def silent_insert2(name):
            try:
                    foo = Foo()
                    foo.name = name
                    session.add(foo)
                    session.commit()
            except(IntegrityError):
                    print("already exists 2")


engine = create_engine('sqlite://', echo=True)
Base.metadata.create_all(engine)
session = sessionmaker(engine)()

Foo.silent_insert1(1)
Foo.silent_insert2(1)

在第一个 silent_insert1 中,当然存在竞争条件。第二个看起来更像是防弹。

哪一个更好?甚至是一种更好的方法或推荐的方法来检查这个?

谢谢

1 个答案:

答案 0 :(得分:4)

您应该将session.add放在try/except

中 其次,这是一个哲学问题。你不会得到明确的答案。

这两种风格被称为在你跳跃之前看(LBYL),这种风格提前使用if语句来看你是否应该这样做。

第二种风格被称为更容易请求宽恕,而不是权限(EAFP),这种风格假设它将起作用并捕获并处理规则发生时引发的异常破碎。作为一般规则,python程序员倾向于支持EAFP,但是像任何规则一样,有许多例外