def delete_historical_data(purge_data_duration):
existingSettingObj = Settings.query.filter(Settings.name == 'purge_data_duration').first()
if existingSettingObj:
current_app.logger.info("Delete duration already set")
else:
Settings.create(name='purge_data_duration', setting=purge_data_duration)
def update_vt_match_count(vt_min_match_count):
existingSettingObj = Settings.query.filter(
Settings.name == 'virustotal_min_match_count').first()
if existingSettingObj:
current_app.logger.info("Match count already set")
else:
Settings.create(name='virustotal_min_match_count',
setting=vt_min_match_count)
def update_or_create_setting(name, setting):
settings_obj = Settings.query.filter(Settings.name == name).first()
if settings_obj:
settings_obj.setting = setting
return settings_obj.update(settings_obj)
else:
return Settings.create(name=name, setting=setting)
def test_email_receipient_and_sender_with_data(self, client, url_prefix,
token, settings):
"""
Test-Case with valid credentials along with existing settings data,
expected output:- status is success
"""
self.payload['email'] = '*****@*****.**'
self.payload['smtpPort'] = None
self.payload['smtpAddress'] = 'smtp.gmail.com'
self.payload['password'] = '******'
self.payload['emailRecipients'] = '[email protected],[email protected]'
settings = Settings.create(setting=json.dumps({
"email":
"*****@*****.**",
"emailRecipients": ["*****@*****.**", "*****@*****.**"],
"password":
"******",
"smtpAddress":
"smtp.googlemail.com",
"smtpPort":
465
}),
name='plgx_config_all_settings')
resp = client.post(url_prefix + '/email/test',
headers={'x-access-token': token},
data=self.payload)
assert resp.status_code == 200
response_dict = json.loads(resp.data)
assert response_dict['status'] == 'success'
def update_settings(purge_data_duration, alert_aggregation_duration):
purge_dur_setting = Settings.query.filter(
Settings.name == 'purge_data_duration').first()
alert_aggr_dur_setting = Settings.query.filter(
Settings.name == 'alert_aggregation_duration').first()
if purge_dur_setting:
current_app.logger.info("Purge duration is already set")
else:
Settings.create(name='purge_data_duration',
setting=purge_data_duration)
if alert_aggr_dur_setting:
current_app.logger.info("Alert aggregation duration is already set")
else:
Settings.create(name='alert_aggregation_duration',
setting=alert_aggregation_duration)
def save_or_update_setting(setting, v, k):
if setting:
setting.setting = v
setting.update(setting)
else:
Settings.create(name=k, setting=v)
def create_settings(name, setting):
return Settings.create(name=name, setting=setting)