refactor directories

refactor

..

..

Generator/user.py

@@ -0,0 +1,41 @@
+import secrets
+import os
+import winreg
+
+
+path = winreg.HKEY_CURRENT_USER
+
+def saveReg(k,v):
+    try:
+        key = winreg.OpenKeyEx(path, r"SOFTWARE\\")
+        newKey = winreg.CreateKey(key,"RotorOps")
+        winreg.SetValueEx(newKey, k, 0, winreg.REG_SZ, str(v))
+        if newKey:
+            winreg.CloseKey(newKey)
+        return True
+    except Exception as e:
+        print(e)
+    return False
+
+
+def readReg(k):
+    try:
+        key = winreg.OpenKeyEx(path, r"SOFTWARE\\RotorOps\\")
+        value = winreg.QueryValueEx(key,k)
+        if key:
+            winreg.CloseKey(key)
+        return value[0]
+    except Exception as e:
+        print(e)
+    return None
+
+def createUserKey():
+    userid = readReg('User')
+    if not userid or userid == 'None':
+        print("Unable to find userid in registry.")
+        userid = secrets.token_urlsafe(10)
+        if saveReg('User', userid):
+            print("Saved userid to registry")
+    return userid
+
+