Linux tray applet for autorandr

autorandrtray.py

#!/usr/bin/python
# For: https://github.com/phillipberndt/autorandr
# Based on: https://fosspost.org/tutorials/custom-system-tray-icon-indicator-linux
import os
from gi import require_version
require_version('Gtk', '3.0')
require_version('AppIndicator3', '0.1')

from gi.repository import Gtk as gtk, AppIndicator3 as appindicator
from subprocess import Popen, PIPE

def main():
  indicator = appindicator.Indicator.new("autorandrtray", "semi-starred-symbolic", appindicator.IndicatorCategory.APPLICATION_STATUS)
  indicator.set_status(appindicator.IndicatorStatus.ACTIVE)
  indicator.set_menu(menu())
  gtk.main()

def menu():
  menu = gtk.Menu()

  profiles = ['common (virtual)', 'clone-largest (virtual)', 'horizontal (virtual)', 'vertical (virtual)']

  pipe = Popen('autorandr', shell=True, stdout=PIPE)
  for line in pipe.stdout:
    profiles.append(line.strip())

  auto = gtk.MenuItem('autodetect')
  auto.connect('activate', change_profile, '')
  menu.append(auto)

  menu.append(gtk.SeparatorMenuItem())
  
  for profile in profiles:
    command = gtk.MenuItem(profile)
    command.connect('activate', change_profile, profile.split(' ')[0])
    menu.append(command)

  menu.append(gtk.SeparatorMenuItem())

  exittray = gtk.MenuItem('Exit Tray')
  exittray.connect('activate', quit)
  menu.append(exittray)
  
  menu.show_all()
  return menu
  
def change_profile(_, profile):
  os.system("autorandr --change " + profile)

def quit(_):
  gtk.main_quit()

if __name__ == "__main__":
  main()