From 08779e3f89b9a5f2c1447cfe2323fd73bec9f99f Mon Sep 17 00:00:00 2001 From: Jeff Becker Date: Thu, 24 Nov 2016 08:30:45 -0500 Subject: [PATCH] update --- baddie-detector/baddiefinder/__main__.py | 4 ++-- baddie-detector/baddiefinder/processor.py | 3 +++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/baddie-detector/baddiefinder/__main__.py b/baddie-detector/baddiefinder/__main__.py index 7a13cec..fa92c2f 100644 --- a/baddie-detector/baddiefinder/__main__.py +++ b/baddie-detector/baddiefinder/__main__.py @@ -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() diff --git a/baddie-detector/baddiefinder/processor.py b/baddie-detector/baddiefinder/processor.py index 634475a..8eeefef 100644 --- a/baddie-detector/baddiefinder/processor.py +++ b/baddie-detector/baddiefinder/processor.py @@ -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