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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
# Copyright (C) 2011-2018 Patrick Totzke <patricktotzke@gmail.com>
# This file is released under the GNU GPL, version 3 or a later revision.
# For further details see the COPYING file
import urwid
from .buffer import Buffer
from ..settings.const import settings
from ..widgets.namedqueries import QuerylineWidget
class NamedQueriesBuffer(Buffer):
"""lists named queries present in the notmuch database"""
modename = 'namedqueries'
_dbman = None
_queries = None
def __init__(self, dbman, filtfun):
self._dbman = dbman
self.filtfun = filtfun
self.isinitialized = False
self.querylist = None
self.rebuild()
super().__init__()
def rebuild(self):
queries = self._dbman.get_named_queries()
sort_hook = settings.get_hook('namedqueries_sort')
if sort_hook:
queries = sort_hook(queries)
else:
queries = list(queries.items())
self._queries = queries
if self.isinitialized:
focusposition = self.querylist.get_focus()[1]
else:
focusposition = 0
lines = []
for num, (key, value) in enumerate(queries):
count = self._dbman.count_messages('query:"%s"' % key)
count_unread = self._dbman.count_messages('query:"%s" and '
'tag:unread' % key)
line = QuerylineWidget(key, value, count, count_unread)
if (num % 2) == 0:
attr = settings.get_theming_attribute('namedqueries',
'line_even')
else:
attr = settings.get_theming_attribute('namedqueries',
'line_odd')
focus_att = settings.get_theming_attribute('namedqueries',
'line_focus')
line = urwid.AttrMap(line, attr, focus_att)
lines.append(line)
self.querylist = urwid.ListBox(urwid.SimpleListWalker(lines))
self.body = self.querylist
self.querylist.set_focus(focusposition % len(queries))
self.isinitialized = True
def focus_first(self):
"""Focus the first line in the query list."""
self.body.set_focus(0)
def focus_last(self):
allpos = self.querylist.body.positions(reverse=True)
if allpos:
lastpos = allpos[0]
self.body.set_focus(lastpos)
def get_selected_query(self):
"""returns selected query"""
return self.querylist.get_focus()[0].original_widget.query
async def get_info(self):
info = {}
info['query_count'] = len(self._queries)
return info
|