update for sqlalchemy 0.6

This commit is contained in:
Damien Elmes 2010-04-21 13:21:16 +09:00
parent 1be5943903
commit a1d3f13f0a
2 changed files with 25 additions and 1 deletions

View file

@ -65,6 +65,27 @@ class SessionHelper(object):
self._lockDB()
self._seen = True
def save(self, obj):
# compat
if sqlalchemy.__version__.startswith("0.4."):
self._session.save(obj)
else:
self._session.add(obj)
def clear(self):
# compat
if sqlalchemy.__version__.startswith("0.4."):
self._session.clear()
else:
self._session.expunge_all()
def update(self, obj):
# compat
if sqlalchemy.__version__.startswith("0.4."):
self._session.update(obj)
else:
self._session.add(obj)
def execute(self, *a, **ka):
x = self._session.execute(*a, **ka)
runHook("dbFinished")

View file

@ -2841,7 +2841,10 @@ class DeckStorage(object):
def _init(s):
"Add a new deck to the database. Return saved deck."
deck = Deck()
if sqlalchemy.__version__.startswith("0.4."):
s.save(deck)
else:
s.add(deck)
s.flush()
return deck
_init = staticmethod(_init)