#!/usr/bin/env python3 import argparse import json import subprocess import sys import os VERBOSE = True def log(*args): if VERBOSE: print(*args) def run_command(cmd): log(f"[CMD] {cmd}") try: subprocess.run(cmd, shell=True, check=True) except subprocess.CalledProcessError as e: log(f"Error executing command: {e}") def get_mode_string(output, mode_id): for mode in output.get('modes', []): if mode['id'] == mode_id: width = mode['size']['width'] height = mode['size']['height'] refresh_rate = round(mode['refreshRate']) return f"{width}x{height}@{refresh_rate}" return None def save_profile(profile_path): log(f"Saving current display profile to {profile_path}...") try: with open(profile_path, 'w') as f: subprocess.run(['kscreen-doctor', '--json'], stdout=f, check=True) log("Profile saved successfully.") except Exception as e: print(f"Error saving profile: {e}", file=sys.stderr) sys.exit(1) def apply_attribute(output_name, attribute, value, value_map=None): if value is None: log(f"Output {output_name} has no attribute {attribute}, skipping...") return # If it's a list or dict, something might be wrong or it's a complex structure we don't handle this way if isinstance(value, (list, dict)): log(f"Output {output_name} attribute {attribute} has complex value {value}, skipping...") return str_value = str(value).lower() if value_map: if str_value in value_map: value = value_map[str_value] else: log(f"Warning: Value {str_value} not found in map for {attribute}, using as is.") cmd = f"kscreen-doctor output.{output_name}.{attribute}.{value}" run_command(cmd) def load_profile(profile_path): log(f"Loading display profile from {profile_path}...") if not os.path.exists(profile_path): print(f"Profile file not found: {profile_path}", file=sys.stderr) sys.exit(1) try: with open(profile_path, 'r') as f: profile = json.load(f) except Exception as e: print(f"Error reading profile JSON: {e}", file=sys.stderr) sys.exit(1) outputs = profile.get('outputs', []) # Maps from JSON values to kscreen-doctor options bool_enable_map = {"true": "enable", "false": "disable"} bool_allow_map = {"true": "allow", "false": "disallow"} rgb_range_map = { "0": "automatic", "1": "full", "2": "limited", "automatic": "automatic", "full": "full", "limited": "limited" } rotation_map = {"1": "normal", "2": "left", "4": "inverted", "8": "right"} vrr_policy_map = { "0": "never", "1": "always", "2": "automatic", "never": "never", "always": "always", "automatic": "automatic" } # 1. Handle Enable/Disable (Enable first to avoid no-output state) log("Enabling enabled outputs...") for out in outputs: if out.get('enabled'): run_command(f"kscreen-doctor output.{out['name']}.enable") log("Disabling disabled outputs...") for out in outputs: if not out.get('enabled'): run_command(f"kscreen-doctor output.{out['name']}.disable") # 2. Apply other attributes for out in outputs: name = out['name'] log(f"Configuring output: {name}") # WCG apply_attribute(name, "wcg", out.get('wcg'), bool_enable_map) # SDR Brightness apply_attribute(name, "sdr-brightness", out.get('sdr-brightness')) # VRR Policy apply_attribute(name, "vrrpolicy", out.get('vrrPolicy'), vrr_policy_map) # RGB Range apply_attribute(name, "rgbrange", out.get('rgbRange'), rgb_range_map) # Overscan apply_attribute(name, "overscan", out.get('overscan')) # HDR apply_attribute(name, "hdr", out.get('hdr'), bool_enable_map) # Brightness (0.0-1.0 to 0-100) brightness = out.get('brightness') if brightness is not None: apply_attribute(name, "brightness", int(float(brightness) * 100)) # Max BPC max_bpc = out.get('maxBpc') if max_bpc == 0: max_bpc = "automatic" apply_attribute(name, "maxbpc", max_bpc) # DDC/CI apply_attribute(name, "ddcCi", out.get('ddcCiAllowed'), bool_allow_map) # Mirroring replication_source_id = out.get('replicationSource', 0) if replication_source_id != 0: # Find the name of the replication source source_name = "none" for other_out in outputs: if other_out.get('id') == replication_source_id: source_name = other_out['name'] break apply_attribute(name, "mirror", source_name) else: apply_attribute(name, "mirror", "none") # ICC Profile icc_path = out.get('iccProfilePath') if icc_path: apply_attribute(name, "iccProfilePath", icc_path) # Mode mode_id = out.get('currentModeId') if mode_id: mode_str = get_mode_string(out, mode_id) if mode_str: apply_attribute(name, "mode", mode_str) # Position pos = out.get('pos') if pos: apply_attribute(name, "position", f"{pos['x']},{pos['y']}") # Scale apply_attribute(name, "scale", out.get('scale')) # Rotation apply_attribute(name, "rotation", out.get('rotation'), rotation_map) # Priority and Primary priority = out.get('priority') if priority is not None: if priority == 1: run_command(f"kscreen-doctor output.{name}.primary") run_command(f"kscreen-doctor output.{name}.priority.{priority}") log("Display configuration restored.") def main(): parser = argparse.ArgumentParser(description="KDE Display Profile Manager") subparsers = parser.add_subparsers(dest="command", required=True) # Save command save_parser = subparsers.add_parser("save", help="Save current display configuration to a profile") save_parser.add_argument("profile", help="Path to the profile file (e.g., profiles/myprofile.json)") # Load command load_parser = subparsers.add_parser("load", help="Load a display configuration from a profile") load_parser.add_argument("profile", help="Path to the profile file") args = parser.parse_args() if args.command == "save": save_profile(args.profile) elif args.command == "load": load_profile(args.profile) if __name__ == "__main__": main()