aboutsummaryrefslogtreecommitdiffstats
path: root/libmproxy/flow.py
diff options
context:
space:
mode:
authorAldo Cortesi <aldo@nullcube.com>2011-01-31 11:44:52 +1300
committerAldo Cortesi <aldo@nullcube.com>2011-01-31 11:44:52 +1300
commit152b97fa0bc9245ccb17353494d5c940b5356d28 (patch)
tree653788766cd21a8b299ce8e59104ee70458f298b /libmproxy/flow.py
parent1619d164ffc4c82da80f00598f3358fe8ca64313 (diff)
downloadmitmproxy-152b97fa0bc9245ccb17353494d5c940b5356d28.tar.gz
mitmproxy-152b97fa0bc9245ccb17353494d5c940b5356d28.tar.bz2
mitmproxy-152b97fa0bc9245ccb17353494d5c940b5356d28.zip
Restructure to make subclassing Flow unnecessary.
Diffstat (limited to 'libmproxy/flow.py')
-rw-r--r--libmproxy/flow.py10
1 files changed, 8 insertions, 2 deletions
diff --git a/libmproxy/flow.py b/libmproxy/flow.py
index 7f4394b1..a014f8cb 100644
--- a/libmproxy/flow.py
+++ b/libmproxy/flow.py
@@ -33,6 +33,12 @@ class Flow:
self.intercepting = False
self._backup = None
+ def run_script(self):
+ """
+ Run a script on a flow, returning the modified flow.
+ """
+ pass
+
def dump(self):
data = dict(
flows = [self.get_state()]
@@ -160,9 +166,9 @@ class State:
)
return bson.dumps(data)
- def load_flows(self, js, klass):
+ def load_flows(self, js):
data = bson.loads(js)
- data = [klass.from_state(i) for i in data["flows"]]
+ data = [Flow.from_state(i) for i in data["flows"]]
self.flow_list.extend(data)
def set_limit(self, limit):