88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
import logging
|
|
import xtraceback
|
|
import pprint
|
|
import sys
|
|
|
|
from cgi import escape
|
|
|
|
from mandaye.http import HTTPResponse, HTTPHeader
|
|
from mandaye.config import debug
|
|
|
|
from template import serve_template
|
|
|
|
|
|
def _get_html_traceback(tb_str, path, env=None):
|
|
msg = "<h2>%s</h2><h3>Exception</h3><pre>%s</pre>" % \
|
|
(path, escape(tb_str))
|
|
if env:
|
|
env = pprint.pformat(env)
|
|
msg += '<h3>Environment</h3><pre>%s</pre>' % escape(env)
|
|
return msg
|
|
|
|
def _302(location, cookies=None):
|
|
headers = HTTPHeader({'Location': [location],
|
|
'Content-Type': ['text/html']})
|
|
msg = "Reload the page to get source for: %s" % location
|
|
return HTTPResponse(302, 'Found', headers, msg, cookies)
|
|
|
|
def _401(msg):
|
|
title='Unauthorized'
|
|
logging.warning('401 %s' % msg)
|
|
headers = HTTPHeader({'Content-Type': ['text/html']})
|
|
return HTTPResponse(401, 'Unauthorized', headers,
|
|
serve_template("response.html", title=title, body=msg))
|
|
|
|
def _404(path):
|
|
logging.warning("404 %s not found" % path)
|
|
title = "Not Found"
|
|
body = "The requested URL %s was not found on this server." % path
|
|
headers = HTTPHeader({'Content-Type': ['text/html']})
|
|
return HTTPResponse(404, 'Not found', headers,
|
|
serve_template("response.html", title=title, body=body))
|
|
|
|
def _500(path, msg, exception=None, env=None):
|
|
""" Return a 500 error
|
|
path: the path of this error
|
|
exception: this exception instance if any """
|
|
if exception:
|
|
tb_str = str(xtraceback.XTraceback(*sys.exc_info(), color=False))
|
|
logging.exception("500 %s: %s, %s, env: %r" % (path, msg, tb_str, env))
|
|
else:
|
|
logging.error("500 %s: %s, env: %r" % (path, msg, env))
|
|
title = 'Internal Server Error'
|
|
headers = HTTPHeader({'Content-Type': ['text/html']})
|
|
if debug:
|
|
if exception:
|
|
msg += _get_html_traceback(tb_str, path, env=env)
|
|
else:
|
|
msg=msg
|
|
else:
|
|
msg = "The server encountered an internal error or misconfiguration \
|
|
and was unable to complete your request. Please contact the server \
|
|
administrator and inform them of the time the error occurred."
|
|
return HTTPResponse(500, 'Internal Server Error', headers,
|
|
serve_template("response.html", title=title, body=msg))
|
|
|
|
def _502(path, target_hostname, exception=None):
|
|
""" Return a 502 error
|
|
path: the path of this error
|
|
exception: this exception instance if any """
|
|
logging.critical("[502] error while reading %s:\n" % path)
|
|
title = 'Bad Gateway'
|
|
headers = HTTPHeader({'Content-Type': ['text/html']})
|
|
msg = "The server can't join the target server or misconfiguration \
|
|
and was unable to complete your request. Please contact the server \
|
|
administrator and inform them of the time the error occurred."
|
|
if debug and exception:
|
|
traceback.print_exc()
|
|
msg = "<h2>Target hostname: %s</h2>" % target_hostname
|
|
msg += _get_html_traceback(exception, path)
|
|
return HTTPResponse(502, 'Bad Gateway', headers,
|
|
serve_template("response.html", title=title, body=msg))
|
|
|
|
def template_response(templatename, values):
|
|
headers = HTTPHeader({'Content-Type': ['text/html']})
|
|
msg = serve_template(templatename, **values)
|
|
return HTTPResponse(200, 'OK', headers, msg)
|
|
|