Wofi Window switcher

sway-list-windows.py

# This script gets the list of all current windows and shows them on wofi
# The selected window get's focused.
import subprocess
from i3ipc import Connection, Event


def get_windows(conn):
    '''Given a sway connection object, return the current windows infromation

    return: Dictionary with window id as key'''
    containers = conn.get_tree().descendants()
    containers = filter(lambda x: x.type in ['con', 'floating_con'], containers)
    windows = []

    for con in containers:
        windows.append(window_crawl(con))

    return {f'{x.id:<5}{", ".join(x.marks):<20}{x.name}': x for x in windows}
    
def window_crawl(con):
    '''Recursively get containers without descendants e.g windows'''
    # TODO: This function is unneccesary the library provides two functions
    # leaves() and scratchpad() that could accomplish this.
    if con.descendants():
        for x in con.descendants():
            return window_crawl(x)
    else:
        return con

def wofi(options: list):
    '''Calls rofi in dmenu mode with the given selection options
   
        It returns the selection
    '''
    return subprocess.check_output(['wofi','-i', '-k', '/dev/null', '-d'],
            input='\n'.join(options), encoding='UTF-8').strip('\n')

if __name__ == "__main__":
    sway = Connection()
    windows = get_windows(sway)
    result = wofi(windows.keys())
    window = windows[result]
    window.command('focus')