- Added automatic monitor.conf setup to hdr-helper script

- Setup script now copies config file and backups up old one
2025-10-08 23:19:04 +01:00
parent 6e5d3906aa
commit 4e16d193f7
2 changed files with 29 additions and 9 deletions
@@ -0,0 +1,95 @@
+import re
+import subprocess
+import os
+import shutil
+
+class MonitorSetup():
+    def __init__(self):
+        self._stg_conf_file = "monitor.conf"
+        self._conf_file_path = os.path.join(os.path.expanduser('~'),
+                                            ".config/hdr-helper",self._stg_conf_file)
+        ################################
+        self.default_mode = None
+        self.alt_rate = None
+        self.alt_res_and_rate = None
+        self.output_name = None
+
+    def extract_display_info(self):
+        # Run process, capture output as a string
+        proc = subprocess.run(
+            ["kscreen-doctor -o"],
+            shell=True,
+            capture_output=True,
+            text=True
+        )
+        output = proc.stdout
+
+        # Remove ANSI escape codes for clean regex matching
+        output_clean = re.sub(r'\x1b\[[0-9;]*m', '', output)
+
+        # Find all resolutions with flags
+        resolutions = re.findall(r'\d+x\d+@\d+\S', output_clean)
+
+        # Identify default display mode and alternatives
+        for i in resolutions:
+            if i.endswith('*'):
+                self.default_mode = i
+            if self.default_mode and i != self.default_mode:
+                if i.startswith(self.default_mode.split('@')[0]):
+                    self.alt_rate = i
+                    break
+                else:
+                    self.alt_res_and_rate = i
+                    
+
+        # Remove asterisk from default
+        if self.default_mode:
+            self.default_mode = self.default_mode.strip('*')
+
+        search_output = re.search(r'Output:\s*\d+\s+(\S+)', output_clean)
+
+        # Extract and print output name
+        if search_output:
+            self.output_name = search_output.group(1)
+            
+    def print_display_info(self):
+        print(f"Output Name: {self.output_name}")
+        print(f"Default Mode: {self.default_mode}")
+        print(f"Alternative Rate: {self.alt_rate}")
+        print(f"Alternative Resolution and Rate: {self.alt_res_and_rate}")
+
+    def edit_config_file(self):
+        if os.path.isfile(self._stg_conf_file):
+            with open(self._stg_conf_file, "r") as file:
+                lines = file.readlines()
+
+            with open(self._stg_conf_file, "w") as file:            
+                for line in lines:
+                    if "MONITOR=" in line:
+                        line = line.replace("MONITOR=\n", f"MONITOR={self.output_name}\n")
+                    elif "NATIVE_RES=" in line:
+                        line = line.replace("NATIVE_RES=\n", f"NATIVE_RES={self.default_mode}\n")
+
+                    # Prefer alt_rate if available, otherwise alt_res_and_rate
+                    if "TEMP_RES=" in line:
+                        if self.alt_rate:
+                            line = line.replace("TEMP_RES=\n", f"TEMP_RES={self.alt_rate}\n")
+                        elif self.alt_res_and_rate:
+                            line = line.replace("TEMP_RES=\n", f"TEMP_RES={self.alt_res_and_rate}\n")
+                        
+                    file.write(line)
+            
+    def copy_config_file(self):
+        # Backup existing config file if it exists
+        if os.path.isfile(self._conf_file_path):
+            shutil.move(self._conf_file_path, self._conf_file_path + ".backup" )
+        
+        # Copy staging config file to target location
+        if os.path.isfile(self._stg_conf_file):
+            shutil.copy(self._stg_conf_file, self._conf_file_path)
+        
+
+setup = MonitorSetup()
+setup.extract_display_info()
+setup.edit_config_file()
+setup.copy_config_file()