Browse Source

update

pull/3/head
Jeff Becker 8 years ago
parent
commit
08779e3f89
  1. 4
      baddie-detector/baddiefinder/__main__.py
  2. 3
      baddie-detector/baddiefinder/processor.py

4
baddie-detector/baddiefinder/__main__.py

@ -22,8 +22,8 @@ def main(): @@ -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()

3
baddie-detector/baddiefinder/processor.py

@ -21,6 +21,9 @@ class BaddieProcessor: @@ -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

Loading…
Cancel
Save