-# Check what to do
-if usedExternalConnector is not None: # there's an external screen connected, we need to ask what to do
- internalResolutions = connectors[internalConnector]
- externalResolutions = connectors[usedExternalConnector]
- extPosition = PositionSelection(usedExternalConnector, map(res2user, internalResolutions), map(res2user, externalResolutions))
- extPosition.exec_()
- if not extPosition.result(): sys.exit(1) # the user canceled
- extResolution = res2xrandr(externalResolutions[extPosition.extResolutions.currentIndex()])
- intResolution = res2xrandr(internalResolutions[extPosition.intResolutions.currentIndex()])
- # build command-line
- args[usedExternalConnector] = ["--mode", extResolution] # set external screen to desired resolution
- if extPosition.extOnly.isChecked():
- args[usedExternalConnector] += ["--primary"]
- else:
- # there are two screens
- args[internalConnector] = ["--mode", intResolution] # set internal screen to desired resolution
- # set position
- if extPosition.posLeft.isChecked():
- args[usedExternalConnector] += ["--left-of", internalConnector]
- else:
- args[usedExternalConnector] += ["--right-of", internalConnector]
- # set primary screen
- if extPosition.primExt.isChecked():
- args[usedExternalConnector] += ["--primary"]
- else:
- args[internalConnector] += ["--primary"]
-else:
- # use first resolution
- args[internalConnector] = ["--mode", res2xrandr(connectors[internalConnector][0]), "--primary"]
-# and do it
-call = ["xrandr"]
-for name in args:
- call += ["--output", name] + args[name]
-print "Call that will be made:",call
-subprocess.check_call(call)
+# if we run top-level
+if __name__ == "__main__":
+ try:
+ # parse command-line arguments
+ parser = argparse.ArgumentParser(description='easy Display Setup for Laptops')
+ parser.add_argument("-f", "--frontend",
+ dest="frontend",
+ help="The frontend to be used for user interaction")
+ parser.add_argument("-r", "--relative-position",
+ dest="rel_position", choices=RelativeScreenPosition.__names__.keys(),
+ help="Position of external screen relative to internal one")
+ parser.add_argument("-i", "--internal-only",
+ dest="internal_only", action='store_true',
+ help="Enable internal screen, disable all the others (as if no external screen was connected")
+ cmdArgs = parser.parse_args()
+
+ # load frontend
+ frontend = getFrontend(cmdArgs.frontend)
+
+ # load connectors and classify them
+ connectors = getXrandrInformation()
+ (internalConnector, externalConnectors) = classifyConnectors(connectors)
+
+ # default: screen off
+ connectorArgs = {} # maps connector names to xrand arguments
+ for c in externalConnectors+[internalConnector]:
+ connectorArgs[c] = ["--off"]
+
+ # check whether we got an external screen or not
+ # Check what to do
+ usedExternalConnector = findAvailableConnector(externalConnectors, connectors) # *the* external connector which is actually used
+ hasExternal = not cmdArgs.internal_only and usedExternalConnector is not None
+ if hasExternal:
+ # compute the list of resolutons available on both
+ commonRes = [res for res in connectors[usedExternalConnector] if res in connectors[internalConnector]]
+ # there's an external screen connected, we need to get a setup
+ if cmdArgs.rel_position is not None:
+ # use command-line arguments
+ relPosition = RelativeScreenPosition.__names__[cmdArgs.rel_position]
+ if relPosition == RelativeScreenPosition.MIRROR:
+ setup = ScreenSetup(relPosition, commonRes[0], commonRes[0]) # use default resolutions
+ else:
+ setup = ScreenSetup(relPosition, connectors[internalConnector][0], connectors[usedExternalConnector][0]) # use default resolutions
+ else:
+ # use GUI
+ setup = frontend.setup(connectors[internalConnector], connectors[usedExternalConnector], commonRes)
+ if setup is None: sys.exit(1) # the user canceled
+ # apply it
+ connectorArgs[internalConnector] = setup.getInternalArgs()
+ connectorArgs[usedExternalConnector] = setup.getExternalArgs(internalConnector)
+ else:
+ # use first resolution of internal connector
+ connectorArgs[internalConnector] = ["--mode", res2xrandr(connectors[internalConnector][0]), "--primary"]
+
+ # and do it
+ call = ["xrandr"]
+ for name in connectorArgs:
+ call += ["--output", name] + connectorArgs[name]
+ print("Call that will be made:",call)
+ subprocess.check_call(call)
+
+ # make sure the internal screen is really, *really* turned on if there is no external screen
+ if not hasExternal:
+ backlight = float(subprocess.check_output(["xbacklight", "-get"]).strip())
+ if backlight == 0: # it's completely turned off, we better enable it
+ subprocess.check_call(["xbacklight", "-set", "100"])
+ except Exception as e:
+ frontend.error(str(e))
+ raise