diff options
author | Aldo Cortesi <aldo@nullcube.com> | 2016-06-02 12:53:12 +1200 |
---|---|---|
committer | Aldo Cortesi <aldo@nullcube.com> | 2016-06-02 12:53:12 +1200 |
commit | 31012d782f64727de1d86662139e9ec6618043c5 (patch) | |
tree | 3887c8e11e30a758e82afdabe764a5e5f4f13e8d /pathod | |
parent | 3d9dbd6a4940cc7fe6f65cbb257aaeacd1eab1ee (diff) | |
download | mitmproxy-31012d782f64727de1d86662139e9ec6618043c5.tar.gz mitmproxy-31012d782f64727de1d86662139e9ec6618043c5.tar.bz2 mitmproxy-31012d782f64727de1d86662139e9ec6618043c5.zip |
Consolidate some functions to strutils.bytes_to_escaped_str
Diffstat (limited to 'pathod')
-rw-r--r-- | pathod/app.py | 5 | ||||
-rw-r--r-- | pathod/pathoc.py | 5 | ||||
-rw-r--r-- | pathod/utils.py | 14 |
3 files changed, 6 insertions, 18 deletions
diff --git a/pathod/app.py b/pathod/app.py index 7e9860b9..e3216c58 100644 --- a/pathod/app.py +++ b/pathod/app.py @@ -3,8 +3,9 @@ import pprint import io import copy from flask import Flask, jsonify, render_template, request, abort, make_response -from . import version, language, utils +from . import version, language from netlib.http import user_agents +from netlib import strutils logging.basicConfig(level="DEBUG") EXAMPLE_HOST = "example.com" @@ -166,7 +167,7 @@ def make_app(noapi, debug): settings.websocket_key = EXAMPLE_WEBSOCKET_KEY language.serve(safe, s, settings) - args["output"] = utils.escape_unprintables(s.getvalue()) + args["output"] = strutils.bytes_to_escaped_str(s.getvalue()) return render(template, False, **args) @app.route('/response_preview') diff --git a/pathod/pathoc.py b/pathod/pathoc.py index ebbd952a..2b7d053c 100644 --- a/pathod/pathoc.py +++ b/pathod/pathoc.py @@ -18,10 +18,11 @@ from netlib.exceptions import HttpException, TcpDisconnect, TcpTimeout, TlsExcep NetlibException from netlib.http import http1, http2 -from . import utils, log, language +from . import log, language import logging from netlib.tutils import treq +from netlib import strutils logging.getLogger("hpack").setLevel(logging.WARNING) @@ -427,7 +428,7 @@ class Pathoc(tcp.TCPClient): finally: if resp: lg("<< %s %s: %s bytes" % ( - resp.status_code, xrepr(resp.reason), len(resp.content) + resp.status_code, strutils.bytes_to_escaped_str(resp.reason), len(resp.content) )) if resp.status_code in self.ignorecodes: lg.suppress() diff --git a/pathod/utils.py b/pathod/utils.py index 96b54ab8..3276198a 100644 --- a/pathod/utils.py +++ b/pathod/utils.py @@ -2,8 +2,6 @@ import os import sys import netlib.utils -from netlib import strutils - class MemBool(object): @@ -28,18 +26,6 @@ def parse_anchor_spec(s): return tuple(s.split("=", 1)) -def escape_unprintables(s): - """ - Like inner_repr, but preserves line breaks. - """ - s = s.replace(b"\r\n", b"PATHOD_MARKER_RN") - s = s.replace(b"\n", b"PATHOD_MARKER_N") - s = strutils.bytes_to_escaped_str(s) - s = s.replace("PATHOD_MARKER_RN", "\n") - s = s.replace("PATHOD_MARKER_N", "\n") - return s - - data = netlib.utils.Data(__name__) |