add cli arg support

1 files changed, 6 insertions, 1 deletions

diff --git a/main.py b/main.py
index ce13568..dfc5581 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,12 @@
+import sys
 
 chars = "qwertyuiopasdfghjklzxcvbnmQWERTYUIOPASDFGHJKLZXCVBNM1234567890!@#$%^&*()[]{}\\|-=_+,./<>?;:\'\""
 
-passwd = input("Put in a password> ")
+global passwd
+try:
+    passwd = sys.argv[1]
+except:
+    passwd = input("Put in a password> ")
 
 guess_pass_index = [0 for i in range(len(passwd))]