Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 00:01:36 +00:00
Code Issues Projects Releases Wiki Activity

closes #1828 script reloads on py file changes

Browse Source
This commit is contained in:
Ujjwal Verma 2017-02-10 20:53:28 +05:30
parent d4264cb719
commit 809207195d
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mitmproxy/addons/script.py
View File

@ -110,11 +110,16 @@ class ReloadHandler(watchdog.events.FileSystemEventHandler):
self.callback = callback self.callback = callback
def filter(self, event): def filter(self, event):
"""
Returns True only when .py file is changed
"""
if event.is_directory: if event.is_directory:
return False return False
if os.path.basename(event.src_path).startswith("."): if os.path.basename(event.src_path).startswith("."):
return False return False
return True if event.src_path.endswith(".py"):
return True
return False
def on_modified(self, event): def on_modified(self, event):
if self.filter(event): if self.filter(event):

7
test/mitmproxy/addons/test_script.py
View File

@ -44,14 +44,19 @@ def test_reloadhandler():
rh = script.ReloadHandler(Called()) rh = script.ReloadHandler(Called())
assert not rh.filter(watchdog.events.DirCreatedEvent("path")) assert not rh.filter(watchdog.events.DirCreatedEvent("path"))
assert not rh.filter(watchdog.events.FileModifiedEvent("/foo/.bar")) assert not rh.filter(watchdog.events.FileModifiedEvent("/foo/.bar"))
assert rh.filter(watchdog.events.FileModifiedEvent("/foo/bar")) assert not rh.filter(watchdog.events.FileModifiedEvent("/foo/bar"))
assert rh.filter(watchdog.events.FileModifiedEvent("/foo/bar.py"))
assert not rh.callback.called assert not rh.callback.called
rh.on_modified(watchdog.events.FileModifiedEvent("/foo/bar")) rh.on_modified(watchdog.events.FileModifiedEvent("/foo/bar"))
assert not rh.callback.called
rh.on_modified(watchdog.events.FileModifiedEvent("/foo/bar.py"))
assert rh.callback.called assert rh.callback.called
rh.callback.called = False rh.callback.called = False
rh.on_created(watchdog.events.FileCreatedEvent("foo")) rh.on_created(watchdog.events.FileCreatedEvent("foo"))
assert not rh.callback.called
rh.on_created(watchdog.events.FileCreatedEvent("foo.py"))
assert rh.callback.called assert rh.callback.called