added menu & added argument that cna pass through to write.py or read.py

1 files changed, 16 insertions, 5 deletions

diff --git a/main.py b/main.py
index a04d1d5..b9cce95 100644
--- a/main.py
+++ b/main.py
@@ -1,8 +1,19 @@
+#!/usr/bin/env python
+
 from sys import argv, executable
 from os import execv
-if argv[1] == 'write':
-    execv(executable, ['python3', 'write.py'])
-elif argv[1] =='read':
-    execv(executable, ['python3','read.py'])
+if len(argv) > 1:
+    if argv[1] == 'write':
+        print(' '.join(argv[2:]))
+        execv(executable, ['python3', 'write.py'] + argv[2:])
+    elif argv[1] =='read':
+        execv(executable, ['python3','read.py'])
+    else:
+        raise Exception("Unknown command\n\tAvailable commands: write, read")
 else:
-    raise Exception("Unknown command")
\ No newline at end of file
+    from simple_term_menu import TerminalMenu
+    choice = TerminalMenu(['Write', 'Read']).show()
+    if choice == 0:
+        execv(executable, ['python3', 'write.py'])
+    elif choice == 1:
+        execv(executable, ['python3','read.py'])
\ No newline at end of file