From d32d6bc5e3cafaa51556da0690f6835cd1b319dc Mon Sep 17 00:00:00 2001
From: Aldo Cortesi <aldo@nullcube.com>
Date: Sun, 19 Feb 2012 00:17:47 +1300
Subject: Add "p" key binding to connection list view to copy a flow.

---
 libmproxy/flow.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'libmproxy/flow.py')

diff --git a/libmproxy/flow.py b/libmproxy/flow.py
index 8ce25c5c..db4cde5e 100644
--- a/libmproxy/flow.py
+++ b/libmproxy/flow.py
@@ -856,8 +856,10 @@ class Flow:
         f = Flow(rc)
         if self.response:
             f.response = self.response.copy()
+            f.response.request = rc
         if self.error:
             f.error = self.error.copy()
+            f.error.request = rc
         return f
 
     @classmethod
@@ -1006,6 +1008,9 @@ class State(object):
     def flow_count(self):
         return len(self._flow_map)
 
+    def index(self, f):
+        return self._flow_list.index(f)
+
     def active_flow_count(self):
         c = 0
         for i in self._flow_list:
@@ -1249,6 +1254,9 @@ class FlowMaster(controller.Master):
         return self.load_flow(f.copy())
 
     def load_flow(self, f):
+        """
+            Loads a flow, and returns a new flow object.
+        """
         if f.request:
             fr = self.handle_request(f.request)
         if f.response:
-- 
cgit v1.2.3