Skip to content
Snippets Groups Projects
Commit 4e480335 authored by Rajmund Hruška's avatar Rajmund Hruška
Browse files

Merge remote-tracking branch 'origin/judiny-bugfix-#7556-abuse_group_mail' into devel

parents 983b96cd 373220e8
No related branches found
No related tags found
No related merge requests found
......@@ -375,7 +375,7 @@ class MentatNetmngrScript(mentat.script.fetcher.FetcherScript):
# from the whois file with those from the database, the name of each group from database is computed
# in the same way as the name of a group from the whois file.
emails = list(set((abg.settings_rep.emails_low or []) + (abg.settings_rep.emails_medium or []) + (abg.settings_rep.emails_high or []) + (abg.settings_rep.emails_critical or [])))
name = ','.join(sorted(map(str.lower, emails if emails else [abg.name])))
name = ','.join(sorted(map(str.lower, emails)))
abuse_group_dict[name] = abg
self.logger.info("Number of abuse groups in database: %d", len(abuse_groups))
......@@ -458,13 +458,10 @@ class MentatNetmngrScript(mentat.script.fetcher.FetcherScript):
if abuse_group in abuse_group_dict:
s = abuse_group_dict[abuse_group].settings_rep
if not (s.emails_low or s.emails_medium or s.emails_high or s.emails_critical):
emails['low'].append([abuse_group])
else:
emails['low'].append(sorted(s.emails_low or []))
emails['medium'].append(sorted(s.emails_medium or []))
emails['high'].append(sorted(s.emails_high or []))
emails['critical'].append(sorted(s.emails_critical or []))
emails['low'].append(sorted(s.emails_low or []))
emails['medium'].append(sorted(s.emails_medium or []))
emails['high'].append(sorted(s.emails_high or []))
emails['critical'].append(sorted(s.emails_critical or []))
for severity in ['low', 'medium', 'high', 'critical']:
if not all(x == emails[severity][0] for x in emails[severity]):
......
......@@ -265,7 +265,6 @@ class TestMentatReportsUtils(unittest.TestCase):
self.assertEqual(str(reporting_settings.filters), "[<Filter(name='FLT1')>, <Filter(name='FLT2')>]")
self.assertEqual(str(reporting_settings.networks), "[<Network(netname='UNET1',network='10.0.0.0/8')>]")
self.assertEqual(reporting_settings.mode, 'summary')
self.assertEqual(reporting_settings.emails, (['abuse@cesnet.cz'], [], [], []))
self.assertEqual(reporting_settings.redirect, False)
self.assertEqual(reporting_settings.template, 'default')
self.assertEqual(reporting_settings.locale, 'en')
......
......@@ -62,9 +62,7 @@ class ReportingSettings: # pylint: disable=locally-disabled,too-many-instance-at
@staticmethod
def _init_emails(group_value, settings):
if settings.emails_low or settings.emails_medium or settings.emails_high or settings.emails_critical:
return (settings.emails_low, settings.emails_medium, settings.emails_high, settings.emails_critical)
return ([group_value], [], [], [])
return (settings.emails_low, settings.emails_medium, settings.emails_high, settings.emails_critical)
@staticmethod
def _init_mode(group_value, **kwargs):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment