Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-22 15:37:45 +00:00
Code Issues Projects Releases Wiki Activity

fix #349

Browse Source
This commit is contained in:
Maximilian Hils 2014-09-07 15:57:36 +02:00
parent b6986b8999
commit 58ea198698
3 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
libmproxy/console/__init__.py
View File

@ -179,7 +179,9 @@ class StatusBar(common.WWrap):
scheme += "2https" if dst[1] else "http"
r.append("[dest:%s]"%utils.unparse_url(scheme, *self.master.server.config.get_upstream_server.dst[2:]))
if self.master.scripts:
r.append("[scripts:%s]"%len(self.master.scripts))
r.append("[")
r.append(("heading_key", "s"))
r.append("cripts:%s]"%len(self.master.scripts))
# r.append("[lt:%0.3f]"%self.master.looptime)
if self.master.stream:
@ -784,8 +786,14 @@ class ConsoleMaster(flow.FlowMaster):
else:
self.view_flowlist()
def edit_scripts(self, *args, **kwargs):
pass
def edit_scripts(self, scripts):
commands = [x[0] for x in scripts] # remove outer array
if commands == [s.command for s in self.scripts]:
return
self.unload_scripts()
for command in commands:
self.load_script(command)
def loop(self):
changed = True
@ -878,7 +886,7 @@ class ConsoleMaster(flow.FlowMaster):
self.view_grideditor(
grideditor.ScriptEditor(
self,
[[i.argv[0]] for i in self.scripts],
[[i.command] for i in self.scripts],
self.edit_scripts
)
)

7
libmproxy/flow.py
View File

@ -494,8 +494,11 @@ class FlowMaster(controller.Master):
def unload_scripts(self):
for s in self.scripts[:]:
s.unload()
self.scripts.remove(s)
self.unload_script(s)
def unload_script(self, script):
script.unload()
self.scripts.remove(script)
def load_script(self, command):
"""

2
libmproxy/script.py
View File

@ -55,6 +55,8 @@ class Script:
@classmethod
def parse_command(klass, command):
if not command or not command.strip():
raise ScriptError("Empty script command.")
if os.name == "nt": # Windows: escape all backslashes in the path.
backslashes = shlex.split(command, posix=False)[0].count("\\")
command = command.replace("\\", "\\\\", backslashes)