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()