made tools/findbits.py importable as a Python module

This commit is contained in:
dn337t@gmail.com 2012-06-20 15:07:30 +00:00
parent 9f69393035
commit 79c4de2a0f

View file

@ -75,7 +75,8 @@ def domatch(number,binary):
print ' Reverse: (%s)' % reversed, print ' Reverse: (%s)' % reversed,
search(inverted,reversed) search(inverted,reversed)
if(len(sys.argv) < 3): def main():
if(len(sys.argv) < 3):
print print
print '\t'+sys.argv[0] + ' - Search bitstream for a known number' print '\t'+sys.argv[0] + ' - Search bitstream for a known number'
print print
@ -92,14 +93,14 @@ if(len(sys.argv) < 3):
print print
os._exit(True) os._exit(True)
bases= { bases= {
2:'BINARY', 2:'BINARY',
8:'OCTAL', 8:'OCTAL',
10:'DECIMAL', 10:'DECIMAL',
16:'HEX', 16:'HEX',
} }
for base in 2,8,10,16: for base in 2,8,10,16:
try: try:
number= int(sys.argv[1],base) number= int(sys.argv[1],base)
print print
@ -111,3 +112,6 @@ for base in 2,8,10,16:
domatch(binstring(number),sys.argv[2]) domatch(binstring(number),sys.argv[2])
except: except:
continue continue
if __name__ == '__main__':
main()