# HG changeset patch
# User Björn Ricks <bjoern.ricks@intevation.de>
# Date 1306323377 -7200
# Node ID 745c644d95641905506a74638e45bd346355e329
# Parent  90b61a24742b61a8a2936b629901ef6e98d4e6f4
Add new state for editing entries

diff -r 90b61a24742b -r 745c644d9564 getan/states.py
--- a/getan/states.py	Wed May 25 13:33:23 2011 +0200
+++ b/getan/states.py	Wed May 25 13:36:17 2011 +0200
@@ -531,3 +531,36 @@
         self.controller.add_project(self.key, description)
         self.controller.view.set_footer_text(self.msg('choose_proj'), 'question')
         return PausedProjectsState(self.controller, self.view)
+
+
+class EditEntryState(HandleUserInputState):
+    messages = {
+        'edit_entry': _('Edit entry text: '),
+    }
+
+    def __init__(self, state, controller, view, entry):
+        controller.view.set_footer_text(self.msg('edit_entry'),
+                                             'question', True)
+        super(EditEntryState, self).__init__(controller, view,
+                None, controller.view.view.get_footer())
+        self.footer.set_edit_text(entry.desc)
+        self.footer.set_edit_pos(len(self.footer.edit_text))
+        self.view.frame.set_focus('footer')
+        self.entry = entry
+        self.state = state
+        logger.debug("EditEntryState: Entry %s" % entry)
+
+    def enter(self):
+        entry_desc = self.footer.get_edit_text()
+        if entry_desc == '':
+            return self
+        entry = self.entry
+        entry.desc = entry_desc
+        self.controller.update_entry(entry)
+        return self.exit()
+
+    def exit(self):
+        self.controller.view.set_footer_text("", 'question', False)
+        return DefaultEntryListState(self.state, self.controller, self.view)
+
+