def edit_item(self):
self.editing = True
self.edit_widget = AdvancedEdit(self.parent_ui, self.key_bindings, caption="> ", edit_text=self.todo.raw + ' ')
self.edit_widget.set_completion_method(self.completions)
self._w = urwid.AttrMap(self.edit_widget, 'plain_selected')
class TodoWidget(urwid.Button):
def __init__(self, todo, key_bindings, colorscheme, parent_ui, editing=False, wrapping='clip', border='no border'):
super(TodoWidget, self).__init__("")
self.todo = todo
self.key_bindings = key_bindings
self.wrapping = wrapping
self.border = border
self.colorscheme = colorscheme
self.parent_ui = parent_ui
self.editing = editing
self.edit_widget = None
urwid.connect_signal(self, 'click', parent_ui.todo_widget_clicked)
if editing:
self.edit_item()
else:
self.update_todo()
def selectable(self):
return True
def update_todo(self):
if self.parent_ui.searching and self.parent_ui.search_string:
text = urwid.Text(self.todo.highlight_search_matches(), wrap=self.wrapping)
elif self.border == 'bordered':
text = urwid.Text(self.todo.highlight(show_due_date=False, show_contexts=False, show_projects=False), wrap=self.wrapping)
else:
text = urwid.Text(self.todo.colored, wrap=self.wrapping)
if self.border == 'bordered':
lt = ''
if self.todo.due_date:
lt = ('due_date', "due:{0}".format(self.todo.due_date))
t = [
('context', ' '.join(self.todo.contexts))
]
if self.todo.contexts and self.todo.projects:
t.append(' ')
t.append(('project', ' '.join(self.todo.projects)))
bc = 'plain'
if self.todo.priority and self.todo.priority in 'ABCDEF':
bc = "priority_{0}".format(self.todo.priority.lower())
text = TodoLineBox(text, top_left_title=lt, bottom_right_title=t, border_color=bc,)
elif self.border == 'tree':
if self.todo.projects:
project = self.todo.projects[0]
depth = project.count('.')
colored_list = self.todo.highlight(show_projects=False)
if isinstance(colored_list[1][0], basestring):
colored_list[1][0] = u' ' * (depth+1) + colored_list[1][0]
text = urwid.Text(colored_list, wrap=self.wrapping)
self._w = urwid.AttrMap(urwid.AttrMap(text, None, 'selected'),
None, self.colorscheme.focus_map)
def edit_item(self):
self.editing = True
self.edit_widget = AdvancedEdit(self.parent_ui, self.key_bindings, caption="> ", edit_text=self.todo.raw + ' ')
self.edit_widget.set_completion_method(self.completions)
self._w = urwid.AttrMap(self.edit_widget, 'plain_selected')
def completions(self, text, completion_data=None):
completion_data = completion_data or {}
space = text.rfind(" ")
start = text[space+1:]
words = self.parent_ui.todos.all_contexts() + self.parent_ui.todos.all_projects()
try:
start_idx = words.index(completion_data['last_word'])+1
if start_idx == len(words):
start_idx = 0
except (KeyError, ValueError):
start_idx = 0
for idx in list(range(start_idx, len(words))) + list(range(0, start_idx)):
if words[idx].lower().startswith(start.lower()):
completion_data['last_word'] = words[idx]
return text[:space+1] + words[idx] + (': ' if space < 0 else '')
return text
def save_item(self):
text = self._w.original_widget.edit_text.strip()
if text:
self.todo.update(text)
self.update_todo()
if self.parent_ui.filter_panel_is_open:
self.parent_ui.update_filters(new_contexts=self.todo.contexts, new_projects=self.todo.projects)
else:
self.parent_ui.delete_todo()
self.parent_ui.save_todos()
self.editing = False
def save_and_append(self):
self.save_item()
self.parent_ui.add_new_todo(position='insert_after')
def noop(self):
pass
def keypress(self, size, key):
context = 'todo' + (':editing' if self.editing else '')
if handle_keypress(self, key, context):
return
elif self.editing:
return self._w.keypress(size, key)
else:
return key
