diff options
author | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2017-02-09 11:56:38 +0100 |
---|---|---|
committer | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2017-02-09 16:08:36 +0100 |
commit | 0299bb5b2e4870363ba0c402c6cf15722ca0ee0f (patch) | |
tree | f3f893401505dd57931497645e4015a2e0ec7aae /mitmproxy | |
parent | 5a3976c43e480b3926691e9f394b8200ca7613f0 (diff) | |
download | mitmproxy-0299bb5b2e4870363ba0c402c6cf15722ca0ee0f.tar.gz mitmproxy-0299bb5b2e4870363ba0c402c6cf15722ca0ee0f.tar.bz2 mitmproxy-0299bb5b2e4870363ba0c402c6cf15722ca0ee0f.zip |
eventsequence: coverage++
Diffstat (limited to 'mitmproxy')
-rw-r--r-- | mitmproxy/addons/script.py | 6 | ||||
-rw-r--r-- | mitmproxy/eventsequence.py (renamed from mitmproxy/events.py) | 4 | ||||
-rw-r--r-- | mitmproxy/master.py | 6 | ||||
-rw-r--r-- | mitmproxy/script/concurrent.py | 4 | ||||
-rw-r--r-- | mitmproxy/test/taddons.py | 4 |
5 files changed, 12 insertions, 12 deletions
diff --git a/mitmproxy/addons/script.py b/mitmproxy/addons/script.py index 07a8975a..b3a93571 100644 --- a/mitmproxy/addons/script.py +++ b/mitmproxy/addons/script.py @@ -8,7 +8,7 @@ import types from mitmproxy import exceptions from mitmproxy import ctx -from mitmproxy import events +from mitmproxy import eventsequence import watchdog.events @@ -141,7 +141,7 @@ class Script: self.last_options = None self.should_reload = threading.Event() - for i in events.Events: + for i in eventsequence.Events: if not hasattr(self, i): def mkprox(): evt = i @@ -211,7 +211,7 @@ class ScriptLoader: raise ValueError(str(e)) sc.load_script() for f in flows: - for evt, o in events.event_sequence(f): + for evt, o in eventsequence.iterate(f): sc.run(evt, o) sc.done() return sc diff --git a/mitmproxy/events.py b/mitmproxy/eventsequence.py index 53f236ca..905cb7bc 100644 --- a/mitmproxy/events.py +++ b/mitmproxy/eventsequence.py @@ -37,7 +37,7 @@ Events = frozenset([ ]) -def event_sequence(f): +def iterate(f): if isinstance(f, http.HTTPFlow): if f.request: yield "requestheaders", f @@ -70,4 +70,4 @@ def event_sequence(f): yield "tcp_error", f yield "tcp_end", f else: - raise NotImplementedError + raise ValueError diff --git a/mitmproxy/master.py b/mitmproxy/master.py index ee240eeb..3a3f4399 100644 --- a/mitmproxy/master.py +++ b/mitmproxy/master.py @@ -7,7 +7,7 @@ import sys from mitmproxy import addonmanager from mitmproxy import options from mitmproxy import controller -from mitmproxy import events +from mitmproxy import eventsequence from mitmproxy import exceptions from mitmproxy import connections from mitmproxy import http @@ -91,7 +91,7 @@ class Master: changed = False try: mtype, obj = self.event_queue.get(timeout=timeout) - if mtype not in events.Events: + if mtype not in eventsequence.Events: raise exceptions.ControlException( "Unknown event %s" % repr(mtype) ) @@ -153,7 +153,7 @@ class Master: f.request.port = self.server.config.upstream_server.address.port f.request.scheme = self.server.config.upstream_server.scheme f.reply = controller.DummyReply() - for e, o in events.event_sequence(f): + for e, o in eventsequence.iterate(f): getattr(self, e)(o) def load_flows(self, fr: io.FlowReader) -> int: diff --git a/mitmproxy/script/concurrent.py b/mitmproxy/script/concurrent.py index 2fd7ad8d..366929a5 100644 --- a/mitmproxy/script/concurrent.py +++ b/mitmproxy/script/concurrent.py @@ -3,7 +3,7 @@ This module provides a @concurrent decorator primitive to offload computations from mitmproxy's main master thread. """ -from mitmproxy import events +from mitmproxy import eventsequence from mitmproxy.types import basethread @@ -12,7 +12,7 @@ class ScriptThread(basethread.BaseThread): def concurrent(fn): - if fn.__name__ not in events.Events - {"start", "configure", "tick"}: + if fn.__name__ not in eventsequence.Events - {"start", "configure", "tick"}: raise NotImplementedError( "Concurrent decorator not supported for '%s' method." % fn.__name__ ) diff --git a/mitmproxy/test/taddons.py b/mitmproxy/test/taddons.py index a25b6891..bb8daa02 100644 --- a/mitmproxy/test/taddons.py +++ b/mitmproxy/test/taddons.py @@ -3,7 +3,7 @@ import contextlib import mitmproxy.master import mitmproxy.options from mitmproxy import proxy -from mitmproxy import events +from mitmproxy import eventsequence from mitmproxy import exceptions @@ -57,7 +57,7 @@ class context: is taken (as in flow interception). """ f.reply._state = "handled" - for evt, arg in events.event_sequence(f): + for evt, arg in eventsequence.iterate(f): h = getattr(addon, evt, None) if h: h(arg) |