diff options
Diffstat (limited to 'libpathod/app.py')
| -rw-r--r-- | libpathod/app.py | 24 |
1 files changed, 23 insertions, 1 deletions
diff --git a/libpathod/app.py b/libpathod/app.py index 685ac22e..0274853b 100644 --- a/libpathod/app.py +++ b/libpathod/app.py @@ -6,6 +6,7 @@ class _Page(tornado.web.RequestHandler): def render(self, name, **kwargs): tornado.web.RequestHandler.render(self, name + ".html", **kwargs) + class Index(_Page): name = "index" section = "main" @@ -16,8 +17,29 @@ class Index(_Page): class Preview(_Page): name = "preview" section = "main" + SANITY = 1024*1024 def get(self): - self.render(self.name, section=self.section) + 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): |