From 5b7e19a77e12ffac2e699d2dc92f96f90114ee01 Mon Sep 17 00:00:00 2001
From: Maximilian Hils <git@maximilianhils.com>
Date: Mon, 18 Aug 2014 00:55:30 +0200
Subject: fix #327

---
 libmproxy/protocol/http.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'libmproxy/protocol')

diff --git a/libmproxy/protocol/http.py b/libmproxy/protocol/http.py
index 21d2c7ea..b8eca7d8 100644
--- a/libmproxy/protocol/http.py
+++ b/libmproxy/protocol/http.py
@@ -261,6 +261,9 @@ class HTTPRequest(HTTPMessage):
         f._load_state(state)
         return f
 
+    def __repr__(self):
+        return "<HTTPRequest: {0}>".format(self._assemble_first_line(self.form_in)[:-9])
+
     @classmethod
     def from_stream(cls, rfile, include_body=True, body_size_limit=None):
         """
@@ -632,6 +635,14 @@ class HTTPResponse(HTTPMessage):
         f._load_state(state)
         return f
 
+    def __repr__(self):
+        return "<HTTPResponse: {code} {msg} ({contenttype}, {size})>".format(
+            code=self.code,
+            msg=self.msg,
+            contenttype=self.headers.get_first("content-type", "?"),
+            size=utils.pretty_size(len(self.content))
+        )
+
     @classmethod
     def from_stream(cls, rfile, request_method, include_body=True, body_size_limit=None):
         """
@@ -803,6 +814,14 @@ class HTTPFlow(Flow):
         f._load_state(state)
         return f
 
+    def __repr__(self):
+        s = "<HTTPFlow"
+        for a in ("request", "response", "error", "client_conn", "server_conn"):
+            if getattr(self, a, False):
+                s += "\r\n  %s = {flow.%s}" % (a,a)
+        s += ">"
+        return s.format(flow=self)
+
     def copy(self):
         f = super(HTTPFlow, self).copy()
         if self.request:
-- 
cgit v1.2.3