diff options
Diffstat (limited to 'mitmproxy/tools/console/grideditor/editors.py')
| -rw-r--r-- | mitmproxy/tools/console/grideditor/editors.py | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/mitmproxy/tools/console/grideditor/editors.py b/mitmproxy/tools/console/grideditor/editors.py index fbe48a1a..fffd782c 100644 --- a/mitmproxy/tools/console/grideditor/editors.py +++ b/mitmproxy/tools/console/grideditor/editors.py @@ -1,3 +1,4 @@ +import urwid import typing from mitmproxy import exceptions @@ -100,12 +101,13 @@ class CookieEditor(base.FocusEditor): flow.request.cookies = vals -class CookieAttributeEditor(base.GridEditor): +class CookieAttributeEditor(base.FocusEditor): title = "Editing Set-Cookie attributes" columns = [ col_text.Column("Name"), col_text.Column("Value"), ] + grideditor = None # type: base.BaseGridEditor def data_in(self, data): return [(k, v or "") for k, v in data] @@ -119,6 +121,20 @@ class CookieAttributeEditor(base.GridEditor): ret.append(i) return ret + def layout_pushed(self, prev): + if self.grideditor.master.view.focus.flow: + self._w = base.BaseGridEditor( + self.grideditor.master, + self.title, + self.columns, + self.grideditor.walker.get_current_value(), + self.grideditor.set_subeditor_value, + self.grideditor.walker.focus, + self.grideditor.walker.focus_col + ) + else: + self._w = urwid.Pile([]) + class SetCookieEditor(base.FocusEditor): title = "Edit SetCookie Header" |