def save(response_elements): session = DBSession() network = LinkShareNetwork(**response_elements['network']) if not session.query(LinkShareNetwork).filter_by(id=response_elements['network']['id']): session.add(network) merchant = LinkShareMerchant(network_id=network.id, **response_elements['merchant']) session.add(merchant) deal = LinkShareDeal(network_id=network.id, merchant_id=merchant.id, **response_elements['deal']) deal.categories=[LinkShareCategory(**category) for category in response_elements['categories']] deal.promotion_types=[LinkSharePromotionType(**promotion_type) for promotion_type in response_elements['promotion_types']] session.add(deal) session.commit()
class BaseTestCase(unittest.TestCase): @classmethod def setUpClass(cls): cls.engine = create_engine('sqlite://', echo=False) cls.connection = cls.engine.connect() initialize_sql(cls.connection) def setUp(self): # begin a non-ORM transaction self.trans = self.connection.begin() # bind an individual Session to the connection self.session = DBSession() def tearDown(self): self.trans.rollback() self.session.close()
def setUp(self): # begin a non-ORM transaction self.trans = self.connection.begin() # bind an individual Session to the connection self.session = DBSession()