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
|
import urwid
import logging
from notmuch import NotmuchError
class IteratorWalker(urwid.ListWalker):
def __init__(self, it, containerclass):
self.it = it
self.containerclass = containerclass
self.lines = []
self.focus = 0
self.empty = False
def get_focus(self):
return self._get_at_pos(self.focus)
def set_focus(self, focus):
self.focus = focus
self._modified()
def get_next(self, start_from):
return self._get_at_pos(start_from + 1)
def get_prev(self, start_from):
return self._get_at_pos(start_from - 1)
def _get_at_pos(self, pos):
if pos < 0: # pos too low
return (None, None)
elif pos > len(self.lines): # pos too high
return (None, None)
elif len(self.lines) > pos: # pos already cached
return (self.lines[pos], pos)
else: # pos not cached yet, look at next item from iterator
if self.empty: # iterator is empty
return (None, None)
else:
widget = self._get_next_item()
if widget:
return (widget, pos)
else:
return (None, None)
def _get_next_item(self):
try:
next_obj = self.it.next()
next_widget = self.containerclass(next_obj)
self.lines.append(next_widget)
except StopIteration:
next_widget = None
self.empty = True
return next_widget
class NotmuchIteratorWalker(IteratorWalker):
def _get_next_item(self):
logging.error("it still there")
try:
next_obj = self.it.next()
logging.error("next obj: %s" % next_obj)
next_widget = self.containerclass(next_obj)
self.lines.append(next_widget)
except NotmuchError:
next_widget = None
return next_widget
|