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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
# (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 <http://python.org/doc/current/lib/module-cgitb.html>`_
for more.
"""
import cgitb
import six
from six.moves import cStringIO as StringIO
import sys
from paste.util import converters
class NoDefault(object):
pass
class CgitbMiddleware(object):
def __init__(self, app,
global_conf=None,
display=NoDefault,
logdir=None,
context=5,
format="html"):
self.app = app
if global_conf is None:
global_conf = {}
if display is NoDefault:
display = global_conf.get('debug')
if isinstance(display, six.string_types):
display = converters.asbool(display)
self.display = 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)
if six.PY3:
response = response.encode('utf8')
return [response]
def catching_iter(self, app_iter, environ):
if not app_iter:
return
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)
if six.PY3:
response = response.encode('utf8')
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()
def make_cgitb_middleware(app, global_conf,
display=NoDefault,
logdir=None,
context=5,
format='html'):
"""
Wraps the application in the ``cgitb`` (standard library)
error catcher.
display:
If true (or debug is set in the global configuration)
then the traceback will be displayed in the browser
logdir:
Writes logs of all errors in that directory
context:
Number of lines of context to show around each line of
source code
"""
from paste.deploy.converters import asbool
if display is not NoDefault:
display = asbool(display)
if 'debug' in global_conf:
global_conf['debug'] = asbool(global_conf['debug'])
return CgitbMiddleware(
app, global_conf=global_conf,
display=display,
logdir=logdir,
context=context,
format=format)
|
