CIRCL
/
url-abuse
mirror of https://github.com/CIRCL/url-abuse
2
0
Fork 0
Code Issues Releases Wiki Activity

fix: Do not send more than one email/day/hostname to the ticketing system. 

Fix #10
pull/14/head
Raphaël Vinot 2019-03-14 10:01:14 +01:00
parent 62ce1b7dc0
commit d95abb15cb
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
urlabuse/urlabuse.py
View File

@ -81,14 +81,18 @@ class Query():
day = date.today().isoformat() day = date.today().isoformat()
else: else:
day = day.isoformat() day = day.isoformat()
return self.cache.sismember(f'{day}_mails', url) self.fex.decode(url)
host = self.fex.get_host()
return self.cache.sismember(f'{day}_mails', host)
def set_mail_sent(self, url, day=None): def set_mail_sent(self, url, day=None):
if day is None: if day is None:
day = date.today().isoformat() day = date.today().isoformat()
else: else:
day = day.isoformat() day = day.isoformat()
return self.cache.sadd(f'{day}_mails', url) self.fex.decode(url)
host = self.fex.get_host()
return self.cache.sadd(f'{day}_mails', host)
def is_valid_url(self, url): def is_valid_url(self, url):
cached = self._cache_get(url, 'valid') cached = self._cache_get(url, 'valid')