1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
from collections import deque
from subprocess import check_output
from . import Lists, App
class Menu:
def __init__(self, i3, menu, menu_args):
self._i3 = i3
self._menu = menu
self._menu_args = menu_args
def show_menu(self, items):
menu_input = bytes(str.join('\n', items), 'UTF-8')
menu_cmd = [self._menu] + ['-l', str(len(items))] + self._menu_args
menu_result = check_output(menu_cmd, input=menu_input)
return menu_result.decode().strip()
def show_menu_app(self, apps):
titles = list(map(lambda a: a.get_title(), apps))
selected_title = self.show_menu(titles)
selected_app = Lists.find_app_by_title(selected_title, apps)
tree = self._i3.get_tree()
con = tree.find_by_id(selected_app.get_con_id())
con.command('focus')
def show_menu_container_info(self, containers_info):
titles = self._get_titles_with_app_prefix(containers_info)
titles_with_suffix = self._add_uniqu_suffix(titles)
infos_by_title = dict(zip(titles_with_suffix, containers_info))
selected_title = self.show_menu(titles_with_suffix)
selected_info = infos_by_title[selected_title]
tree = self._i3.get_tree()
con = tree.find_by_id(selected_info["id"])
con.command('focus')
def _get_titles_with_app_prefix(self, containers_info):
return list(map(lambda i: App(i).get_title() + ': ' + i["window_title"], containers_info))
def _add_uniqu_suffix(self, titles):
counters = dict()
titles_with_suffix = []
for title in titles:
counters[title] = counters[title] + 1 if title in counters else 1
if counters[title] > 1:
title = f'{title} ({counters[title]})'
titles_with_suffix.append(title)
return titles_with_suffix
|
