def keypress(self, xxx_todo_changeme, key): (maxcol, ) = xxx_todo_changeme key = common.shortcuts(key) if key == "E": signals.status_prompt_onekey.send( self, prompt="Export to file", keys=[(e[0], e[1]) for e in export.EXPORTERS], callback=common.export_to_clip_or_file, args=(None, self.flow, common.ask_save_path)) # elif key == "C": # signals.status_prompt_onekey.send( # self, # prompt = "Export to clipboard", # keys = [(e[0], e[1]) for e in export.EXPORTERS], # callback = common.export_to_clip_or_file, # args = (None, self.flow, common.copy_to_clipboard_or_prompt) # ) elif key == "b": common.ask_save_body(None, self.flow) else: return key
def keypress(self, xxx_todo_changeme, key): (maxcol, ) = xxx_todo_changeme key = common.shortcuts(key) if key == "a": self.flow.resume(self.master) signals.flowlist_change.send(self) elif key == "d": if self.flow.killable: self.flow.kill(self.master) self.master.view.remove(self.master.view.focus.flow) elif key == "D": cp = self.flow.copy() self.master.view.add(cp) self.master.view.focus.flow = cp elif key == "m": self.flow.marked = not self.flow.marked signals.flowlist_change.send(self) elif key == "M": self.master.view.toggle_marked() elif key == "r": try: self.master.replay_request(self.flow) except exceptions.ReplayException as e: signals.add_log("Replay error: %s" % e, "warn") signals.flowlist_change.send(self) elif key == "S": def stop_server_playback(response): if response == "y": self.master.options.server_replay = [] a = self.master.addons.get("serverplayback") if a.count(): signals.status_prompt_onekey.send( prompt="Stop current server replay?", keys=( ("yes", "y"), ("no", "n"), ), callback=stop_server_playback, ) else: signals.status_prompt_onekey.send( prompt="Server Replay", keys=( ("all flows", "a"), ("this flow", "t"), ), callback=self.server_replay_prompt, ) elif key == "U": for f in self.master.view: f.marked = False signals.flowlist_change.send(self) elif key == "V": if not self.flow.modified(): signals.status_message.send(message="Flow not modified.") return self.flow.revert() signals.flowlist_change.send(self) signals.status_message.send(message="Reverted.") elif key == "w": signals.status_prompt_onekey.send( self, prompt="Save", keys=( ("listed flows", "l"), ("this flow", "t"), ), callback=self.save_flows_prompt, ) elif key == "X": if self.flow.killable: self.flow.kill(self.master) elif key == "enter": if self.flow.request: self.master.view_flow(self.flow) elif key == "|": signals.status_prompt_path.send( prompt="Send flow to script", callback=self.master.run_script_once, args=(self.flow, )) elif key == "E": signals.status_prompt_onekey.send( self, prompt="Export to file", keys=[(e[0], e[1]) for e in export.EXPORTERS], callback=common.export_to_clip_or_file, args=(None, self.flow, common.ask_save_path)) elif key == "C": signals.status_prompt_onekey.send( self, prompt="Export to clipboard", keys=[(e[0], e[1]) for e in export.EXPORTERS], callback=common.export_to_clip_or_file, args=(None, self.flow, common.copy_to_clipboard_or_prompt)) elif key == "b": common.ask_save_body(None, self.flow) else: return key
def keypress(self, xxx_todo_changeme, key): (maxcol,) = xxx_todo_changeme key = common.shortcuts(key) if key == "a": self.flow.resume(self.master) signals.flowlist_change.send(self) elif key == "d": if self.flow.killable: self.flow.kill(self.master) self.master.view.remove(self.master.view.focus.flow) elif key == "D": cp = self.flow.copy() self.master.view.add(cp) self.master.view.focus.flow = cp elif key == "m": self.flow.marked = not self.flow.marked signals.flowlist_change.send(self) elif key == "M": self.master.view.toggle_marked() elif key == "r": try: self.master.replay_request(self.flow) except exceptions.ReplayException as e: signals.add_log("Replay error: %s" % e, "warn") signals.flowlist_change.send(self) elif key == "S": def stop_server_playback(response): if response == "y": self.master.options.server_replay = [] a = self.master.addons.get("serverplayback") if a.count(): signals.status_prompt_onekey.send( prompt = "Stop current server replay?", keys = ( ("yes", "y"), ("no", "n"), ), callback = stop_server_playback, ) else: signals.status_prompt_onekey.send( prompt = "Server Replay", keys = ( ("all flows", "a"), ("this flow", "t"), ), callback = self.server_replay_prompt, ) elif key == "U": for f in self.master.view: f.marked = False signals.flowlist_change.send(self) elif key == "V": if not self.flow.modified(): signals.status_message.send(message="Flow not modified.") return self.flow.revert() signals.flowlist_change.send(self) signals.status_message.send(message="Reverted.") elif key == "w": signals.status_prompt_onekey.send( self, prompt = "Save", keys = ( ("listed flows", "l"), ("this flow", "t"), ), callback = self.save_flows_prompt, ) elif key == "X": if self.flow.killable: self.flow.kill(self.master) elif key == "enter": if self.flow.request: self.master.view_flow(self.flow) elif key == "|": signals.status_prompt_path.send( prompt = "Send flow to script", callback = self.master.run_script_once, args = (self.flow,) ) elif key == "E": signals.status_prompt_onekey.send( self, prompt = "Export to file", keys = [(e[0], e[1]) for e in export.EXPORTERS], callback = common.export_to_clip_or_file, args = (None, self.flow, common.ask_save_path) ) elif key == "C": signals.status_prompt_onekey.send( self, prompt = "Export to clipboard", keys = [(e[0], e[1]) for e in export.EXPORTERS], callback = common.export_to_clip_or_file, args = (None, self.flow, common.copy_to_clipboard_or_prompt) ) elif key == "b": common.ask_save_body(None, self.flow) else: return key
def keypress(self, size, key): conn = None # type: Optional[Union[http.HTTPRequest, http.HTTPResponse]] if self.tab_offset == TAB_REQ: conn = self.flow.request elif self.tab_offset == TAB_RESP: conn = self.flow.response key = super().keypress(size, key) # Special case: Space moves over to the next flow. # We need to catch that before applying common.shortcuts() if key == " ": self.view_next_flow(self.flow) return key = common.shortcuts(key) if key in ("up", "down", "page up", "page down"): # Pass scroll events to the wrapped widget self._w.keypress(size, key) elif key == "a": self.flow.resume(self.master) signals.flow_change.send(self, flow=self.flow) elif key == "A": self.master.accept_all() signals.flow_change.send(self, flow=self.flow) elif key == "d": if self.flow.killable: self.flow.kill(self.master) self.view.remove(self.flow) if not self.view.focus.flow: self.master.view_flowlist() else: self.view_flow(self.view.focus.flow) elif key == "D": cp = self.flow.copy() self.master.view.add(cp) self.master.view.focus.flow = cp self.view_flow(cp) signals.status_message.send(message="Duplicated.") elif key == "p": self.view_prev_flow(self.flow) elif key == "r": try: self.master.replay_request(self.flow) except exceptions.ReplayException as e: signals.add_log("Replay error: %s" % e, "warn") signals.flow_change.send(self, flow=self.flow) elif key == "V": if self.flow.modified(): self.flow.revert() signals.flow_change.send(self, flow=self.flow) signals.status_message.send(message="Reverted.") else: signals.status_message.send(message="Flow not modified.") elif key == "W": signals.status_prompt_path.send(prompt="Save this flow", callback=self.master.save_one_flow, args=(self.flow, )) elif key == "|": signals.status_prompt_path.send( prompt="Send flow to script", callback=self.master.run_script_once, args=(self.flow, )) elif key == "e": if self.tab_offset == TAB_REQ: signals.status_prompt_onekey.send(prompt="Edit request", keys=( ("cookies", "c"), ("query", "q"), ("path", "p"), ("url", "u"), ("header", "h"), ("form", "f"), ("raw body", "r"), ("method", "m"), ), callback=self.edit) elif self.tab_offset == TAB_RESP: signals.status_prompt_onekey.send(prompt="Edit response", keys=( ("cookies", "c"), ("code", "o"), ("message", "m"), ("header", "h"), ("raw body", "r"), ), callback=self.edit) else: signals.status_message.send( message="Tab to the request or response", expire=1) elif key in set("bfgmxvzEC") and not conn: signals.status_message.send( message="Tab to the request or response", expire=1) return elif key == "b": if self.tab_offset == TAB_REQ: common.ask_save_body("q", self.flow) else: common.ask_save_body("s", self.flow) elif key == "f": self.view.settings[self.flow][(self.tab_offset, "fullcontents")] = True signals.flow_change.send(self, flow=self.flow) signals.status_message.send(message="Loading all body data...") elif key == "m": p = list(contentviews.view_prompts) p.insert(0, ("Clear", "C")) signals.status_prompt_onekey.send( self, prompt="Display mode", keys=p, callback=self.change_this_display_mode) elif key == "E": if self.tab_offset == TAB_REQ: scope = "q" else: scope = "s" signals.status_prompt_onekey.send( self, prompt="Export to file", keys=[(e[0], e[1]) for e in export.EXPORTERS], callback=common.export_to_clip_or_file, args=(scope, self.flow, common.ask_save_path)) elif key == "C": if self.tab_offset == TAB_REQ: scope = "q" else: scope = "s" signals.status_prompt_onekey.send( self, prompt="Export to clipboard", keys=[(e[0], e[1]) for e in export.EXPORTERS], callback=common.export_to_clip_or_file, args=(scope, self.flow, common.copy_to_clipboard_or_prompt)) elif key == "x": conn.content = None signals.flow_change.send(self, flow=self.flow) elif key == "v": if conn.raw_content: t = conn.headers.get("content-type") if "EDITOR" in os.environ or "PAGER" in os.environ: self.master.spawn_external_viewer( conn.get_content(strict=False), t) else: signals.status_message.send( message="Error! Set $EDITOR or $PAGER.") elif key == "z": self.flow.backup() e = conn.headers.get("content-encoding", "identity") if e != "identity": try: conn.decode() except ValueError: signals.status_message.send( message="Could not decode - invalid data?") else: signals.status_prompt_onekey.send( prompt="Select encoding: ", keys=( ("gzip", "z"), ("deflate", "d"), ("brotli", "b"), ), callback=self.encode_callback, args=(conn, )) signals.flow_change.send(self, flow=self.flow) else: # Key is not handled here. return key
def keypress(self, size, key): conn = None # type: Optional[Union[http.HTTPRequest, http.HTTPResponse]] if self.tab_offset == TAB_REQ: conn = self.flow.request elif self.tab_offset == TAB_RESP: conn = self.flow.response key = super().keypress(size, key) # Special case: Space moves over to the next flow. # We need to catch that before applying common.shortcuts() if key == " ": self.view_next_flow(self.flow) return key = common.shortcuts(key) if key in ("up", "down", "page up", "page down"): # Pass scroll events to the wrapped widget self._w.keypress(size, key) elif key == "a": self.flow.resume(self.master) signals.flow_change.send(self, flow = self.flow) elif key == "A": self.master.accept_all() signals.flow_change.send(self, flow = self.flow) elif key == "d": if self.flow.killable: self.flow.kill(self.master) self.view.remove(self.flow) if not self.view.focus.flow: self.master.view_flowlist() else: self.view_flow(self.view.focus.flow) elif key == "D": cp = self.flow.copy() self.master.view.add(cp) self.master.view.focus.flow = cp self.view_flow(cp) signals.status_message.send(message="Duplicated.") elif key == "p": self.view_prev_flow(self.flow) elif key == "r": try: self.master.replay_request(self.flow) except exceptions.ReplayException as e: signals.add_log("Replay error: %s" % e, "warn") signals.flow_change.send(self, flow = self.flow) elif key == "V": if self.flow.modified(): self.flow.revert() signals.flow_change.send(self, flow = self.flow) signals.status_message.send(message="Reverted.") else: signals.status_message.send(message="Flow not modified.") elif key == "W": signals.status_prompt_path.send( prompt = "Save this flow", callback = self.master.save_one_flow, args = (self.flow,) ) elif key == "|": signals.status_prompt_path.send( prompt = "Send flow to script", callback = self.master.run_script_once, args = (self.flow,) ) elif key == "e": if self.tab_offset == TAB_REQ: signals.status_prompt_onekey.send( prompt="Edit request", keys=( ("cookies", "c"), ("query", "q"), ("path", "p"), ("url", "u"), ("header", "h"), ("form", "f"), ("raw body", "r"), ("method", "m"), ), callback=self.edit ) elif self.tab_offset == TAB_RESP: signals.status_prompt_onekey.send( prompt="Edit response", keys=( ("cookies", "c"), ("code", "o"), ("message", "m"), ("header", "h"), ("raw body", "r"), ), callback=self.edit ) else: signals.status_message.send( message="Tab to the request or response", expire=1 ) elif key in set("bfgmxvzEC") and not conn: signals.status_message.send( message = "Tab to the request or response", expire = 1 ) return elif key == "b": if self.tab_offset == TAB_REQ: common.ask_save_body("q", self.flow) else: common.ask_save_body("s", self.flow) elif key == "f": self.view.settings[self.flow][(self.tab_offset, "fullcontents")] = True signals.flow_change.send(self, flow = self.flow) signals.status_message.send(message="Loading all body data...") elif key == "m": p = list(contentviews.view_prompts) p.insert(0, ("Clear", "C")) signals.status_prompt_onekey.send( self, prompt = "Display mode", keys = p, callback = self.change_this_display_mode ) elif key == "E": if self.tab_offset == TAB_REQ: scope = "q" else: scope = "s" signals.status_prompt_onekey.send( self, prompt = "Export to file", keys = [(e[0], e[1]) for e in export.EXPORTERS], callback = common.export_to_clip_or_file, args = (scope, self.flow, common.ask_save_path) ) elif key == "C": if self.tab_offset == TAB_REQ: scope = "q" else: scope = "s" signals.status_prompt_onekey.send( self, prompt = "Export to clipboard", keys = [(e[0], e[1]) for e in export.EXPORTERS], callback = common.export_to_clip_or_file, args = (scope, self.flow, common.copy_to_clipboard_or_prompt) ) elif key == "x": conn.content = None signals.flow_change.send(self, flow=self.flow) elif key == "v": if conn.raw_content: t = conn.headers.get("content-type") if "EDITOR" in os.environ or "PAGER" in os.environ: self.master.spawn_external_viewer(conn.get_content(strict=False), t) else: signals.status_message.send( message = "Error! Set $EDITOR or $PAGER." ) elif key == "z": self.flow.backup() e = conn.headers.get("content-encoding", "identity") if e != "identity": try: conn.decode() except ValueError: signals.status_message.send( message = "Could not decode - invalid data?" ) else: signals.status_prompt_onekey.send( prompt = "Select encoding: ", keys = ( ("gzip", "z"), ("deflate", "d"), ("brotli", "b"), ), callback = self.encode_callback, args = (conn,) ) signals.flow_change.send(self, flow = self.flow) else: # Key is not handled here. return key