diff options
Diffstat (limited to 'docs/wiki-example-code/example.py')
| -rw-r--r-- | docs/wiki-example-code/example.py | 115 |
1 files changed, 61 insertions, 54 deletions
diff --git a/docs/wiki-example-code/example.py b/docs/wiki-example-code/example.py index 2d452c7..f5edd9f 100644 --- a/docs/wiki-example-code/example.py +++ b/docs/wiki-example-code/example.py @@ -4,7 +4,8 @@ from webob import Request, Response from webob import exc from tempita import HTMLTemplate -VIEW_TEMPLATE = HTMLTemplate("""\ +VIEW_TEMPLATE = HTMLTemplate( + """\ <html> <head> <title>{{page.title}}</title> @@ -21,9 +22,11 @@ VIEW_TEMPLATE = HTMLTemplate("""\ <a href="{{req.url}}?action=edit">Edit</a> </body> </html> -""") +""" +) -EDIT_TEMPLATE = HTMLTemplate("""\ +EDIT_TEMPLATE = HTMLTemplate( + """\ <html> <head> <title>Edit: {{page.title}}</title> @@ -47,7 +50,9 @@ EDIT_TEMPLATE = HTMLTemplate("""\ <a href="{{req.path_url}}">Cancel</a> </form> </body></html> -""") +""" +) + class WikiApp(object): @@ -59,67 +64,63 @@ class WikiApp(object): def __call__(self, environ, start_response): req = Request(environ) - action = req.params.get('action', 'view') + action = req.params.get("action", "view") page = self.get_page(req.path_info) try: try: - meth = getattr(self, 'action_%s_%s' % (action, req.method)) + meth = getattr(self, "action_%s_%s" % (action, req.method)) except AttributeError: - raise exc.HTTPBadRequest('No such action %r' % action) + raise exc.HTTPBadRequest("No such action %r" % action) resp = meth(req, page) except exc.HTTPException, e: resp = e return resp(environ, start_response) def get_page(self, path): - path = path.lstrip('/') + path = path.lstrip("/") if not path: - path = 'index' + path = "index" path = os.path.join(self.storage_dir, path) path = os.path.normpath(path) - if path.endswith('/'): - path += 'index' + if path.endswith("/"): + path += "index" if not path.startswith(self.storage_dir): raise exc.HTTPBadRequest("Bad path") - path += '.html' + path += ".html" return Page(path) def action_view_GET(self, req, page): if not page.exists: - return exc.HTTPTemporaryRedirect( - location=req.url + '?action=edit') - if req.cookies.get('message'): - message = req.cookies['message'] + return exc.HTTPTemporaryRedirect(location=req.url + "?action=edit") + if req.cookies.get("message"): + message = req.cookies["message"] else: message = None - text = self.view_template.substitute( - page=page, req=req, message=message) + text = self.view_template.substitute(page=page, req=req, message=message) resp = Response(text) if message: - resp.delete_cookie('message') + resp.delete_cookie("message") else: resp.last_modified = page.mtime resp.conditional_response = True return resp def action_view_POST(self, req, page): - submit_mtime = int(req.params.get('mtime') or '0') or None + submit_mtime = int(req.params.get("mtime") or "0") or None if page.mtime != submit_mtime: return exc.HTTPPreconditionFailed( - "The page has been updated since you started editing it") - page.set( - title=req.params['title'], - content=req.params['content']) - resp = exc.HTTPSeeOther( - location=req.path_url) - resp.set_cookie('message', 'Page updated') + "The page has been updated since you started editing it" + ) + page.set(title=req.params["title"], content=req.params["content"]) + resp = exc.HTTPSeeOther(location=req.path_url) + resp.set_cookie("message", "Page updated") return resp def action_edit_GET(self, req, page): - text = self.edit_template.substitute( - page=page, req=req) + text = self.edit_template.substitute(page=page, req=req) return Response(text) + class Page(object): def __init__(self, filename): self.filename = filename @@ -133,15 +134,15 @@ class Page(object): if not self.exists: # we need to guess the title basename = os.path.splitext(os.path.basename(self.filename))[0] - basename = re.sub(r'[_-]', ' ', basename) + basename = re.sub(r"[_-]", " ", basename) return basename.capitalize() content = self.full_content - match = re.search(r'<title>(.*?)</title>', content, re.I|re.S) + match = re.search(r"<title>(.*?)</title>", content, re.I | re.S) return match.group(1) @property def full_content(self): - f = open(self.filename, 'rb') + f = open(self.filename, "rb") try: return f.read() finally: @@ -150,9 +151,9 @@ class Page(object): @property def content(self): if not self.exists: - return '' + return "" content = self.full_content - match = re.search(r'<body[^>]*>(.*?)</body>', content, re.I|re.S) + match = re.search(r"<body[^>]*>(.*?)</body>", content, re.I | re.S) return match.group(1) @property @@ -166,35 +167,41 @@ class Page(object): dir = os.path.dirname(self.filename) if not os.path.exists(dir): os.makedirs(dir) - new_content = """<html><head><title>%s</title></head><body>%s</body></html>""" % ( - title, content) - f = open(self.filename, 'wb') + new_content = ( + """<html><head><title>%s</title></head><body>%s</body></html>""" + % (title, content) + ) + f = open(self.filename, "wb") f.write(new_content) f.close() -if __name__ == '__main__': + +if __name__ == "__main__": import optparse - parser = optparse.OptionParser( - usage='%prog --port=PORT' - ) + + parser = optparse.OptionParser(usage="%prog --port=PORT") parser.add_option( - '-p', '--port', - default='8080', - dest='port', - type='int', - help='Port to serve on (default 8080)') + "-p", + "--port", + default="8080", + dest="port", + type="int", + help="Port to serve on (default 8080)", + ) parser.add_option( - '--wiki-data', - default='./wiki', - dest='wiki_data', - help='Place to put wiki data into (default ./wiki/)') + "--wiki-data", + default="./wiki", + dest="wiki_data", + help="Place to put wiki data into (default ./wiki/)", + ) options, args = parser.parse_args() - print 'Writing wiki pages to %s' % options.wiki_data + print("Writing wiki pages to %s" % options.wiki_data) app = WikiApp(options.wiki_data) from wsgiref.simple_server import make_server - httpd = make_server('localhost', options.port, app) - print 'Serving on http://localhost:%s' % options.port + + httpd = make_server("localhost", options.port, app) + print("Serving on http://localhost:%s" % options.port) try: httpd.serve_forever() except KeyboardInterrupt: - print '^C' + print("^C") |