diff options
Diffstat (limited to 'alot/buffers/namedqueries.py')
-rw-r--r-- | alot/buffers/namedqueries.py | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/alot/buffers/namedqueries.py b/alot/buffers/namedqueries.py index ad7c5fb6..2ff7ac08 100644 --- a/alot/buffers/namedqueries.py +++ b/alot/buffers/namedqueries.py @@ -13,16 +13,20 @@ class NamedQueriesBuffer(Buffer): modename = 'namedqueries' - def __init__(self, ui, filtfun): - self.ui = ui + _dbman = None + + def __init__(self, dbman, filtfun): + self._dbman = dbman + self.filtfun = filtfun self.isinitialized = False self.querylist = None self.rebuild() - Buffer.__init__(self, ui, self.body) + + super().__init__() def rebuild(self): - self.queries = self.ui.dbman.get_named_queries() + self.queries = self._dbman.get_named_queries() if self.isinitialized: focusposition = self.querylist.get_focus()[1] @@ -32,8 +36,8 @@ class NamedQueriesBuffer(Buffer): lines = [] for (num, key) in enumerate(self.queries): value = self.queries[key] - count = self.ui.dbman.count_messages('query:"%s"' % key) - count_unread = self.ui.dbman.count_messages('query:"%s" and ' + 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) |