From e14e593872ccd3e2fb02df778c5cd9e9c95a0abb Mon Sep 17 00:00:00 2001 From: Ad Schellevis Date: Sun, 6 Sep 2015 19:03:26 +0200 Subject: [PATCH] style fix list_osfp.py --- src/opnsense/scripts/filter/list_osfp.py | 33 ++++++++++++------------ 1 file changed, 17 insertions(+), 16 deletions(-) diff --git a/src/opnsense/scripts/filter/list_osfp.py b/src/opnsense/scripts/filter/list_osfp.py index 8b7397098..d2425f47d 100755 --- a/src/opnsense/scripts/filter/list_osfp.py +++ b/src/opnsense/scripts/filter/list_osfp.py @@ -34,20 +34,21 @@ import os import sys import ujson -result=[] -with tempfile.NamedTemporaryFile() as output_stream: - subprocess.call(['/sbin/pfctl','-s', 'osfp'], stdout=output_stream, stderr=open(os.devnull, 'wb')) - output_stream.seek(0) - data = output_stream.read().strip() - if (data.count('\n') > 2): - for line in data.split('\n')[2:]: - result.append(line.replace('\t',' ').strip()) +if __name__ == '__main__': + result = [] + with tempfile.NamedTemporaryFile() as output_stream: + subprocess.call(['/sbin/pfctl', '-s', 'osfp'], stdout=output_stream, stderr=open(os.devnull, 'wb')) + output_stream.seek(0) + data = output_stream.read().strip() + if data.count('\n') > 2: + for line in data.split('\n')[2:]: + result.append(line.replace('\t', ' ').strip()) -# handle command line argument (type selection) -if len(sys.argv) > 1 and sys.argv[1] == 'json': - print(ujson.dumps(result)) -else: - # output plain - print ('------------------------- OS fingerprints -------------------------') - for ostype in result: - print (ostype) + # handle command line argument (type selection) + if len(sys.argv) > 1 and sys.argv[1] == 'json': + print(ujson.dumps(result)) + else: + # output plain + print ('------------------------- OS fingerprints -------------------------') + for ostype in result: + print (ostype)