update cli prompt

usc-isi-i2 · Nov 21, 2018 · 727114c · 727114c
1 parent 57ef371
commit 727114c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/rltk/__main__.py b/rltk/__main__.py
@@ -15,14 +15,17 @@ def help_info():
     if len(sys.argv) <= 1:
         print('No command\n')
         help_info()
+        sys.exit()
 
     cmd = sys.argv[1]
     sub_cmd = sys.argv[2:] if len(sys.argv) >= 3 else []
     sys.argv.pop(1)
 
     if cmd in ('help', '--help', 'h', '-h'):
         help_info()
+        sys.exit()
 
+    sys.argv[0] = cmd  # replace prog name
     temp_path = os.path.join(tempfile.gettempdir(), 'rltk', 'remote')
     if not os.path.exists(temp_path):
         os.makedirs(temp_path, exist_ok=True)