Add CLI args

z1pti3 · Mar 6, 2021 · e6d8792 · e6d8792
1 parent 8deda03
commit e6d8792
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/jimi_cli.py b/jimi_cli.py
@@ -1,2 +1,17 @@
-from screens import mainScreen
-screen = mainScreen.mainScreen()
+import sys
+
+if len(sys.argv) > 1:
+    if sys.argv[1] == "reset":
+        if sys.argv[2] == "root":
+            from core import auth
+            from system import install
+            rootUser = auth._user().getAsClass(query={ "username" : "root" })
+            if len(rootUser) == 1:
+                rootUser = rootUser[0]
+                rootPass = install.randomString(30)
+                rootUser.setAttribute("passwordHash",rootPass)
+                rootUser.update(["passwordHash"])
+                print("Password={0}".format(rootPass))
+else:
+    from screens import mainScreen
+    screen = mainScreen.mainScreen()