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._lockDB()
self._seen = True 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): def execute(self, *a, **ka):
x = self._session.execute(*a, **ka) x = self._session.execute(*a, **ka)
runHook("dbFinished") runHook("dbFinished")

View file

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