added package

Author: wrecks-code

SimpleMonitorControlTray/__init__.py

@@ -0,0 +1,64 @@
+import configparser
+import os
+
+import SimpleMonitorControlTray.monitorHandler as mH
+import SimpleMonitorControlTray.notificationHandler as nH
+import SimpleMonitorControlTray.trayHandler as tH
+
+currentPath = os.getcwd()
+config_file_path = "config.ini"
+assets_folder = "assets"
+asset_iconEnabled = "assets\iconEnabled.png"
+asset_iconDisabled = "assets\iconDisabled.png"
+
+fileNotFound = " file not found. Exiting."
+
+MULTIMONITORTOOL_PATH, CSV_FILE_PATH, MM_CONFIG_FILE_PATH, MONITOR_NAME = (
+    None,
+    None,
+    None,
+    None,
+)
+
+
+def check_for_missing_files():
+
+    if not os.path.exists(MULTIMONITORTOOL_PATH):
+        nH.sendError(MULTIMONITORTOOL_PATH + fileNotFound)
+        tH.quitItemClicked()
+    if not os.path.exists(os.path.join(currentPath, asset_iconEnabled)):
+        nH.sendError(asset_iconDisabled + fileNotFound)
+        tH.quitItemClicked()
+
+    if not os.path.exists(CSV_FILE_PATH):
+        mH.saveMultiMonitorToolConfig()
+
+    multiMonitorToolOutputPath = os.path.join(currentPath, "MultiMonitorTool")
+
+    if not os.path.exists(multiMonitorToolOutputPath):
+        os.makedirs(multiMonitorToolOutputPath)
+
+    nH.sendNotification(
+        "If you do not have all Monitors enabled and configured as you like right now, please do so and then right click the tray icon and save the configuration",
+        20,
+    )
+
+
+def read_config():
+
+    global 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()
+
+    config = configparser.ConfigParser()
+
+    config.read(config_file_path, encoding="utf-8")
+
+    MULTIMONITORTOOL_PATH = config.get("SETTINGS", "multimonitorpath")
+    MONITOR_NAME = config.get("SETTINGS", "monitor_name")
+    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()

SimpleMonitorControlTray/configHandler.py

@@ -0,0 +1,62 @@
+import csv
+import subprocess
+
+import SimpleMonitorControlTray.configHandler as ch
+
+
+def saveMultiMonitorToolConfig():
+    subprocess.run(
+        [
+            ch.MULTIMONITORTOOL_PATH,
+            "/SaveConfig",
+            ch.MM_CONFIG_FILE_PATH,
+        ]
+    )
+
+
+def updateMonitorsCSV():
+    # Run MultiMonitorTool.exe with the /scomma option to save the monitors list into a CSV file
+    subprocess.run(
+        [
+            ch.MULTIMONITORTOOL_PATH,
+            "/scomma",
+            ch.CSV_FILE_PATH,
+        ]
+    )
+
+
+def enableMonitor():
+    subprocess.run(
+        [
+            ch.MULTIMONITORTOOL_PATH,
+            "/LoadConfig",
+            ch.MM_CONFIG_FILE_PATH,
+        ]
+    )
+
+
+def disableMonitor():
+    # Run MultiMonitorTool.exe with the /disable option to disable the third monitor
+    subprocess.run(
+        [
+            ch.MULTIMONITORTOOL_PATH,
+            "/disable",
+            ch.MONITOR_NAME[-1],
+        ]
+    )
+
+
+def isMonitorEnabled():
+    # Check if the third monitor is listed in the CSV file
+    with open(ch.CSV_FILE_PATH, "r") as file:
+        reader = csv.DictReader(file)
+
+        for row in reader:
+            try:
+                monitor_name = row["Name"]
+                monitor_active = row["Active"]
+                if monitor_name == ch.MONITOR_NAME and monitor_active.upper() == "YES":
+                    return True
+            except KeyError:
+                pass  # Ignore rows with missing keys
+    return False  # No match found

SimpleMonitorControlTray/monitorHandler.py

@@ -0,0 +1,13 @@
+import threading
+
+from win10toast import ToastNotifier
+
+
+def sendError(text):
+    toaster = ToastNotifier()
+    toaster.show_toast("Error", text, duration=7)
+
+
+def sendNotification(text, duration):
+    # toaster.show_toast("It's your first startup!", text, duration=duration)
+    pass

SimpleMonitorControlTray/notificationHandler.py

@@ -0,0 +1,66 @@
+import os
+
+import pystray
+from PIL import Image
+from pystray import MenuItem as item
+
+import SimpleMonitorControlTray.configHandler as cH
+import SimpleMonitorControlTray.monitorHandler as mH
+
+script_dir = os.getcwd()
+
+imageIconEnabled = Image.open(os.path.join(script_dir, "assets\iconEnabled.png"))
+imageIconDisabled = Image.open(os.path.join(script_dir, "assets\iconDisabled.png"))
+
+title = "SimpleMonitorControlTray"
+
+icon = None
+
+
+def saveMultiMonitorToolConfigClicked():
+    mH.saveMultiMonitorToolConfig()
+
+
+def iconTrayClicked():
+    mH.updateMonitorsCSV()
+    if mH.isMonitorEnabled():
+        mH.disableMonitor()
+        icon.icon = imageIconDisabled
+    else:
+        mH.enableMonitor()
+        icon.icon = imageIconEnabled
+
+
+def quitItemClicked():
+    icon.stop()
+
+
+def openConfigClicked():
+    os.startfile(os.path.join(script_dir, cH.config_file_path))
+
+
+def initTray():
+    global icon
+    menu = (
+        item(title, iconTrayClicked, default=True, visible=False),
+        item(
+            "Save current monitor layout (used when enabling)",
+            saveMultiMonitorToolConfigClicked,
+        ),
+        item("Open Config.ini", openConfigClicked),
+        item("Quit", quitItemClicked),
+    )
+
+    firstImageIcon = None
+
+    mH.updateMonitorsCSV()
+
+    if mH.isMonitorEnabled():
+        firstImageIcon = imageIconEnabled
+        if not os.path.exists(cH.MM_CONFIG_FILE_PATH):
+            mH.saveMultiMonitorToolConfig()
+    else:
+        firstImageIcon = imageIconDisabled
+
+    icon = pystray.Icon(title, firstImageIcon, title, menu)
+    icon.run()