added autostart option

Author: wrecks-code

SimpleMonitorControlTray/configHandler.py

@@ -26,10 +26,10 @@ def check_for_missing_files():
 
     if not os.path.exists(MULTIMONITORTOOL_PATH):
         nH.sendError(MULTIMONITORTOOL_PATH + fileNotFound)
-        tH.quitItemClicked()
+        tH.exitItemClicked()
     if not os.path.exists(os.path.join(currentPath, asset_iconEnabled)):
         nH.sendError(asset_iconDisabled + fileNotFound)
-        tH.quitItemClicked()
+        tH.exitItemClicked()
 
     if not os.path.exists(CSV_FILE_PATH):
         mH.saveMultiMonitorToolConfig()
@@ -47,19 +47,35 @@ def check_for_missing_files():
 
 def read_config():
 
-    global MULTIMONITORTOOL_PATH, CSV_FILE_PATH, MM_CONFIG_FILE_PATH, MONITOR_NAME
+    global AUTOSTART, MULTIMONITORTOOL_PATH, CSV_FILE_PATH, MM_CONFIG_FILE_PATH, MONITOR_NAME
 
     if not os.path.exists(config_file_path):
         nH.sendError(config_file_path + fileNotFound)
-        tH.quitItemClicked()
+        tH.exitItemClicked()
 
     config = configparser.ConfigParser()
 
     config.read(config_file_path, encoding="utf-8")
 
     MULTIMONITORTOOL_PATH = config.get("SETTINGS", "multimonitorpath")
     MONITOR_NAME = config.get("SETTINGS", "monitor_name")
+    AUTOSTART = config.get("SETTINGS", "autostart")
     CSV_FILE_PATH = config.get("DEV", "mm_csv_export_path")
     MM_CONFIG_FILE_PATH = config.get("DEV", "mm_config_file_path")
 
     check_for_missing_files()
+
+
+def set_config_value(category, key, value):
+    config = configparser.ConfigParser()
+    config.read(config_file_path, encoding="utf-8")
+    config[category][key] = value
+    with open(config_file_path, "w") as configfile:
+        config.write(configfile)
+    print("Setting config value: " + key + " to " + value)
+
+
+def get_config_value(category, key):
+    config = configparser.ConfigParser()
+    config.read(config_file_path, encoding="utf-8")
+    return config.get(category, key)

SimpleMonitorControlTray/registryHandler.py

@@ -14,6 +14,9 @@ def add_to_autostart():
 
 
 def remove_from_autostart():
-    key = reg.OpenKey(reg.HKEY_CURRENT_USER, registry_key, 0, reg.KEY_ALL_ACCESS)
-    reg.DeleteValue(key, keyName)
-    reg.CloseKey(key)
+    try:
+        key = reg.OpenKey(reg.HKEY_CURRENT_USER, registry_key, 0, reg.KEY_ALL_ACCESS)
+        reg.DeleteValue(key, keyName)
+        reg.CloseKey(key)
+    except:
+        pass

SimpleMonitorControlTray/trayHandler.py

@@ -6,6 +6,7 @@
 
 import SimpleMonitorControlTray.configHandler as cH
 import SimpleMonitorControlTray.monitorHandler as mH
+import SimpleMonitorControlTray.registryHandler as rH
 
 script_dir = os.getcwd()
 
@@ -15,6 +16,7 @@
 title = "SimpleMonitorControlTray"
 
 icon = None
+itemTitle = ""
 
 
 def saveMultiMonitorToolConfigClicked():
@@ -31,42 +33,58 @@ def iconTrayClicked():
         icon.icon = imageIconEnabled
 
 
-def quitItemClicked():
+def exitItemClicked():
     icon.stop()
 
 
 def openConfigClicked():
     os.startfile(os.path.join(script_dir, cH.config_file_path))
 
 
-checked = False
+def toggleAutostartInConfig():
+    global itemTitle
+    if cH.AUTOSTART == "False":
+        itemTitle = "Disable Autostart"
+        rH.add_to_autostart()
+        cH.set_config_value("SETTINGS", "autostart", "True")
+        cH.AUTOSTART = "True"
+    else:
+        itemTitle = "Enable Autostart"
+        rH.remove_from_autostart()
+        cH.set_config_value("SETTINGS", "autostart", "False")
+        cH.AUTOSTART = "False"
 
 
 def toggleAutostart(icon):
-    # Recreate the menu with the updated title for "Autostart"
+    toggleAutostartInConfig()
     new_menu = (
-        item("New Title", toggleAutostart),
+        item(itemTitle, toggleAutostart),
         item(
             "Save current monitor layout (used when enabling)",
             saveMultiMonitorToolConfigClicked,
         ),
         item("Open Config.ini", openConfigClicked),
-        item("Quit", quitItemClicked),
+        item("Exit", exitItemClicked),
     )
     icon.menu = new_menu
 
 
 def initTray():
     global icon
+    if cH.AUTOSTART == "True":
+        itemTitle = "Disable Autostart"
+    else:
+        itemTitle = "Enable Autostart"
+
     menu = (
         item(title, iconTrayClicked, default=True, visible=False),
-        item("Autostart", toggleAutostart),
+        item(itemTitle, toggleAutostart),
         item(
             "Save current monitor layout (used when enabling)",
             saveMultiMonitorToolConfigClicked,
         ),
         item("Open Config.ini", openConfigClicked),
-        item("Quit", quitItemClicked),
+        item("Quit", exitItemClicked),
     )
 
     firstImageIcon = None

config.ini

@@ -1,7 +1,9 @@
 [SETTINGS]
 monitor_name = \\.\DISPLAY3
 multimonitorpath = C:\\App_Install\\multimonitortool-x64\\MultiMonitorTool.exe
-autostart = False
+autostart = True
+
 [DEV]
 mm_csv_export_path = MultiMonitorTool\multiMonitorToolOutput.csv
 mm_config_file_path = MultiMonitorTool\MultiMonitorToolConfig
+