Optimizing user commands to prevent conflicts (v1.0.4)

keyvault.py

@@ -187,8 +187,8 @@ print("\nkeyvault initialized.")
 print("keyvault is ready to use! Type 'help' for a list of commands.\n")
 
 while True:
-    command = input("> ")[0].lower()
+    command = input("> ")[0:2].lower()
-    if command == 'h':
+    if command[0] == 'h':
         print("""
 Usage:
 
@@ -204,22 +204,22 @@ version - print current version
 quit - quit the program
 
         """)
-    elif command == 's' or command == 'f':
+    elif command[0] == 's' or command[0] == 'f':
         find()
-    elif command == 'l':
+    elif command[0] == 'l':
         ls()
-    elif command == 'a':
+    elif command[0] == 'a':
         add()
-    elif command == 'r':
+    elif command[0] == 'r':
         rm()
-    elif command == 'e':
+    elif command == 'ed':
         edit()
-    elif command == 'v':
+    elif command[0] == 'v':
-        print("keyvault: v1.0.3")
+        print("keyvault: v1.0.4")
-    elif command == 'g':
+    elif command[0] == 'g':
         gen()
-    elif command == 's':
+    elif command[0] == 's':
         strength()
-    elif command == 'q':
+    elif command[0] == 'q' or command == 'ex':
         print("Thank for you using keyvault!")
         exit()