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
|
"""Middleware that can dynamically change the stack based on status or headers
Used in the authentication middleware to intercept 401 responses and put the
appropriate middleware into the stack to prompt the user to sign in.
"""
from paste.auth import multi
import logging
log = logging.getLogger('authkit.authenticate.multi')
class MultiHandler(multi.MultiHandler):
def __init__(self, application):
multi.MultiHandler.__init__(self, application)
self.checker = []
def add_checker(self, name, checker):
self.checker.append((checker,self.binding[name]))
def __call__(self, environ, start_response):
status_ = []
headers_ = []
exc_info_ = []
def app(environ, start_response):
def find(status, headers, exc_info=None):
status_.append(status)
headers_.append(headers)
exc_info_.append(exc_info)
log.debug("Status: %r, Headers: %r", status, headers)
return self.default(environ, find)
def logging_start_response(status, headers, exc_info=None):
log.debug("Matched binding returns status: %r, headers: %r, "
"exc_info: %r", status, headers, exc_info)
return start_response(status, headers, exc_info)
def check():
for (checker, binding) in self.predicate:
if checker(environ):
log.debug(
"MultMiddleware self.predicate check() returning %r",
binding)
environ['authkit.multi'] = True
return binding(environ, logging_start_response)
for (checker, binding) in self.checker:
if not len(status_):
raise Exception('No status was returned by the applicaiton')
if not len(headers_):
raise Exception('No headers were returned by the '
'application')
if checker(environ, status_[0], headers_ and headers_[0] or []):
log.debug(
"MultiMiddleware self.checker check() returning %r",
binding
)
environ['authkit.multi'] = True
return binding(environ, logging_start_response)
return None
app_iter = app(environ, start_response)
if not status_:
raise Exception('WSGI start_response was not called before a result'
' was returned')
result = check()
if result is None:
log.debug("Multi: No binding was found for the check")
start_response(status_[0],
headers_ and headers_[0] or [],
exc_info_[0])
return app_iter
# Close app_iter if necessary
if hasattr(app_iter, 'close'):
app_iter.close()
return result
def status_checker(environ, status, headers):
"""
Used by AuthKit to intercept statuses specified in the config file
option ``authkit.intercept``.
"""
log.debug(
"Status checker recieved status %r, headers %r, intecept %r",
status,
headers,
environ['authkit.intercept']
)
if str(status[:3]) in environ['authkit.intercept']:
log.debug("Status checker returns True")
return True
log.debug("Status checker returns False")
return False
class AuthSwitcher:
def __init__(self):
pass
def __call__(self, environ, status, headers):
if status_checker(environ, status, headers):
return self.switch(environ, status, headers)
return False
def switch(self, environ, status, headers):
return False
|
