better name handling

aczid · Apr 22, 2018 · 98593da · 98593da
1 parent 5818f36
commit 98593da
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)