An try to fix autorun in utils.py. In case, when system has problem…

…s with registry permissions
denipolis · Oct 11, 2023 · 4ae90c0 · 4ae90c0
1 parent cb06c17
commit 4ae90c0
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/src/utils.py b/src/utils.py
@@ -32,20 +32,25 @@ def openURL(link: str) -> None:
 def enableAutoRun() -> None:
     currentAppPath = sys.argv[0]
 
-    key = reg.OpenKey(reg.HKEY_CURRENT_USER, 'Software\Microsoft\Windows\CurrentVersion\Run', 0, reg.KEY_SET_VALUE)
-    reg.SetValueEx(key, "ArciBinder", 0, reg.REG_SZ, currentAppPath)
-    key.Close()
+    try:
+        key = reg.OpenKey(reg.HKEY_CURRENT_USER, 'Software\Microsoft\Windows\CurrentVersion\Run', 0, reg.KEY_SET_VALUE)
+        reg.SetValueEx(key, "ArciBinder", 0, reg.REG_SZ, currentAppPath)
+        key.Close()
+    except Exception as e:
+        print(e.message)
 
 def disableAutoRun() -> None:
-    key = reg.OpenKey(reg.HKEY_CURRENT_USER, 'Software\Microsoft\Windows\CurrentVersion\Run', 0, reg.KEY_WRITE)
-    reg.DeleteValue(key, "ArciBinder")
-    key.Close()
+    try:
+        key = reg.OpenKey(reg.HKEY_CURRENT_USER, 'Software\Microsoft\Windows\CurrentVersion\Run', 0, reg.KEY_WRITE)
+        reg.DeleteValue(key, "ArciBinder")
+        key.Close()
+    except Exception as e:
+        print(e.mesasge)
 
 def isAutoRun() -> bool:
-    key = reg.OpenKey(reg.HKEY_CURRENT_USER, 'Software\Microsoft\Windows\CurrentVersion\Run', 0, reg.KEY_QUERY_VALUE)
-
     try:
+        key = reg.OpenKey(reg.HKEY_CURRENT_USER, 'Software\Microsoft\Windows\CurrentVersion\Run', 0, reg.KEY_QUERY_VALUE)
         reg.QueryValueEx(key, "ArciBinder")
         return True
-    except:
+    except Exception as e:
         return False