def get_module_campaigns(ctx, module_id): scoped_session = start_engine() session = scoped_session() facade = CampaignFacade(session) campaigns = facade.get_campaigns_from_module_id(module_id=module_id) session.close() return campaigns
def delete_module(self, module: Module): campaign_facade = CampaignFacade(self.session) for campaign in campaign_facade.get_campaigns_from_module_id( module.id): campaign_facade.delete_campaign(campaign) self.session.delete(self.get_module_by_id(module.id)) self.session.commit()
def get_campaign_by_id(ctx, campaign_id): scoped_session = start_engine() session = scoped_session() facade = CampaignFacade(session) campaign = facade.get_campaign_by_id(campaign_id=campaign_id) session.close() return campaign
def create_campaign(ctx, campaign): campaign_id = -1 if campaign.id is None: scoped_session = start_engine() session = scoped_session() campaign_facade = CampaignFacade(session) campaign_facade.create_campaign(campaign=campaign) campaign_id = campaign_facade.get_campaign_by_name( name=campaign.name).id session.close() return campaign_id
def test_create(): scoped_session = start_engine() session = scoped_session() campaign = Campaign(name='sample', date=time.localtime(time.time()), module_id=1) facade = CampaignFacade(session) facade.create_campaign(campaign=campaign) query = session.query(Campaign).filter_by(id=campaign.id) campaign_query = query.all() assert len(campaign_query) == 1 assert campaign_query[0] is campaign session.close()
def test_delete(): scoped_session = start_engine() session = scoped_session() # get first campaign query = session.query(Campaign).filter_by(id=1) campaign_query = query.all() campaign = campaign_query[0] # delete facade = CampaignFacade(session) facade.delete_campaign(campaign) # get campaign again query = session.query(Campaign).filter_by(id=1) campaign_query = query.all() assert len(campaign_query) == 0 session.close()
def test_update(): scoped_session = start_engine() session = scoped_session() # get first campaign query = session.query(Campaign).filter_by(id=1) campaign_query = query.all() campaign = campaign_query[0] campaign.name = campaign.name + "_update" # update facade = CampaignFacade(session) facade.update_campaign(campaign) # get campaign again query = session.query(Campaign).filter_by(id=1) campaign_query = query.all() assert len(campaign_query) == 1 assert campaign_query[0] is campaign session.close()
def update_campaign(ctx, campaign): scoped_session = start_engine() session = scoped_session() facade = CampaignFacade(session) facade.update_module(module=campaign) session.close()
def delete_campaign(ctx, campaign): scoped_session = start_engine() session = scoped_session() facade = CampaignFacade(session) facade.delete_module(campaign=campaign) session.close()