dotfiles/scripts/start-firefox.py

62 lines
1.9 KiB
Python

#!/usr/bin/env python3
from i3ipc import Connection, Event
import os
# moveWindowToWorkspace() moves a given window to a given workspace.
def moveWindowToWorkspace(window, workspace):
window.command('move window to workspace ' + workspace)
# getWindows() returns a list of all open windows on the desktop.
def getWindows(i3):
windows = []
for con in i3.get_tree():
if con.window and con.parent.type != 'dockarea':
windows.append(con)
return windows
# getWindowByName() returns a window with the given name.
def getWindowByName(name, windows):
for win in windows:
if name in win.name:
return win
return
# filterWindowsByClass() returns a filter list of windows by class.
def filterWindowsByClass(windowClass, windows):
return [w for w in windows if w.window_class == windowClass]
# doesWindowExist() returns if a given window exists.
def doesWindowExist(window):
return window != None
# launchProgram() launches a program on a given workspace.
def launchProgram(program, workspace):
i3.command('workspace ' + workspace + '; exec ' + program)
# isProgramRunning() returns if a program is running and if it is
# moves it to a given workspace.
def isProgramRunning(name, windows, workspace):
program = getWindowByName(name, windows)
if doesWindowExist(program):
moveWindowToWorkspace(program, workspace)
return True
return False
i3 = Connection()
firefoxWindows = filterWindowsByClass('Firefox', getWindows(i3))
# Music
if not isProgramRunning('YouTube Music', firefoxWindows, '10'):
launchProgram('exec firefox --new-window music.youtube.com', '10')
# Stocks
if not isProgramRunning('Robinhood', firefoxWindows, '10'):
os.system('python ~/.config/scripts/launch-stocks-tracker.py')
# YouTube
if not isProgramRunning('Subscriptions - YouTube', firefoxWindows, '10'):
launchProgram('firefox --new-window youtube.com/feed/subscriptions', '10')