This commit is contained in:
Jeff Becker 2016-11-24 08:30:45 -05:00
parent e548512f1e
commit 08779e3f89
2 changed files with 5 additions and 2 deletions

View file

@ -22,8 +22,8 @@ def main():
p = processor.BaddieProcessor([f])
netdb.inspect(p.hook)
with open(s.get("output", "file", fallback="baddies.txt"), 'w') as f:
p.write_blocklist(f)
print ('wrote {} blocklist entries'.format(p.write_blocklist(f)))
if __name__ == "__main__":
main()

View file

@ -21,6 +21,9 @@ class BaddieProcessor:
self._baddies[addr] = reason
def write_blocklist(self, f):
wrote = 0
f.write('# baddies blocklist generated on {}\n'.format(datetime.datetime.now()))
for k in self._baddies:
f.write('{}:{}\n'.format(self._baddies[k], k))
wrote += 1
return wrote