diff options
Diffstat (limited to 'libmproxy/script.py')
| -rw-r--r-- | libmproxy/script.py | 17 |
1 files changed, 9 insertions, 8 deletions
diff --git a/libmproxy/script.py b/libmproxy/script.py index 896f5245..464ac4aa 100644 --- a/libmproxy/script.py +++ b/libmproxy/script.py @@ -6,7 +6,6 @@ import shlex import sys from watchdog.observers import Observer from watchdog.events import PatternMatchingEventHandler, FileModifiedEvent -from .console import signals class ScriptError(Exception): @@ -71,6 +70,7 @@ class Script: self.ctx = ScriptContext(master) self.ns = None self.load() + observe_scripts(master, self.args[0]) @classmethod def parse_command(cls, command): @@ -199,18 +199,19 @@ def concurrent(fn): class ScriptModified(PatternMatchingEventHandler): - def __init__(self, FlowMaster): - self.FlowMaster = FlowMaster + def __init__(self, flow_master): + self.flow_master = flow_master PatternMatchingEventHandler.__init__(self, ignore_directories=True, patterns=["*.py"]) - + self.context = ScriptContext(self.flow_master) + def on_modified(self, event=FileModifiedEvent): - self.FlowMaster.reload_scripts() - signals.status_message.send(message="script: <{0}> reloaded.".format(event.src_path)) + self.flow_master.reload_scripts() + self.context.log("script: <{}> reloaded.".format(event.src_path)) -def ObserveScripts(FlowMaster, path): +def observe_scripts(flow_master, path): script_dir = os.path.dirname(path) - event_handler = ScriptModified(FlowMaster) + event_handler = ScriptModified(flow_master) observer = Observer() observer.schedule(event_handler, script_dir) observer.start() |