From: Scott Torborg Date: Mon, 2 Nov 2009 20:51:22 +0000 (-1000) Subject: brought up to date with latest sqlalchemy conventions X-Git-Url: https://asedeno.scripts.mit.edu/gitweb/?a=commitdiff_plain;h=d24e23a6a01ed245e204c8c29fc10a5fbc6f8bc4;p=bluechips.git brought up to date with latest sqlalchemy conventions --- diff --git a/bluechips/controllers/spend.py b/bluechips/controllers/spend.py index fd88453..1a1c346 100644 --- a/bluechips/controllers/spend.py +++ b/bluechips/controllers/spend.py @@ -42,12 +42,12 @@ class SpendController(BaseController): # old one if id is None: e = model.Expenditure() + meta.Session.add(e) else: e = meta.Session.query(model.Expenditure).get(id) # Set the fields that were submitted update_sar(e, self.form_result) - meta.Session.save_or_update(e) if id is None: e.even_split() diff --git a/bluechips/controllers/transfer.py b/bluechips/controllers/transfer.py index 68d8b44..b4b375a 100644 --- a/bluechips/controllers/transfer.py +++ b/bluechips/controllers/transfer.py @@ -38,11 +38,11 @@ class TransferController(BaseController): if id is None: t = model.Transfer() + meta.Session.add(t) else: t = meta.Session.query(model.Transfer).get(id) update_sar(t, self.form_result) - meta.Session.save_or_update(t) meta.Session.commit() h.flash('Transfer recorded.') diff --git a/bluechips/model/__init__.py b/bluechips/model/__init__.py index 4b016ea..ba5404f 100644 --- a/bluechips/model/__init__.py +++ b/bluechips/model/__init__.py @@ -16,7 +16,7 @@ from datetime import datetime def init_model(engine): """Call me before using any of the tables or classes in the model""" - sm = orm.sessionmaker(autoflush=True, transactional=True, bind=engine) + sm = orm.sessionmaker(autoflush=True, bind=engine) meta.engine = engine meta.Session = orm.scoped_session(sm) diff --git a/bluechips/model/expenditure.py b/bluechips/model/expenditure.py index 8d3bbbc..f7e6934 100644 --- a/bluechips/model/expenditure.py +++ b/bluechips/model/expenditure.py @@ -84,6 +84,6 @@ class Expenditure(object): for user, share in amounts_dict.iteritems(): s = Split(self, user, share) - meta.Session.save(s) + meta.Session.add(s) __all__ = ['Expenditure'] diff --git a/bluechips/tests/__init__.py b/bluechips/tests/__init__.py index cbc0ad4..79de019 100644 --- a/bluechips/tests/__init__.py +++ b/bluechips/tests/__init__.py @@ -32,7 +32,7 @@ def setUpPackage(): SetupCommand('setup-app').run([config['__file__']]) test_user = bluechips.model.User(u'root', u'Charlie Root', False) - meta.Session.save(test_user) + meta.Session.add(test_user) meta.Session.commit() def tearDownPackage(): @@ -50,7 +50,7 @@ def createUsers(n=None): n = random.randint(2, 5) for i in xrange(n): u = bluechips.model.User(sample_users[i].lower(), resident=True) - meta.Session.save(u) + meta.Session.add(u) meta.Session.commit() def createExpenditures(n=None): @@ -60,7 +60,7 @@ def createExpenditures(n=None): for i in xrange(n): e = bluechips.model.Expenditure(random.choice(users), Currency(random.randint(1000, 100000))) - meta.Session.save(e) + meta.Session.add(e) e.even_split() meta.Session.commit() diff --git a/bluechips/tests/splits/test_fixed.py b/bluechips/tests/splits/test_fixed.py index 0de3786..46fcd84 100644 --- a/bluechips/tests/splits/test_fixed.py +++ b/bluechips/tests/splits/test_fixed.py @@ -14,7 +14,7 @@ class TestSplitFixed(TestCase): e = model.Expenditure(meta.Session.query(model.User).first(), Currency("100")) - meta.Session.save(e) + meta.Session.add(e) e.even_split() meta.Session.commit() @@ -34,7 +34,7 @@ class TestSplitFixed(TestCase): users = meta.Session.query(model.User).all() e = model.Expenditure(users[0], Currency("100")) - meta.Session.save(e) + meta.Session.add(e) split_dict = {users[0]: Decimal("20"), users[1]: Decimal("80")} @@ -61,7 +61,7 @@ class TestSplitFixed(TestCase): users = meta.Session.query(model.User).all() e = model.Expenditure(users[0], Currency("100.00")) - meta.Session.save(e) + meta.Session.add(e) split_dict = {users[0]: Decimal(10), users[1]: Decimal(15)} @@ -87,7 +87,7 @@ class TestSplitFixed(TestCase): users = meta.Session.query(model.User).all() e = model.Expenditure(users[0], Currency("100.00")) - meta.Session.save(e) + meta.Session.add(e) # Force a split that will result in needing to distribute # pennies