93 lines
2.5 KiB
Python
93 lines
2.5 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
#
|
||
|
|
# Qtile window switcher for Wayland only
|
||
|
|
#
|
||
|
|
# Justine Smithies - 2021
|
||
|
|
#
|
||
|
|
# Requirements are: qtile running under Wayland and rofi-lbonn-wayland-git
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
from libqtile.command.client import InteractiveCommandClient
|
||
|
|
|
||
|
|
client = InteractiveCommandClient()
|
||
|
|
windows = client.windows()
|
||
|
|
# Get the number of open windows
|
||
|
|
lines = (len(windows))
|
||
|
|
x = 0
|
||
|
|
names = []
|
||
|
|
ids = []
|
||
|
|
groups = []
|
||
|
|
appids = []
|
||
|
|
while x < lines:
|
||
|
|
# client.window[window_id].focus() etc
|
||
|
|
win = windows[x]
|
||
|
|
name = win.get("name")
|
||
|
|
names.append(name)
|
||
|
|
id = win.get("id")
|
||
|
|
ids.append(id)
|
||
|
|
group = win.get("group")
|
||
|
|
groups.append(group)
|
||
|
|
appid = win.get("wm_class")[0]
|
||
|
|
appids.append(appid)
|
||
|
|
print(name, id, group, appid)
|
||
|
|
x += 1
|
||
|
|
print(windows)
|
||
|
|
|
||
|
|
# Reverse the tuples
|
||
|
|
ids = ids[::-1]
|
||
|
|
groups = groups[::-1]
|
||
|
|
names = names[::-1]
|
||
|
|
appids = appids[::-1]
|
||
|
|
|
||
|
|
# Combine all the information into a string to be piped through rofi -dmenu
|
||
|
|
d = dict(os.environ)
|
||
|
|
outputs = []
|
||
|
|
output = []
|
||
|
|
icon = []
|
||
|
|
x = 0
|
||
|
|
print(lines)
|
||
|
|
while x < lines:
|
||
|
|
id = ids[x]
|
||
|
|
id = str(id)
|
||
|
|
appid = (appids[x]).strip('\n')
|
||
|
|
name = names[x]
|
||
|
|
# If name is > 40 characters then shorten it to max 40 and append ...
|
||
|
|
if len(name) > 40:
|
||
|
|
name = name[:40] + '...'
|
||
|
|
group = groups[x]
|
||
|
|
# Hacky fix to get correct icon for appids that begin with org.
|
||
|
|
if 'org.' in appid:
|
||
|
|
icon = appid.split(".")
|
||
|
|
icon = str(icon[2].lower())
|
||
|
|
else:
|
||
|
|
icon = appid
|
||
|
|
output = ' ' + appid + ' - ' + name + ' on group: ' + group + ' (' + id + ')' + r'\0icon\x1f' + icon + '\n'
|
||
|
|
outputs.append(str(output))
|
||
|
|
x += 1
|
||
|
|
|
||
|
|
outputs = ''.join(outputs)
|
||
|
|
# Remove last blank line
|
||
|
|
outputs = os.linesep.join([s for s in outputs.splitlines() if s])
|
||
|
|
print(outputs)
|
||
|
|
|
||
|
|
# Send options to rofi
|
||
|
|
d["OPTIONS"] = outputs
|
||
|
|
d["LINES"] = str(lines)
|
||
|
|
result = subprocess.Popen('echo -e "$OPTIONS" | rofi -theme themes/windowswitcher.rasi -i -dmenu -l $LINES -mesg "Window Switcher"', shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, env=d)
|
||
|
|
|
||
|
|
# Get selected option and process
|
||
|
|
selected_option = str(result.communicate())
|
||
|
|
print(selected_option)
|
||
|
|
# Search for the content between the last set of brackets
|
||
|
|
selected_option = re.sub('^.*\((.*?)\)[^\(]*$', '\g<1>', selected_option)
|
||
|
|
print(selected_option)
|
||
|
|
if True in [char.isdigit() for char in selected_option]:
|
||
|
|
selected_option = int(selected_option)
|
||
|
|
client.window[selected_option].group.toscreen()
|
||
|
|
client.window[selected_option].focus()
|
||
|
|
exit()
|
||
|
|
else:
|
||
|
|
exit()
|