Merge pull request #28 from dkgitdev/patch-1

better name handling
2018-04-22 23:30:55 +02:00
parent 5818f369da 98593da476
commit 89de1ba5ef
1 changed files with 2 additions and 1 deletions
--- a/txttobin.py
+++ b/txttobin.py
@ -1,6 +1,7 @@
 import sys
 data = file(sys.argv[1], 'rb').readlines()
-uid = sys.argv[1].replace('.txt','')
+if sys.argv[1].endswith('.txt'):
+    uid = sys.argv[1][:-4]

 def parity(x):
    return ("{0:08b}".format(x).count('1') & 1)