def campaign_object(template_object, smtp_object):
"""Return a Campaign object."""
return Campaign(
name="RVXXX1-C1",
launch_date="01/01/2025 13:00",
completed_date="01/01/2025 14:00",
url="http://bad.domain/camp1",
group_name="RVXXX1-G1",
template=template_object,
smtp=smtp_object,
)
def build_campaigns(assessment, campaign_number, template_smtp):
"""Build a campaign."""
# Set up component holders
logging.info("Building Campaign %s", assessment.id)
campaign = Campaign(name=assessment.id)
campaign_tz = pytz.timezone(assessment.timezone)
# Get Launch Time
while True:
campaign.launch_date = get_time_input("start", assessment.timezone)
if campaign.launch_date > datetime.now(campaign_tz).isoformat():
break
else:
logging.error("Launch date is not after the current datetime")
while True:
campaign.complete_date = get_time_input("end", assessment.timezone)
if campaign.complete_date > campaign.launch_date:
pass # Do nothing yet and continue checks
else:
logging.error("Complete date is not after launch date.")
if campaign.complete_date > datetime.now(campaign_tz).isoformat():
break # Valid input, break out of loop
else:
logging.error("Complete date is not after the current datetime.")
campaign.smtp, campaign.template = import_email(
assessment, campaign_number, template_smtp
)
# Select Group:
campaign.group_name = select_group(assessment)
# Select page:
campaign.page_name = select_page(assessment)
campaign.url = prompt(
" Campaign URL: ",
default="http://" + assessment.domain,
validator=BlankInputValidator(),
)
campaign = review_campaign(campaign)
logging.info("Successfully Added Campaign {}".format(campaign.name))
return campaign
def multiple_campaign_object(template_object, smtp_object):
"""Return list of campaign objects."""
campaigns = list()
for x in range(1, 8):
campaigns.append(
Campaign(
name=f"RVXXX1-C{x}",
launch_date=f"01/0{x}/2025 13:00",
completed_date=f"01/0{x}/2025 14:00",
url=f"http://bad.domain/camp{x}",
group_name="RVXXX1-G1",
template=template_object,
smtp=smtp_object,
))
# Make a campaign from a different assessment.
campaigns[6].name = "RVXXX2-C7"
return campaigns
def build_campaigns(assessment, campaign_number, template_smtp):
"""Build a campaign."""
# Set up component holders
logging.info(f"Building Campaign {assessment.id}-C{campaign_number}")
campaign = Campaign(name=f"{assessment.id}-C{campaign_number}")
# Get Launch Time
campaign.launch_date = get_time_input("start", assessment.timezone)
while True:
campaign.complete_date = get_time_input("end", assessment.timezone)
if campaign.complete_date > campaign.launch_date:
break
else:
logging.error("Complete Date is not after Launch Date.")
campaign.smtp, campaign.template = import_email(assessment,
campaign_number,
template_smtp)
# Select Group:
campaign.group_name = select_group(assessment)
# Select page:
campaign.page_name = select_page(assessment)
campaign.url = prompt(
" Campaign URL: ",
default="http://" + assessment.domain,
validator=BlankInputValidator(),
)
campaign = review_campaign(campaign)
logging.info("Successfully Added Campaign {}".format(campaign.name))
return campaign
def test_campaign_parse(self, campaign_object, campaign_json):
"""Test parsing of campaign JSON."""
assert campaign_object.as_dict() == Campaign.parse(
campaign_json).as_dict()