#!/usr/bin/env python3

import sys
import os

# 1. Run `xrandr -q | grep connected` to identify what the correct display IDs are
# 2. set keyboard shortcuts like <ctrl>+<super>+<alt>+1 to call /home/username/set_display.py desktop 
if len(sys.argv) <= 1:
   sys.argv.append("all")
if sys.argv[1].strip() == "laptop":
    COMMAND = "xrandr --output eDP1 --auto --output DP1-2 --off"
elif sys.argv[1].strip() == "desktop":
    COMMAND = "xrandr --output eDP1 --off --output DP1-2 --auto"
else:
    COMMAND = "xrandr --output eDP1 --auto --output DP1-2 --auto"

os.system(COMMAND)