autorandr.py

   1 #!/usr/bin/env python
   2 # encoding: utf-8
   3 #
   4 # autorandr.py
   5 # Copyright (c) 2015, Phillip Berndt
   6 #
   7 # Autorandr rewrite in Python
   8 #
   9 # This script aims to be fully compatible with the original autorandr.
  10 #
  11 # This program is free software: you can redistribute it and/or modify
  12 # it under the terms of the GNU General Public License as published by
  13 # the Free Software Foundation, either version 3 of the License, or
  14 # (at your option) any later version.
  15 #
  16 # This program is distributed in the hope that it will be useful,
  17 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  19 # GNU General Public License for more details.
  20 #
  21 # You should have received a copy of the GNU General Public License
  22 # along with this program. If not, see <http://www.gnu.org/licenses/>.
  23 #
  24
  25 from __future__ import print_function
  26 import copy
  27 import getopt
  28
  29 import binascii
  30 import hashlib
  31 import os
  32 import re
  33 import subprocess
  34 import sys
  35 from distutils.version import LooseVersion as Version
  36
  37 from itertools import chain
  38 from collections import OrderedDict
  39
  40 virtual_profiles = [
  41     # (name, description, callback)
  42     ("common", "Clone all connected outputs at the largest common resolution", None),
  43     ("horizontal", "Stack all connected outputs horizontally at their largest resolution", None),
  44     ("vertical", "Stack all connected outputs vertically at their largest resolution", None),
  45 ]
  46
  47 help_text = """
  48 Usage: autorandr [options]
  49
  50 -h, --help              get this small help
  51 -c, --change            reload current setup
  52 -s, --save <profile>    save your current setup to profile <profile>
  53 -l, --load <profile>    load profile <profile>
  54 -d, --default <profile> make profile <profile> the default profile
  55 --force                 force (re)loading of a profile
  56 --fingerprint           fingerprint your current hardware setup
  57 --config                dump your current xrandr setup
  58 --dry-run               don't change anything, only print the xrandr commands
  59
  60  To prevent a profile from being loaded, place a script call "block" in its
  61  directory. The script is evaluated before the screen setup is inspected, and
  62  in case of it returning a value of 0 the profile is skipped. This can be used
  63  to query the status of a docking station you are about to leave.
  64
  65  If no suitable profile can be identified, the current configuration is kept.
  66  To change this behaviour and switch to a fallback configuration, specify
  67  --default <profile>.
  68
  69  Another script called "postswitch" can be placed in the directory
  70  ~/.config/autorandr (or ~/.autorandr if you have an old installation) as well
  71  as in any profile directories: The scripts are executed after a mode switch
  72  has taken place and can notify window managers.
  73
  74  The following virtual configurations are available:
  75 """.strip()
  76
  77 class XrandrOutput(object):
  78     "Represents an XRandR output"
  79
  80     # This regular expression is used to parse an output in `xrandr --verbose'
  81     XRANDR_OUTPUT_REGEXP = """(?x)
  82         ^(?P<output>[^ ]+)\s+                                                           # Line starts with output name
  83         (?:                                                                             # Differentiate disconnected and connected in first line
  84             disconnected |
  85             unknown\ connection |
  86             (?P<connected>connected)\s+                                                 # If connected:
  87             (?:
  88             (?P<primary>primary\ )?                                                     # Might be primary screen
  89             (?P<width>[0-9]+)x(?P<height>[0-9]+)                                        # Resolution (might be overridden below!)
  90             \+(?P<x>[0-9]+)\+(?P<y>[0-9]+)\s+                                           # Position
  91             (?:\(0x[0-9a-fA-F]+\)\s+)?                                                  # XID
  92             (?P<rotate>(?:normal|left|right|inverted))\s+                               # Rotation
  93             (?:(?P<reflect>X\ and\ Y|X|Y)\ axis)?                                       # Reflection
  94             )?                                                                          # .. but everything of the above only if the screen is in use.
  95         ).*
  96         (?:\s*(?:                                                                       # Properties of the output
  97             Gamma: (?P<gamma>[0-9\.: ]+) |                                              # Gamma value
  98             Transform: (?P<transform>(?:[\-0-9\. ]+\s+){3}) |                           # Transformation matrix
  99             EDID: (?P<edid>\s*?(?:\\n\\t\\t[0-9a-f]+)+) |                               # EDID of the output
 100             (?![0-9])[^:\s][^:\n]+:.*(?:\s\\t[\\t ].+)*                                 # Other properties
 101         ))+
 102         \s*
 103         (?P<modes>(?:
 104             (?P<mode_width>[0-9]+)x(?P<mode_height>[0-9]+).+?\*current.+\s+
 105                 h:.+\s+v:.+clock\s+(?P<rate>[0-9\.]+)Hz\s* |                            # Interesting (current) resolution: Extract rate
 106             [0-9]+x[0-9]+.+\s+h:.+\s+v:.+\s*                                            # Other resolutions
 107         )*)
 108     """
 109
 110     XRANDR_OUTPUT_MODES_REGEXP = """(?x)
 111         (?P<width>[0-9]+)x(?P<height>[0-9]+)
 112         .*?(?P<preferred>\+preferred)?
 113         \s+h:.+
 114         \s+v:.+clock\s+(?P<rate>[0-9\.]+)Hz
 115     """
 116
 117     XRANDR_13_DEFAULTS = {
 118         "transform": "1,0,0,0,1,0,0,0,1",
 119     }
 120
 121     XRANDR_12_DEFAULTS = {
 122         "reflect": "normal",
 123         "rotate": "normal",
 124         "gamma": "1.0:1.0:1.0",
 125     }
 126
 127     XRANDR_DEFAULTS = dict(list(XRANDR_13_DEFAULTS.items()) + list(XRANDR_12_DEFAULTS.items()))
 128
 129     def __repr__(self):
 130         return "<%s%s %s>" % (self.output, (" %s..%s" % (self.edid[:5], self.edid[-5:])) if self.edid else "", " ".join(self.option_vector))
 131
 132     @property
 133     def options_with_defaults(self):
 134         "Return the options dictionary, augmented with the default values that weren't set"
 135         if "off" in self.options:
 136             return self.options
 137         options = {}
 138         if xrandr_version() >= Version("1.3"):
 139             options.update(self.XRANDR_13_DEFAULTS)
 140         if xrandr_version() >= Version("1.2"):
 141             options.update(self.XRANDR_12_DEFAULTS)
 142         options.update(self.options)
 143         return options
 144
 145     @property
 146     def option_vector(self):
 147         "Return the command line parameters for XRandR for this instance"
 148         return sum([["--%s" % option[0], option[1]] if option[1] else ["--%s" % option[0]] for option in chain((("output", self.output),), sorted(self.options_with_defaults.items()))], [])
 149
 150     @property
 151     def option_string(self):
 152         "Return the command line parameters in the configuration file format"
 153         return "\n".join([ " ".join(option) if option[1] else option[0] for option in chain((("output", self.output),), sorted(self.options.items()))])
 154
 155     @property
 156     def sort_key(self):
 157         "Return a key to sort the outputs for xrandr invocation"
 158         if not self.edid:
 159             return -1
 160         if "pos" in self.options:
 161             x, y = map(float, self.options["pos"].split("x"))
 162         else:
 163             x, y = 0, 0
 164         return x + 10000 * y
 165
 166     def __init__(self, output, edid, options):
 167         "Instanciate using output name, edid and a dictionary of XRandR command line parameters"
 168         self.output = output
 169         self.edid = edid
 170         self.options = options
 171         self.remove_default_option_values()
 172
 173     def remove_default_option_values(self):
 174         "Remove values from the options dictionary that are superflous"
 175         if "off" in self.options and len(self.options.keys()) > 1:
 176             self.options = { "off": None }
 177             return
 178         for option, default_value in self.XRANDR_DEFAULTS.items():
 179             if option in self.options and self.options[option] == default_value:
 180                 del self.options[option]
 181
 182     @classmethod
 183     def from_xrandr_output(cls, xrandr_output):
 184         """Instanciate an XrandrOutput from the output of `xrandr --verbose'
 185
 186         This method also returns a list of modes supported by the output.
 187         """
 188         try:
 189             xrandr_output = xrandr_output.replace("\r\n", "\n")
 190             match_object = re.search(XrandrOutput.XRANDR_OUTPUT_REGEXP, xrandr_output)
 191         except:
 192             raise RuntimeError("Parsing XRandR output failed, there is an error in the regular expression.")
 193         if not match_object:
 194             debug = debug_regexp(XrandrOutput.XRANDR_OUTPUT_REGEXP, xrandr_output)
 195             raise RuntimeError("Parsing XRandR output failed, the regular expression did not match: %s" % debug)
 196         remainder = xrandr_output[len(match_object.group(0)):]
 197         if remainder:
 198             raise RuntimeError(("Parsing XRandR output failed, %d bytes left unmatched after regular expression, "
 199                                 "starting at byte %d with ..'%s'.") % (len(remainder), len(len(match_object.group(0))), remainder[:10]))
 200
 201         match = match_object.groupdict()
 202
 203         modes = []
 204         if match["modes"]:
 205             modes = [ x.groupdict() for x in re.finditer(XrandrOutput.XRANDR_OUTPUT_MODES_REGEXP, match["modes"]) ]
 206             if not modes:
 207                 raise RuntimeError("Parsing XRandR output failed, couldn't find any display modes")
 208
 209         options = {}
 210         if not match["connected"]:
 211             options["off"] = None
 212             edid = None
 213         elif not match["width"]:
 214             options["off"] = None
 215             edid = "".join(match["edid"].strip().split())
 216         else:
 217             if match["mode_width"]:
 218                 options["mode"] = "%sx%s" % (match["mode_width"], match["mode_height"])
 219             else:
 220                 if match["rotate"] not in ("left", "right"):
 221                     options["mode"] = "%sx%s" % (match["width"], match["height"])
 222                 else:
 223                     options["mode"] = "%sx%s" % (match["height"], match["width"])
 224             options["rotate"] = match["rotate"]
 225             if match["primary"]:
 226                 options["primary"] = None
 227             if match["reflect"] == "X":
 228                 options["reflect"] = "x"
 229             elif match["reflect"] == "Y":
 230                 options["reflect"] = "y"
 231             elif match["reflect"] == "X and Y":
 232                 options["reflect"] = "xy"
 233             options["pos"] = "%sx%s" % (match["x"], match["y"])
 234             if match["transform"]:
 235                 transformation = ",".join(match["transform"].strip().split())
 236                 if transformation != "1.000000,0.000000,0.000000,0.000000,1.000000,0.000000,0.000000,0.000000,1.000000":
 237                     options["transform"] = transformation
 238                     if not match["mode_width"]:
 239                         # TODO We'd need to apply the reverse transformation here. Let's see if someone complains, I doubt that this
 240                         # special case is actually required.
 241                         print("Warning: Output %s has a transformation applied. Could not determine correct mode!", file=sys.stderr)
 242             if match["gamma"]:
 243                 gamma = match["gamma"].strip()
 244                 options["gamma"] = gamma
 245             if match["rate"]:
 246                 options["rate"] = match["rate"]
 247             edid = "".join(match["edid"].strip().split())
 248
 249         return XrandrOutput(match["output"], edid, options), modes
 250
 251     @classmethod
 252     def from_config_file(cls, edid_map, configuration):
 253         "Instanciate an XrandrOutput from the contents of a configuration file"
 254         options = {}
 255         for line in configuration.split("\n"):
 256             if line:
 257                 line = line.split(None, 1)
 258                 options[line[0]] = line[1] if len(line) > 1 else None
 259
 260         edid = None
 261
 262         if options["output"] in edid_map:
 263             edid = edid_map[options["output"]]
 264         else:
 265             # This fuzzy matching is for legacy autorandr that used sysfs output names
 266             fuzzy_edid_map = [ re.sub("(card[0-9]+|-)", "", x) for x in edid_map.keys() ]
 267             fuzzy_output = re.sub("(card[0-9]+|-)", "", options["output"])
 268             if fuzzy_output in fuzzy_edid_map:
 269                 edid = edid_map[list(edid_map.keys())[fuzzy_edid_map.index(fuzzy_output)]]
 270             elif "off" not in options:
 271                 raise RuntimeError("Failed to find an EDID for output `%s' in setup file, required as `%s' is not off in config file."
 272                                    % (options["output"], options["output"]))
 273         output = options["output"]
 274         del options["output"]
 275
 276         return XrandrOutput(output, edid, options)
 277
 278     def edid_equals(self, other):
 279         "Compare to another XrandrOutput's edid and on/off-state, taking legacy autorandr behaviour (md5sum'ing) into account"
 280         if self.edid and other.edid:
 281             if len(self.edid) == 32 and len(other.edid) != 32:
 282                 return hashlib.md5(binascii.unhexlify(other.edid)).hexdigest() == self.edid
 283             if len(self.edid) != 32 and len(other.edid) == 32:
 284                 return hashlib.md5(binascii.unhexlify(self.edid)).hexdigest() == other.edid
 285         return self.edid == other.edid
 286
 287     def __eq__(self, other):
 288         return self.edid_equals(other) and self.output == other.output and self.options == other.options
 289
 290 def xrandr_version():
 291     "Return the version of XRandR that this system uses"
 292     if getattr(xrandr_version, "version", False) is False:
 293         version_string = os.popen("xrandr -v").read()
 294         try:
 295             version = re.search("xrandr program version\s+([0-9\.]+)", version_string).group(1)
 296             xrandr_version.version = Version(version)
 297         except AttributeError:
 298             xrandr_version.version = Version("1.3.0")
 299
 300     return xrandr_version.version
 301
 302 def debug_regexp(pattern, string):
 303     "Use the partial matching functionality of the regex module to display debug info on a non-matching regular expression"
 304     try:
 305         import regex
 306         bounds = ( 0, len(string) )
 307         while bounds[0] != bounds[1]:
 308             half = int((bounds[0] + bounds[1]) / 2)
 309             if half == bounds[0]:
 310                 break
 311             bounds = (half, bounds[1]) if regex.search(pattern, string[:half], partial=True) else (bounds[0], half - 1)
 312         partial_length = bounds[0]
 313         return ("Regular expression matched until position "
 314               "%d, ..'%s', and did not match from '%s'.." % (partial_length, string[max(0, partial_length-20):partial_length],
 315                                                              string[partial_length:partial_length+10]))
 316     except ImportError:
 317         pass
 318     return "Debug information available if `regex' module is installed."
 319
 320 def parse_xrandr_output():
 321     "Parse the output of `xrandr --verbose' into a list of outputs"
 322     xrandr_output = os.popen("xrandr -q --verbose").read()
 323     if not xrandr_output:
 324         raise RuntimeError("Failed to run xrandr")
 325
 326     # We are not interested in screens
 327     xrandr_output = re.sub("(?m)^Screen [0-9].+", "", xrandr_output).strip()
 328
 329     # Split at output boundaries and instanciate an XrandrOutput per output
 330     split_xrandr_output = re.split("(?m)^([^ ]+ (?:(?:dis)?connected|unknown connection).*)$", xrandr_output)
 331     if len(split_xrandr_output) < 2:
 332         raise RuntimeError("No output boundaries found")
 333     outputs = OrderedDict()
 334     modes = OrderedDict()
 335     for i in range(1, len(split_xrandr_output), 2):
 336         output_name = split_xrandr_output[i].split()[0]
 337         output, output_modes = XrandrOutput.from_xrandr_output("".join(split_xrandr_output[i:i+2]))
 338         outputs[output_name] = output
 339         if output_modes:
 340             modes[output_name] = output_modes
 341
 342     return outputs, modes
 343
 344 def load_profiles(profile_path):
 345     "Load the stored profiles"
 346
 347     profiles = {}
 348     for profile in os.listdir(profile_path):
 349         config_name = os.path.join(profile_path, profile, "config")
 350         setup_name  = os.path.join(profile_path, profile, "setup")
 351         if not os.path.isfile(config_name) or not os.path.isfile(setup_name):
 352             continue
 353
 354         edids = dict([ x.strip().split() for x in open(setup_name).readlines() ])
 355
 356         config = {}
 357         buffer = []
 358         for line in chain(open(config_name).readlines(), ["output"]):
 359             if line[:6] == "output" and buffer:
 360                 config[buffer[0].strip().split()[-1]] = XrandrOutput.from_config_file(edids, "".join(buffer))
 361                 buffer = [ line ]
 362             else:
 363                 buffer.append(line)
 364
 365         for output_name in list(config.keys()):
 366             if config[output_name].edid is None:
 367                 del config[output_name]
 368
 369         profiles[profile] = { "config": config, "path": os.path.join(profile_path, profile), "config-mtime": os.stat(config_name).st_mtime }
 370
 371     return profiles
 372
 373 def find_profiles(current_config, profiles):
 374     "Find profiles matching the currently connected outputs"
 375     detected_profiles = []
 376     for profile_name, profile in profiles.items():
 377         config = profile["config"]
 378         matches = True
 379         for name, output in config.items():
 380             if not output.edid:
 381                 continue
 382             if name not in current_config or not output.edid_equals(current_config[name]):
 383                 matches = False
 384                 break
 385         if not matches or any(( name not in config.keys() for name in current_config.keys() if current_config[name].edid )):
 386             continue
 387         if matches:
 388             detected_profiles.append(profile_name)
 389     return detected_profiles
 390
 391 def profile_blocked(profile_path):
 392     "Check if a profile is blocked"
 393     script = os.path.join(profile_path, "blocked")
 394     if not os.access(script, os.X_OK | os.F_OK):
 395         return False
 396     return subprocess.call(script) == 0
 397
 398 def output_configuration(configuration, config):
 399     "Write a configuration file"
 400     outputs = sorted(configuration.keys(), key=lambda x: configuration[x].sort_key)
 401     for output in outputs:
 402         print(configuration[output].option_string, file=config)
 403
 404 def output_setup(configuration, setup):
 405     "Write a setup (fingerprint) file"
 406     outputs = sorted(configuration.keys())
 407     for output in outputs:
 408         if configuration[output].edid:
 409             print(output, configuration[output].edid, file=setup)
 410
 411 def save_configuration(profile_path, configuration):
 412     "Save a configuration into a profile"
 413     if not os.path.isdir(profile_path):
 414         os.makedirs(profile_path)
 415     with open(os.path.join(profile_path, "config"), "w") as config:
 416         output_configuration(configuration, config)
 417     with open(os.path.join(profile_path, "setup"), "w") as setup:
 418         output_setup(configuration, setup)
 419
 420 def update_mtime(filename):
 421     "Update a file's mtime"
 422     try:
 423         os.utime(filename, None)
 424         return True
 425     except:
 426         return False
 427
 428 def apply_configuration(configuration, dry_run=False):
 429     "Apply a configuration"
 430     outputs = sorted(configuration.keys(), key=lambda x: configuration[x].sort_key)
 431     if dry_run:
 432         base_argv = [ "echo", "xrandr" ]
 433     else:
 434         base_argv = [ "xrandr" ]
 435
 436     # Disable all unused outputs
 437     argv = base_argv[:]
 438     disable_argv = []
 439     for output in outputs:
 440         if not configuration[output].edid or "off" in configuration[output].options:
 441             disable_argv += configuration[output].option_vector
 442     if disable_argv:
 443         if subprocess.call(base_argv + disable_argv) != 0:
 444             # Disabling the outputs failed. Retry with the next command:
 445             # Sometimes disabling of outputs fails due to an invalid RRSetScreenSize.
 446             # This does not occur if simultaneously the primary screen is reset.
 447             pass
 448         else:
 449             disable_argv = []
 450
 451     # Enable remaining outputs in pairs of two
 452     remaining_outputs = [ x for x in outputs if configuration[x].edid ]
 453     for index in range(0, len(remaining_outputs), 2):
 454         argv = base_argv[:]
 455         if disable_argv:
 456             argv += disable_argv
 457             disable_argv = []
 458         argv += configuration[remaining_outputs[index]].option_vector + (configuration[remaining_outputs[index + 1]].option_vector if index < len(remaining_outputs) - 1 else [])
 459         if subprocess.call(argv) != 0:
 460             raise RuntimeError("Command failed: %s" % " ".join(argv))
 461
 462 def add_unused_outputs(source_configuration, target_configuration):
 463     "Add outputs that are missing in target to target, in 'off' state"
 464     for output_name, output in source_configuration.items():
 465         if output_name not in target_configuration:
 466             target_configuration[output_name] = XrandrOutput(output_name, output.edid, { "off": None })
 467
 468 def generate_virtual_profile(configuration, modes, profile_name):
 469     "Generate one of the virtual profiles"
 470     configuration = copy.deepcopy(configuration)
 471     if profile_name == "common":
 472         common_resolution = [ set(( ( mode["width"], mode["height"] ) for mode in output )) for output in modes.values() ]
 473         common_resolution = reduce(lambda a, b: a & b, common_resolution[1:], common_resolution[0])
 474         common_resolution = sorted(common_resolution, key=lambda a: int(a[0])*int(a[1]))
 475         if common_resolution:
 476             for output in configuration:
 477                 configuration[output].options = {}
 478                 if output in modes:
 479                     configuration[output].options["mode"] = "%sx%s" % common_resolution[-1]
 480                     configuration[output].options["pos"] = "0x0"
 481                 else:
 482                     configuration[output].options["off"] = None
 483     elif profile_name in ("horizontal", "vertical"):
 484         shift = 0
 485         if profile_name == "horizontal":
 486             shift_index = "width"
 487             pos_specifier = "%sx0"
 488         else:
 489             shift_index = "height"
 490             pos_specifier = "0x%s"
 491
 492         for output in configuration:
 493             configuration[output].options = {}
 494             if output in modes:
 495                 mode = sorted(modes[output], key=lambda a: int(a["width"])*int(a["height"]) + (10**6 if a["preferred"] else 0))[-1]
 496                 configuration[output].options["mode"] = "%sx%s" % (mode["width"], mode["height"])
 497                 configuration[output].options["rate"] = mode["rate"]
 498                 configuration[output].options["pos"] = pos_specifier % shift
 499                 shift += int(mode[shift_index])
 500             else:
 501                 configuration[output].options["off"] = None
 502     return configuration
 503
 504 def exit_help():
 505     "Print help and exit"
 506     print(help_text)
 507     for profile in virtual_profiles:
 508         print("  %-10s %s" % profile[:2])
 509     sys.exit(0)
 510
 511 def exec_scripts(profile_path, script_name):
 512     "Run userscripts"
 513     for script in (os.path.join(profile_path, script_name), os.path.join(os.path.dirname(profile_path), script_name)):
 514         if os.access(script, os.X_OK | os.F_OK):
 515             subprocess.call(script)
 516
 517 def main(argv):
 518     try:
 519        options = dict(getopt.getopt(argv[1:], "s:l:d:cfh", [ "dry-run", "change", "default=", "save=", "load=", "force", "fingerprint", "config", "help" ])[0])
 520     except getopt.GetoptError as e:
 521         print(str(e))
 522         options = { "--help": True }
 523
 524     profiles = {}
 525     try:
 526         # Load profiles from each XDG config directory
 527         for directory in os.environ.get("XDG_CONFIG_DIRS", "").split(":"):
 528             system_profile_path = os.path.join(directory, "autorandr")
 529             if os.path.isdir(system_profile_path):
 530                 profiles.update(load_profiles(system_profile_path))
 531         # For the user's profiles, prefer the legacy ~/.autorandr if it already exists
 532         # profile_path is also used later on to store configurations
 533         profile_path = os.path.expanduser("~/.autorandr")
 534         if not os.path.isdir(profile_path):
 535             # Elsewise, follow the XDG specification
 536             profile_path = os.path.join(os.environ.get("XDG_CONFIG_HOME", os.path.expanduser("~/.config")), "autorandr")
 537         if os.path.isdir(profile_path):
 538             profiles.update(load_profiles(profile_path))
 539         # Sort by descending mtime
 540         profiles = OrderedDict(sorted(profiles.items(), key=lambda x: -x[1]["config-mtime"]))
 541     except Exception as e:
 542         print("Failed to load profiles:\n%s" % str(e), file=sys.stderr)
 543         sys.exit(1)
 544
 545     try:
 546         config, modes = parse_xrandr_output()
 547     except Exception as e:
 548         print("Failed to parse current configuration from XRandR:\n%s" % str(e), file=sys.stderr)
 549         sys.exit(1)
 550
 551     if "--fingerprint" in options:
 552         output_setup(config, sys.stdout)
 553         sys.exit(0)
 554
 555     if "--config" in options:
 556         output_configuration(config, sys.stdout)
 557         sys.exit(0)
 558
 559     if "-s" in options:
 560         options["--save"] = options["-s"]
 561     if "--save" in options:
 562         if options["--save"] in ( x[0] for x in virtual_profiles ):
 563             print("Cannot save current configuration as profile '%s':\nThis configuration name is a reserved virtual configuration." % options["--save"])
 564             sys.exit(1)
 565         try:
 566             save_configuration(os.path.join(profile_path, options["--save"]), config)
 567         except Exception as e:
 568             print("Failed to save current configuration as profile '%s':\n%s" % (options["--save"], str(e)), file=sys.stderr)
 569             sys.exit(1)
 570         print("Saved current configuration as profile '%s'" % options["--save"])
 571         sys.exit(0)
 572
 573     if "-h" in options or "--help" in options:
 574         exit_help()
 575
 576     detected_profiles = find_profiles(config, profiles)
 577     load_profile = False
 578
 579     if "-l" in options:
 580         options["--load"] = options["-l"]
 581     if "--load" in options:
 582         load_profile = options["--load"]
 583     else:
 584         for profile_name in profiles.keys():
 585             if profile_blocked(os.path.join(profile_path, profile_name)):
 586                 print("%s (blocked)" % profile_name, file=sys.stderr)
 587                 continue
 588             if profile_name in detected_profiles:
 589                 print("%s (detected)" % profile_name, file=sys.stderr)
 590                 if ("-c" in options or "--change" in options) and not load_profile:
 591                     load_profile = profile_name
 592             else:
 593                 print(profile_name, file=sys.stderr)
 594
 595     if "-d" in options:
 596         options["--default"] = options["-d"]
 597     if not load_profile and "--default" in options:
 598         load_profile = options["--default"]
 599
 600     if load_profile:
 601         if load_profile in ( x[0] for x in virtual_profiles ):
 602             load_config = generate_virtual_profile(config, modes, load_profile)
 603             scripts_path = os.path.join(profile_path, load_profile)
 604         else:
 605             try:
 606                 profile = profiles[load_profile]
 607                 load_config = profile["config"]
 608                 scripts_path = profile["path"]
 609             except KeyError:
 610                 print("Failed to load profile '%s':\nProfile not found" % load_profile, file=sys.stderr)
 611                 sys.exit(1)
 612             if load_profile in detected_profiles and detected_profiles[0] != load_profile:
 613                 update_mtime(os.path.join(scripts_path, "config"))
 614         add_unused_outputs(config, load_config)
 615         if load_config == dict(config) and not "-f" in options and not "--force" in options:
 616             print("Config already loaded", file=sys.stderr)
 617             sys.exit(0)
 618
 619         try:
 620             if "--dry-run" in options:
 621                 apply_configuration(load_config, True)
 622             else:
 623                 exec_scripts(scripts_path, "preswitch")
 624                 apply_configuration(load_config, False)
 625                 exec_scripts(scripts_path, "postswitch")
 626         except Exception as e:
 627             print("Failed to apply profile '%s':\n%s" % (load_profile, str(e)), file=sys.stderr)
 628             sys.exit(1)
 629
 630     sys.exit(0)
 631
 632 if __name__ == '__main__':
 633     try:
 634         main(sys.argv)
 635     except Exception as e:
 636         print("General failure. Please report this as a bug:\n%s" % (str(e),), file=sys.stderr)
 637         sys.exit(1)