在SQLAlchemy中禁用提交对象更改

时间:2012-07-25 10:33:09

标签: python mysql sqlalchemy

我在不是Web应用程序的项目中使用SQLAlchemy。它是一个服务器应用程序,它从数据库加载许多不同的对象并在本地修改它们,但是每次发出提交时都不希望将这些更新保存到数据库中。我以前和Django ORM合作过一些网络项目,发现它更适合我想要实现的目标。在Django ORM中我可以随时随地使用.save()每个对象而不保存其他我可能不想保存的东西。我理解为什么它在SQLAlchemy中的工作方式如此,但我想知道如何以Django的方式做到这一点?


更新 为了更容易理解我想要实现的目标,我将为您提供一个示例。

这实际上是如何运作的:

a = MyModel.query.get(1)
b = MyModel.query.get(1)

a.somefield = 1
b.somefield = 2

# this will save both of changed models
session.commit()

这就是我希望它发挥作用的方式:

a = MyModel.query.get(1)
b = MyModel.query.get(1)

a.somefield = 1
b.somefield = 2

a.save()
# I didn't want to save b, changes of b weren't committed

我想更好地控制实际保存的内容。我想每隔5分钟左右保存每个对象的更改。

3 个答案:

答案 0 :(得分:3)

我用的是:

class BaseModel(object):
    def save(self, commit=True):
        # this part can be optimized.
        try:
            db.session.add(self)
        except FlushError:
            # In case of an update operation.
            pass

        if commit:
            db.session.commit()

    def delete(self, commit=True):
        db.session.delete(self)

        if commit:
            db.session.commit()

然后我将我的模型定义为:

class User(db.Model, BaseModel)

所以,现在我可以做到:

u = User(username='foo', password='bar')
u.save()

这是你打算实现的目标吗?

答案 1 :(得分:2)

我不确定我是否理解你的困境。

在Django,

foo = MyModel(field1='value1', field2='value2')
foo.save()

或者

foo = MyModel.objects.create(field1='value1', field2='value2')

在SQLAlchemy中,

foo = MyModel(field1='value1', field2='value2')
session.add(foo)

此时您只将对象添加到会话中,并且尚未提交事务。只有在完成所需的任何更改后才需要提交

session.commit()

看看这个link。我认为这将使从Django ORM到SqlAlchemy的过渡更容易。

<强>更新

对于这种情况,您可以使用多个会话。

engine = create_engine("postgresql+psycopg2://user:password@localhost/test")
metadata = MetaData(bind=engine)
Session = sessionmaker(bind=engine)
session1 = Session()
session2 = Session()
Base = declarative_base()
class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    age = Column(Integer)
    def __init__(self, name, age):
        self.name = name
        self.age = age
    def __repr__(self):
        return "<User('%s','%s')>" % (self.name, self.age)
Base.metadata.create_all(engine)

在'test'db中创建了一个表'users'。此外,还初始化了2个会话对象session1和session2。

a = User('foo','10')
b = User('bar', '20')
session1.add(a)
session1.add(b)
session1.commit()

表用户现在将有2条记录

1: foo, 10
2: bar, 20

使用session2获取'foo'记录会话session1和'bar'。

foo = session1.query(User).filter(User.name == "foo").first()
bar = session2.query(User).filter(User.name == "bar").first()

更改2条记录

foo.age = 11
bar.age = 21

现在,如果你想让foo的更改继续进行,

session1.commit()

和吧,

session2.commit() 

答案 2 :(得分:0)

不要挑起旧职位,但是

你说:

  

我想每隔5分钟左右保存每个对象的更改。

那么为什么不使用像Celery这样的调度程序。(我使用pyramid_celery

使用此功能,您可以每5分钟保存一个对象,即您可以添加装饰器:

@periodic_task(run_every=crontab(minute="*/5")
def somefunction():
    #your code here

这非常有用,尤其是当您需要更新数据库以确保它是最新的时(如果有许多用户使用您的系统)

希望这可以帮助某人每隔5分钟保存一次部分。