From 98593da4767826ad84989719e2cb157126795dc3 Mon Sep 17 00:00:00 2001
From: dkgitdev <36101416+dkgitdev@users.noreply.github.com>
Date: Mon, 23 Apr 2018 01:16:10 +0700
Subject: [PATCH] better name handling

---
 txttobin.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/txttobin.py b/txttobin.py
index 670e9ed..ac53400 100644
--- 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)