1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
# (c) 2005 Ian Bicking and contributors; written for Paste (http://pythonpaste.org)
# Licensed under the MIT license: http://www.opensource.org/licenses/mit-license.php
"""
WSGI middleware
Captures any exceptions and prints a pretty report. See the cgitb
documentation for more:
http://python.org/doc/current/lib/module-cgitb.html
"""
import cgitb
from cStringIO import StringIO
import sys
from paste.deploy import converters
class NoDefault:
pass
class CgitbMiddleware(object):
def __init__(self, app,
global_conf,
display=NoDefault,
logdir=None,
context=5,
format="html"):
self.app = app
if display is NoDefault:
display = global_conf.get('debug')
self.display = converters.asbool(display)
self.logdir = logdir
self.context = int(context)
self.format = format
def __call__(self, environ, start_response):
try:
app_iter = self.app(environ, start_response)
return self.catching_iter(app_iter, environ)
except:
exc_info = sys.exc_info()
start_response('500 Internal Server Error',
[('content-type', 'text/html')],
exc_info)
response = self.exception_handler(exc_info, environ)
return [response]
def catching_iter(self, app_iter, environ):
if not app_iter:
raise StopIteration
error_on_close = False
try:
for v in app_iter:
yield v
if hasattr(app_iter, 'close'):
error_on_close = True
app_iter.close()
except:
response = self.exception_handler(sys.exc_info(), environ)
if not error_on_close and hasattr(app_iter, 'close'):
try:
app_iter.close()
except:
close_response = self.exception_handler(
sys.exc_info(), environ)
response += (
'<hr noshade>Error in .close():<br>%s'
% close_response)
yield response
def exception_handler(self, exc_info, environ):
dummy_file = StringIO()
hook = cgitb.Hook(file=dummy_file,
display=self.display,
logdir=self.logdir,
context=self.context,
format=self.format)
hook(*exc_info)
return dummy_file.getvalue()
|