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