57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from werkzeug import BaseRequest, BaseResponse, ETagResponseMixin, escape, run_simple
|
|
from werkzeug.exceptions import HTTPException
|
|
from werkzeug.routing import RequestRedirect
|
|
from jinja import Environment, FileSystemLoader, MemcachedFileSystemLoader
|
|
from jinja.exceptions import TemplateNotFound
|
|
import os
|
|
import sha
|
|
from time import time
|
|
from random import randint
|
|
|
|
class Request(BaseRequest):
|
|
"""Useful subclass of the default request that knows how to build urls."""
|
|
|
|
def __init__(self, environ):
|
|
BaseRequest.__init__(self, environ)
|
|
|
|
|
|
class Response(BaseResponse, ETagResponseMixin):
|
|
"""Subclass of base response that has a default mimetype of text/html."""
|
|
default_mimetype = 'text/html'
|
|
|
|
|
|
# setup jinja
|
|
try:
|
|
env = Environment(loader=MemcachedFileSystemLoader('pages', memcache_host=['127.0.0.1:11211']))
|
|
except RuntimeError:
|
|
env = Environment(loader=FileSystemLoader('pages', use_memcache=True))
|
|
|
|
def app(environ, start_response):
|
|
"""The WSGI application that connects all together."""
|
|
req = Request(environ)
|
|
path = req.path[1:].lower()
|
|
if path == '':
|
|
path = 'index'
|
|
mime_type='text/html'
|
|
try:
|
|
try:
|
|
path.index('.')
|
|
if path.split('.')[-1].isdigit() and not path.split('.')[-1].isalpha():
|
|
raise ValueError()
|
|
tmpl = env.get_template(path)
|
|
if path[-3:] == 'txt':
|
|
mime_type='text/plain'
|
|
if path[-3:] == 'xml':
|
|
mime_type='text/xml'
|
|
except ValueError:
|
|
tmpl = env.get_template(path + '.html')
|
|
except TemplateNotFound:
|
|
tmpl = env.get_template('not_found.html')
|
|
resp = Response(tmpl.render(static=False), mimetype=mime_type)
|
|
resp.add_etag()
|
|
resp.make_conditional(req)
|
|
return resp(environ, start_response)
|
|
|
|
if __name__ == '__main__':
|
|
run_simple('localhost', 5009, app)
|