Improved requests highlight
This commit is contained in:
parent
f7e44b6870
commit
6d79787633
1 changed files with 14 additions and 6 deletions
|
@ -19,6 +19,7 @@ from pappyproxy.util import PappyException
|
|||
from pappyproxy.macros import load_macros, macro_from_requests, gen_imacro
|
||||
from pappyproxy.repeater import start_editor
|
||||
from pygments.lexers import get_lexer_for_mimetype
|
||||
from pygments.lexers import HttpLexer
|
||||
from pygments.formatters import TerminalFormatter
|
||||
|
||||
"""
|
||||
|
@ -1124,9 +1125,12 @@ def check_reqid(reqid):
|
|||
|
||||
def view_full_request(request, headers_only=False):
|
||||
if headers_only:
|
||||
print printable_data(request.raw_headers)
|
||||
to_print = printable_data(request.raw_headers)
|
||||
else:
|
||||
print printable_data(request.full_request)
|
||||
to_print = printable_data(request.full_request)
|
||||
to_print = pygments.highlight(to_print, HttpLexer(), TerminalFormatter())
|
||||
|
||||
print to_print
|
||||
|
||||
def view_full_response(response, headers_only=False):
|
||||
def check_type(response, against):
|
||||
|
@ -1135,13 +1139,17 @@ def view_full_response(response, headers_only=False):
|
|||
return False
|
||||
|
||||
if headers_only:
|
||||
print printable_data(response.raw_headers)
|
||||
to_print = printable_data(response.raw_headers)
|
||||
to_print = pygments.highlight(to_print, HttpLexer(), TerminalFormatter())
|
||||
print to_print
|
||||
else:
|
||||
print response.raw_headers,
|
||||
headers = printable_data(response.raw_headers)
|
||||
headers = pygments.highlight(headers, HttpLexer(), TerminalFormatter())
|
||||
print headers
|
||||
to_print = printable_data(response.raw_data)
|
||||
if 'content-type' in response.headers:
|
||||
if 'Content-Type' in response.headers:
|
||||
try:
|
||||
lexer = get_lexer_for_mimetype(response.headers['content-type'].split(';')[0])
|
||||
lexer = get_lexer_for_mimetype(response.headers['Content-Type'].split(';')[0])
|
||||
to_print = pygments.highlight(to_print, lexer, TerminalFormatter())
|
||||
except ClassNotFound:
|
||||
pass
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue