def test_gtk_calendar_and_pydate_rounding(self): gtk_calendar = Gtk.Calendar() year, month, day = 2018, 4, 31 # April 31st, 2018 (does not exist) gtk_calendar.select_month(month - 1, year) # GTK months start at 0 gtk_calendar.select_day(day) date = gui_utilities.gtk_calendar_get_pydate(gtk_calendar) self.assertEquals((date.year, date.month, date.day), (year, month, 30))
def test_gtk_calendar_and_pydate_rounding(self):
gtk_calendar = Gtk.Calendar()
year, month, day = 2018, 4, 31 # April 31st, 2018 (does not exist)
gtk_calendar.select_month(month - 1, year) # GTK months start at 0
gtk_calendar.select_day(day)
date = gui_utilities.gtk_calendar_get_pydate(gtk_calendar)
self.assertEquals((date.year, date.month, date.day), (year, month, 30))
def test_gtk_calendar_and_pydate(self): gtk_calendar = Gtk.Calendar() year, month, day = 2018, 2, 28 # February 28th, 2018 gtk_calendar.select_month(month - 1, year) # GTK months start at 0 gtk_calendar.select_day(day) date = gui_utilities.gtk_calendar_get_pydate(gtk_calendar) self.assertIsInstance(date, datetime.date) self.assertEquals((date.year, date.month, date.day), (year, month, day)) month, day = 1, 27 # January 27th, 2018 date = date.replace(month=month, day=day) gui_utilities.gtk_calendar_set_pydate(gtk_calendar, date) self.assertEquals(gtk_calendar.get_date(), (year, month - 1, day))
def test_gtk_calendar_and_pydate(self):
gtk_calendar = Gtk.Calendar()
year, month, day = 2018, 2, 28 # February 28th, 2018
gtk_calendar.select_month(month - 1, year) # GTK months start at 0
gtk_calendar.select_day(day)
date = gui_utilities.gtk_calendar_get_pydate(gtk_calendar)
self.assertIsInstance(date, datetime.date)
self.assertEquals((date.year, date.month, date.day),
(year, month, day))
month, day = 1, 27 # January 27th, 2018
date = date.replace(month=month, day=day)
gui_utilities.gtk_calendar_set_pydate(gtk_calendar, date)
self.assertEquals(gtk_calendar.get_date(), (year, month - 1, day))
def signal_calendar_prev(self, calendar): today = datetime.date.today() calendar_day = gui_utilities.gtk_calendar_get_pydate(calendar) if calendar_day >= today: return gui_utilities.gtk_calendar_set_pydate(calendar, today)
def signal_assistant_apply(self, _):
self._close_ready = False
# have to do it this way because the next page will be selected when the apply signal is complete
set_current_page = lambda page_name: self.assistant.set_current_page(max(0, self._page_titles[page_name] - 1))
# get and validate the campaign name
campaign_name = self.gobjects['entry_campaign_name'].get_text()
campaign_name = campaign_name.strip()
if not campaign_name:
gui_utilities.show_dialog_error('Invalid Campaign Name', self.parent, 'A unique and valid campaign name must be specified.')
set_current_page('Basic Settings')
return True
# validate the company
company_id = None
if self.gobjects['radiobutton_company_existing'].get_active():
company_id = self._get_company_existing_id()
if company_id is None:
gui_utilities.show_dialog_error('Invalid Company', self.parent, 'A valid existing company must be specified.')
set_current_page('Company')
return True
elif self.gobjects['radiobutton_company_new'].get_active():
company_id = self._get_company_new_id()
if company_id is None:
gui_utilities.show_dialog_error('Invalid Company', self.parent, 'The new company settings are invalid.')
set_current_page('Company')
return True
# get and validate the campaign expiration
expiration = None
if self.gobjects['checkbutton_expire_campaign'].get_property('active'):
expiration = datetime.datetime.combine(
gui_utilities.gtk_calendar_get_pydate(self.gobjects['calendar_campaign_expiration']),
datetime.time(
int(self.gobjects['spinbutton_campaign_expiration_hour'].get_value()),
int(self.gobjects['spinbutton_campaign_expiration_minute'].get_value())
)
)
expiration = utilities.datetime_local_to_utc(expiration)
if self.is_new_campaign and expiration <= datetime.datetime.now():
gui_utilities.show_dialog_error('Invalid Campaign Expiration', self.parent, 'The expiration date is set in the past.')
set_current_page('Expiration')
return True
properties = {}
# point of no return
campaign_description = self.gobjects['entry_campaign_description'].get_text()
campaign_description = campaign_description.strip()
if campaign_description == '':
campaign_description = None
if self.campaign_id:
properties['name'] = self.campaign_name
properties['description'] = campaign_description
cid = self.campaign_id
else:
try:
cid = self.application.rpc('campaign/new', campaign_name, description=campaign_description)
except AdvancedHTTPServer.AdvancedHTTPServerRPCError as error:
if not error.is_remote_exception:
raise error
if not error.remote_exception['name'] == 'exceptions.ValueError':
raise error
error_message = error.remote_exception.get('message', 'an unknown error occurred').capitalize() + '.'
gui_utilities.show_dialog_error('Failed To Create Campaign', self.parent, error_message)
set_current_page('Basic Settings')
return True
properties['campaign_type_id'] = self._get_tag_from_combobox(self.gobjects['combobox_campaign_type'], 'campaign_types')
properties['company_id'] = company_id
properties['expiration'] = expiration
properties['reject_after_credentials'] = self.gobjects['checkbutton_reject_after_credentials'].get_property('active')
self.application.rpc('db/table/set', 'campaigns', cid, properties.keys(), properties.values())
should_subscribe = self.gobjects['checkbutton_alert_subscribe'].get_property('active')
if should_subscribe != self.application.rpc('campaign/alerts/is_subscribed', cid):
if should_subscribe:
self.application.rpc('campaign/alerts/subscribe', cid)
else:
self.application.rpc('campaign/alerts/unsubscribe', cid)
self.application.emit('campaign-changed', cid)
self._close_ready = True
return
def signal_calendar_prev(self, calendar): today = datetime.date.today() calendar_day = gui_utilities.gtk_calendar_get_pydate(calendar) if calendar_day >= today: return gui_utilities.gtk_calendar_set_pydate(calendar, today)
def signal_assistant_apply(self, _):
self._close_ready = False
# have to do it this way because the next page will be selected when the apply signal is complete
set_current_page = lambda page_name: self.assistant.set_current_page(max(0, self._page_titles[page_name] - 1))
# get and validate the campaign name
campaign_name = self.gobjects['entry_campaign_name'].get_text()
campaign_name = campaign_name.strip()
if not campaign_name:
gui_utilities.show_dialog_error('Invalid Campaign Name', self.parent, 'A unique and valid campaign name must be specified.')
set_current_page('Basic Settings')
return True
properties = {}
# validate the credential validation regular expressions
for field in ('username', 'password', 'mfa_token'):
regex = self.gobjects['entry_validation_regex_' + field].get_text()
if regex:
try:
re.compile(regex)
except re.error:
label = self.gobjects['label_validation_regex_' + field].get_text()
gui_utilities.show_dialog_error('Invalid Regex', self.parent, "The '{0}' regular expression is invalid.".format(label))
return True
else:
regex = None # keep empty strings out of the database
properties['credential_regex_' + field] = regex
# validate the company
company_id = None
if self.gobjects['radiobutton_company_existing'].get_active():
company_id = self._get_company_existing_id()
if company_id is None:
gui_utilities.show_dialog_error('Invalid Company', self.parent, 'A valid existing company must be specified.')
set_current_page('Company')
return True
elif self.gobjects['radiobutton_company_new'].get_active():
company_id = self._get_company_new_id()
if company_id is None:
gui_utilities.show_dialog_error('Invalid Company', self.parent, 'The new company settings are invalid.')
set_current_page('Company')
return True
# get and validate the campaign expiration
expiration = None
if self.gobjects['checkbutton_expire_campaign'].get_property('active'):
expiration = datetime.datetime.combine(
gui_utilities.gtk_calendar_get_pydate(self.gobjects['calendar_campaign_expiration']),
self._expiration_time.time
)
expiration = utilities.datetime_local_to_utc(expiration)
if self.is_new_campaign and expiration <= datetime.datetime.now():
gui_utilities.show_dialog_error('Invalid Campaign Expiration', self.parent, 'The expiration date is set in the past.')
set_current_page('Expiration')
return True
# point of no return
campaign_description = self.get_entry_value('campaign_description')
if self.campaign_id:
properties['name'] = self.campaign_name
properties['description'] = campaign_description
cid = self.campaign_id
else:
try:
cid = self.application.rpc('campaign/new', campaign_name, description=campaign_description)
except advancedhttpserver.RPCError as error:
if not error.is_remote_exception:
raise error
if not error.remote_exception['name'] == 'exceptions.ValueError':
raise error
error_message = error.remote_exception.get('message', 'an unknown error occurred').capitalize() + '.'
gui_utilities.show_dialog_error('Failed To Create Campaign', self.parent, error_message)
set_current_page('Basic Settings')
return True
self.application.emit('campaign-created', cid)
properties['campaign_type_id'] = self._get_tag_from_combobox(self.gobjects['combobox_campaign_type'], 'campaign_types')
properties['company_id'] = company_id
properties['expiration'] = expiration
properties['max_credentials'] = (1 if self.gobjects['checkbutton_reject_after_credentials'].get_property('active') else None)
self.application.rpc('db/table/set', 'campaigns', cid, tuple(properties.keys()), tuple(properties.values()))
should_subscribe = self.gobjects['checkbutton_alert_subscribe'].get_property('active')
if should_subscribe != self.application.rpc('campaign/alerts/is_subscribed', cid):
if should_subscribe:
self.application.rpc('campaign/alerts/subscribe', cid)
else:
self.application.rpc('campaign/alerts/unsubscribe', cid)
self.application.emit('campaign-changed', cid)
self._close_ready = True
return
def signal_assistant_apply(self, _):
self._close_ready = False
# have to do it this way because the next page will be selected when the apply signal is complete
set_current_page = lambda page_name: self.assistant.set_current_page(max(0, self._page_titles[page_name] - 1))
# get and validate the campaign name
campaign_name = self.gobjects['entry_campaign_name'].get_text()
campaign_name = campaign_name.strip()
if not campaign_name:
gui_utilities.show_dialog_error('Invalid Campaign Name', self.parent, 'A unique and valid campaign name must be specified.')
set_current_page('Basic Settings')
return True
properties = {}
# validate the credential validation regular expressions
for field in ('username', 'password', 'mfa_token'):
regex = self.gobjects['entry_validation_regex_' + field].get_text()
if regex:
try:
re.compile(regex)
except re.error:
label = self.gobjects['label_validation_regex_' + field].get_text()
gui_utilities.show_dialog_error('Invalid Regex', self.parent, "The '{0}' regular expression is invalid.".format(label))
return True
else:
regex = None # keep empty strings out of the database
properties['credential_regex_' + field] = regex
# validate the company
company_id = None
if self.gobjects['radiobutton_company_existing'].get_active():
company_id = self._get_company_existing_id()
if company_id is None:
gui_utilities.show_dialog_error('Invalid Company', self.parent, 'A valid existing company must be specified.')
set_current_page('Company')
return True
elif self.gobjects['radiobutton_company_new'].get_active():
company_id = self._get_company_new_id()
if company_id is None:
gui_utilities.show_dialog_error('Invalid Company', self.parent, 'The new company settings are invalid.')
set_current_page('Company')
return True
# get and validate the campaign expiration
expiration = None
if self.gobjects['checkbutton_expire_campaign'].get_property('active'):
expiration = datetime.datetime.combine(
gui_utilities.gtk_calendar_get_pydate(self.gobjects['calendar_campaign_expiration']),
self._expiration_time.time
)
expiration = utilities.datetime_local_to_utc(expiration)
if self.is_new_campaign and expiration <= datetime.datetime.now():
gui_utilities.show_dialog_error('Invalid Campaign Expiration', self.parent, 'The expiration date is set in the past.')
set_current_page('Expiration')
return True
# point of no return
campaign_description = self.get_entry_value('campaign_description')
if self.campaign_id:
properties['name'] = self.campaign_name
properties['description'] = campaign_description
cid = self.campaign_id
else:
try:
cid = self.application.rpc('campaign/new', campaign_name, description=campaign_description)
properties['name'] = campaign_name
except advancedhttpserver.RPCError as error:
if not error.is_remote_exception:
raise error
if not error.remote_exception['name'] == 'exceptions.ValueError':
raise error
error_message = error.remote_exception.get('message', 'an unknown error occurred').capitalize() + '.'
gui_utilities.show_dialog_error('Failed To Create Campaign', self.parent, error_message)
set_current_page('Basic Settings')
return True
self.application.emit('campaign-created', cid)
properties['campaign_type_id'] = self._get_tag_from_combobox(self.gobjects['combobox_campaign_type'], 'campaign_types')
properties['company_id'] = company_id
properties['expiration'] = expiration
properties['max_credentials'] = (1 if self.gobjects['checkbutton_reject_after_credentials'].get_property('active') else None)
self.application.rpc('db/table/set', 'campaigns', cid, tuple(properties.keys()), tuple(properties.values()))
should_subscribe = self.gobjects['checkbutton_alert_subscribe'].get_property('active')
if should_subscribe != self.application.rpc('campaign/alerts/is_subscribed', cid):
if should_subscribe:
self.application.rpc('campaign/alerts/subscribe', cid)
else:
self.application.rpc('campaign/alerts/unsubscribe', cid)
self.application.emit('campaign-changed', cid)
old_cid = self.config.get('campaign_id')
self.config['campaign_id'] = cid
self.config['campaign_name'] = properties['name']
_kpm_pathing = self._get_kpm_path()
if all(_kpm_pathing):
if not self.application.main_tabs['mailer'].emit('message-data-import', _kpm_pathing.kpm_file, _kpm_pathing.destination_folder):
gui_utilities.show_dialog_info('Failure', self.parent, 'Failed to import the message configuration.')
else:
gui_utilities.show_dialog_info('Success', self.parent, 'Successfully imported the message configuration.')
url_model, url_iter = self.gobjects['treeselection_url_selector'].get_selected()
if url_iter:
selected_row = []
for column_n in range(0, url_model.get_n_columns()):
selected_row.append(url_model.get_value(url_iter, column_n))
selected_row = _ModelNamedRow(*selected_row)
self.config['mailer.webserver_url'] = selected_row.url
self.application.emit('campaign-set', old_cid, cid)
self._close_ready = True
return
def signal_assistant_apply(self, _):
self._close_ready = False
# have to do it this way because the next page will be selected when the apply signal is complete
set_current_page = lambda page_name: self.assistant.set_current_page(max(0, self._page_titles[page_name] - 1))
# get and validate the campaign name
campaign_name = self.gobjects["entry_campaign_name"].get_text()
campaign_name = campaign_name.strip()
if not campaign_name:
gui_utilities.show_dialog_error(
"Invalid Campaign Name", self.parent, "A unique and valid campaign name must be specified."
)
set_current_page("Basic Settings")
return True
# validate the company
company_id = None
if self.gobjects["radiobutton_company_existing"].get_active():
company_id = self._get_company_existing_id()
if company_id is None:
gui_utilities.show_dialog_error(
"Invalid Company", self.parent, "A valid existing company must be specified."
)
set_current_page("Company")
return True
elif self.gobjects["radiobutton_company_new"].get_active():
company_id = self._get_company_new_id()
if company_id is None:
gui_utilities.show_dialog_error("Invalid Company", self.parent, "The new company settings are invalid.")
set_current_page("Company")
return True
# get and validate the campaign expiration
expiration = None
if self.gobjects["checkbutton_expire_campaign"].get_property("active"):
expiration = datetime.datetime.combine(
gui_utilities.gtk_calendar_get_pydate(self.gobjects["calendar_campaign_expiration"]),
datetime.time(
int(self.gobjects["spinbutton_campaign_expiration_hour"].get_value()),
int(self.gobjects["spinbutton_campaign_expiration_minute"].get_value()),
),
)
expiration = utilities.datetime_local_to_utc(expiration)
if expiration <= datetime.datetime.now():
gui_utilities.show_dialog_error(
"Invalid Campaign Expiration", self.parent, "The expiration date is set in the past."
)
set_current_page("Expiration")
return True
properties = {}
# point of no return
campaign_description = self.gobjects["entry_campaign_description"].get_text()
campaign_description = campaign_description.strip()
if campaign_description == "":
campaign_description = None
if self.campaign_id:
properties["name"] = self.campaign_name
properties["description"] = campaign_description
cid = self.campaign_id
else:
try:
cid = self.application.rpc("campaign/new", campaign_name, description=campaign_description)
except AdvancedHTTPServer.AdvancedHTTPServerRPCError as error:
if not error.is_remote_exception:
raise error
if not error.remote_exception["name"] == "exceptions.ValueError":
raise error
error_message = error.remote_exception.get("message", "an unknown error occurred").capitalize() + "."
gui_utilities.show_dialog_error("Failed To Create Campaign", self.parent, error_message)
set_current_page("Basic Settings")
return True
properties["campaign_type_id"] = self._get_tag_from_combobox(
self.gobjects["combobox_campaign_type"], "campaign_types"
)
properties["company_id"] = company_id
properties["expiration"] = expiration
properties["reject_after_credentials"] = self.gobjects["checkbutton_reject_after_credentials"].get_property(
"active"
)
self.application.rpc("db/table/set", "campaigns", cid, properties.keys(), properties.values())
should_subscribe = self.gobjects["checkbutton_alert_subscribe"].get_property("active")
if should_subscribe != self.application.rpc("campaign/alerts/is_subscribed", cid):
if should_subscribe:
self.application.rpc("campaign/alerts/subscribe", cid)
else:
self.application.rpc("campaign/alerts/unsubscribe", cid)
self._close_ready = True
return
def signal_assistant_apply(self, _):
self._close_ready = False
# have to do it this way because the next page will be selected when the apply signal is complete
set_current_page = lambda page_name: self.assistant.set_current_page(
max(0, self._page_titles[page_name] - 1))
# get and validate the campaign name
campaign_name = self.gobjects['entry_campaign_name'].get_text()
campaign_name = campaign_name.strip()
if not campaign_name:
gui_utilities.show_dialog_error(
'Invalid Campaign Name', self.parent,
'A unique and valid campaign name must be specified.')
set_current_page('Basic Settings')
return True
# validate the company
company_id = None
if self.gobjects['radiobutton_company_existing'].get_active():
company_id = self._get_company_existing_id()
if company_id is None:
gui_utilities.show_dialog_error(
'Invalid Company', self.parent,
'A valid existing company must be specified.')
set_current_page('Company')
return True
elif self.gobjects['radiobutton_company_new'].get_active():
company_id = self._get_company_new_id()
if company_id is None:
gui_utilities.show_dialog_error(
'Invalid Company', self.parent,
'The new company settings are invalid.')
set_current_page('Company')
return True
# get and validate the campaign expiration
expiration = None
if self.gobjects['checkbutton_expire_campaign'].get_property('active'):
expiration = datetime.datetime.combine(
gui_utilities.gtk_calendar_get_pydate(
self.gobjects['calendar_campaign_expiration']),
datetime.time(
int(self.gobjects['spinbutton_campaign_expiration_hour'].
get_value()),
int(self.gobjects['spinbutton_campaign_expiration_minute'].
get_value())))
expiration = utilities.datetime_local_to_utc(expiration)
if self.is_new_campaign and expiration <= datetime.datetime.now():
gui_utilities.show_dialog_error(
'Invalid Campaign Expiration', self.parent,
'The expiration date is set in the past.')
set_current_page('Expiration')
return True
properties = {}
# point of no return
campaign_description = self.gobjects[
'entry_campaign_description'].get_text()
campaign_description = campaign_description.strip()
if campaign_description == '':
campaign_description = None
if self.campaign_id:
properties['name'] = self.campaign_name
properties['description'] = campaign_description
cid = self.campaign_id
else:
try:
cid = self.application.rpc('campaign/new',
campaign_name,
description=campaign_description)
except AdvancedHTTPServer.AdvancedHTTPServerRPCError as error:
if not error.is_remote_exception:
raise error
if not error.remote_exception[
'name'] == 'exceptions.ValueError':
raise error
error_message = error.remote_exception.get(
'message', 'an unknown error occurred').capitalize() + '.'
gui_utilities.show_dialog_error('Failed To Create Campaign',
self.parent, error_message)
set_current_page('Basic Settings')
return True
properties['campaign_type_id'] = self._get_tag_from_combobox(
self.gobjects['combobox_campaign_type'], 'campaign_types')
properties['company_id'] = company_id
properties['expiration'] = expiration
properties['reject_after_credentials'] = self.gobjects[
'checkbutton_reject_after_credentials'].get_property('active')
self.application.rpc('db/table/set', 'campaigns', cid,
properties.keys(), properties.values())
should_subscribe = self.gobjects[
'checkbutton_alert_subscribe'].get_property('active')
if should_subscribe != self.application.rpc(
'campaign/alerts/is_subscribed', cid):
if should_subscribe:
self.application.rpc('campaign/alerts/subscribe', cid)
else:
self.application.rpc('campaign/alerts/unsubscribe', cid)
self.application.emit('campaign-changed', cid)
self._close_ready = True
return
