mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-27 02:24:18 +00:00
Finalize porting built-in web app to Flask.
This commit is contained in:
parent
f88e899274
commit
cd8fba1d70
110
libpathod/app.py
110
libpathod/app.py
@ -1,6 +1,6 @@
|
||||
import logging
|
||||
from flask import Flask, jsonify, render_template
|
||||
import version
|
||||
import logging, pprint, cStringIO
|
||||
from flask import Flask, jsonify, render_template, request
|
||||
import version, rparse
|
||||
|
||||
logging.basicConfig(level="DEBUG")
|
||||
app = Flask(__name__)
|
||||
@ -29,75 +29,47 @@ def api_clear_log():
|
||||
@app.route('/')
|
||||
@app.route('/index.html')
|
||||
def index():
|
||||
return render_template("index.html", name="index", section="main")
|
||||
return render_template("index.html", section="main")
|
||||
|
||||
|
||||
@app.route('/help')
|
||||
def help():
|
||||
return render_template("help.html", section="help")
|
||||
|
||||
|
||||
@app.route('/log')
|
||||
def log():
|
||||
return render_template("log.html", section="log", log=app.config["pathod"].get_log())
|
||||
|
||||
|
||||
"""
|
||||
class _Page(tornado.web.RequestHandler):
|
||||
def render(self, name, **kwargs):
|
||||
tornado.web.RequestHandler.render(self, name + ".html", **kwargs)
|
||||
@app.route('/log/<int:lid>')
|
||||
def onelog(lid):
|
||||
l = pprint.pformat(app.config["pathod"].log_by_id(int(lid)))
|
||||
return render_template("onelog.html", section="log", alog=l, lid=lid)
|
||||
|
||||
|
||||
class Index(_Page):
|
||||
name = "index"
|
||||
section = "main"
|
||||
def get(self):
|
||||
self.render(self.name, section=self.section, spec="")
|
||||
|
||||
|
||||
class Preview(_Page):
|
||||
name = "preview"
|
||||
section = "main"
|
||||
SANITY = 1024*1024
|
||||
def get(self):
|
||||
spec = self.get_argument("spec", None)
|
||||
args = dict(
|
||||
spec = spec,
|
||||
section = self.section,
|
||||
syntaxerror = None,
|
||||
error = None
|
||||
)
|
||||
try:
|
||||
r = rparse.parse(self.application.settings, spec)
|
||||
except rparse.ParseException, v:
|
||||
args["syntaxerror"] = str(v)
|
||||
args["marked"] = v.marked()
|
||||
return self.render(self.name, **args)
|
||||
if r.length() > self.SANITY:
|
||||
error = "Refusing to preview a response of %s bytes. This is for your own good."%r.length()
|
||||
args["error"] = error
|
||||
else:
|
||||
d = utils.DummyRequest()
|
||||
r.serve(d)
|
||||
args["output"] = d.getvalue()
|
||||
self.render(self.name, **args)
|
||||
|
||||
|
||||
class Help(_Page):
|
||||
name = "help"
|
||||
section = "help"
|
||||
def get(self):
|
||||
self.render(self.name, section=self.section)
|
||||
|
||||
|
||||
class Log(_Page):
|
||||
name = "log"
|
||||
section = "log"
|
||||
def get(self):
|
||||
self.render(self.name, section=self.section, log=self.application.log)
|
||||
|
||||
|
||||
class OneLog(_Page):
|
||||
name = "onelog"
|
||||
section = "log"
|
||||
def get(self, lid):
|
||||
l = pprint.pformat(self.application.log_by_id(int(lid)))
|
||||
self.render(self.name, section=self.section, alog=l, lid=lid)
|
||||
|
||||
|
||||
class ClearLog(_Page):
|
||||
def post(self):
|
||||
self.application.clear_logs()
|
||||
self.redirect("/log")
|
||||
"""
|
||||
SANITY = 1024*1024
|
||||
@app.route('/preview')
|
||||
def preview():
|
||||
spec = request.args["spec"]
|
||||
args = dict(
|
||||
spec = spec,
|
||||
section = "main",
|
||||
syntaxerror = None,
|
||||
error = None
|
||||
)
|
||||
try:
|
||||
r = rparse.parse(app.config["pathod"].request_settings, spec)
|
||||
except rparse.ParseException, v:
|
||||
args["syntaxerror"] = str(v)
|
||||
args["marked"] = v.marked()
|
||||
return render_template("preview.html", **args)
|
||||
if r.length() > SANITY:
|
||||
error = "Refusing to preview a response of %s bytes. This is for your own good."%r.length()
|
||||
args["error"] = error
|
||||
else:
|
||||
s = cStringIO.StringIO()
|
||||
r.serve(s)
|
||||
args["output"] = s.getvalue()
|
||||
return render_template("preview.html", **args)
|
||||
|
@ -35,6 +35,17 @@ class PathodHandler(tcp.BaseHandler):
|
||||
ret = presp.serve(self.wfile)
|
||||
if ret["disconnect"]:
|
||||
self.close()
|
||||
|
||||
ret["request"] = dict(
|
||||
path = path,
|
||||
method = method,
|
||||
headers = headers.lst,
|
||||
#remote_address = self.request.connection.address,
|
||||
#full_url = self.request.full_url(),
|
||||
#query = self.request.query,
|
||||
httpversion = httpversion,
|
||||
#uri = self.request.uri,
|
||||
)
|
||||
self.server.add_log(ret)
|
||||
else:
|
||||
cc = wsgi.ClientConn(self.client_address)
|
||||
@ -60,6 +71,10 @@ class Pathod(tcp.TCPServer):
|
||||
self.log = []
|
||||
self.logid = 0
|
||||
|
||||
@property
|
||||
def request_settings(self):
|
||||
return {}
|
||||
|
||||
def handle_connection(self, request, client_address):
|
||||
PathodHandler(request, client_address, self)
|
||||
|
||||
|
@ -1,4 +1,4 @@
|
||||
{% extends frame.html %}
|
||||
{% extends "frame.html" %}
|
||||
{% block body %}
|
||||
<!-- Text below generated with "markdown2 README.mkd" -->
|
||||
|
||||
@ -251,4 +251,4 @@ ascii
|
||||
bytes
|
||||
</code></pre>
|
||||
|
||||
{% end %}
|
||||
{% endblock %}
|
||||
|
@ -1,4 +1,4 @@
|
||||
{% extends frame.html %}
|
||||
{% extends "frame.html" %}
|
||||
{% block body %}
|
||||
<form style="float: right" method="POST" action="/log/clear">
|
||||
<button type="submit" class="btn">clear</button>
|
||||
@ -10,17 +10,19 @@
|
||||
<thead>
|
||||
<tr>
|
||||
<th>id</th>
|
||||
<th>url</th>
|
||||
<th>method</th>
|
||||
<th>path</th>
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody>
|
||||
{% for i in log %}
|
||||
<tr>
|
||||
<td>{{ i["id"] }}</td>
|
||||
<td><a href="/log/{{ i["id"] }}">{{ i["request"]["full_url"] }}</a></td>
|
||||
<td>{{ i["request"]["method"] }}</td>
|
||||
<td><a href="/log/{{ i["id"] }}">{{ i["request"]["path"] }}</a></td>
|
||||
</tr>
|
||||
{% end %}
|
||||
{% endfor %}
|
||||
</tbody>
|
||||
</table>
|
||||
{% end %}
|
||||
{% endblock %}
|
||||
|
||||
|
@ -1,4 +1,4 @@
|
||||
{% extends frame.html %}
|
||||
{% extends "frame.html" %}
|
||||
{% block body %}
|
||||
<h2> Log entry {{ lid }} </h2>
|
||||
<hr>
|
||||
@ -6,5 +6,5 @@
|
||||
{{ alog }}
|
||||
</pre>
|
||||
|
||||
{% end %}
|
||||
{% endblock %}
|
||||
|
||||
|
@ -1,4 +1,4 @@
|
||||
{% extends frame.html %}
|
||||
{% extends "frame.html" %}
|
||||
{% block body %}
|
||||
|
||||
<h1>Preview</h1>
|
||||
@ -10,7 +10,6 @@
|
||||
<h2> Error </h2>
|
||||
<p style="color: #ff0000">{{ error }}</p>
|
||||
{% else %}
|
||||
|
||||
<h2>Spec:</h2>
|
||||
|
||||
<pre>{{ spec }}</pre>
|
||||
@ -19,6 +18,6 @@
|
||||
|
||||
<pre>{{ output }}</pre>
|
||||
|
||||
{% end %}
|
||||
{% include previewform.html %}
|
||||
{% end %}
|
||||
{% endif %}
|
||||
{% include "previewform.html" %}
|
||||
{% endblock %}
|
||||
|
Loading…
Reference in New Issue
Block a user