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
|
from itertools import izip, islice
from inspect import getargspec
import cherrypy
from formencode.api import Invalid
from _utils import decorated
__all__ = ["validate", "error_handler", "exception_handler"]
def to_kw(func, args, kw):
argnames, defaults = getargspec(func)[::3]
defaults = defaults or []
kv = zip(islice(argnames, 0, len(argnames) - len(defaults)), args)
kw.update(kv)
return args[len(argnames)-len(defaults):], kw
def from_kw(func, args, kw):
argnames, defaults = getargspec(func)[::3]
defaults = defaults or []
newargs = [kw.pop(name) for name in islice(argnames, 0, len(argnames) - len(defaults)) if name in kw]
newargs.extend(args)
return newargs, kw
def validate(form=None, validators=None):
def validate_wrapper(func):
if callable(form) and not hasattr(form, "validate"):
init_form = lambda self: form(self)
else:
init_form = lambda self: form
def func_wrapper(*args, **kw):
# do not validate a second time if already validated
if hasattr(cherrypy.request, 'validation_state'):
return func(*args, **kw)
form = init_form(args and args[0] or kw["self"])
args, kw = to_kw(func, args, kw)
errors = {}
if form:
value = kw.copy()
value.pop('self', None)
try:
kw.update(form.validate(value, None))
except Invalid, e:
errors = e.unpack_errors()
cherrypy.request.validation_exception = e
cherrypy.request.validation_value = value
cherrypy.request.validated_form = form
if validators:
if isinstance(validators, dict):
for field, validator in validators.iteritems():
try:
kw[field] = validator.to_python(
kw.get(field, None), None)
except Invalid, error:
errors[field] = error
else:
try:
value = kw.copy()
kw.update(validators.to_python(value, None))
except Invalid, e:
errors = e.unpack_errors()
cherrypy.request.validation_exception = e
cherrypy.request.validation_value = value
cherrypy.request.validation_errors = errors
cherrypy.request.input_values = kw.copy()
cherrypy.request.validation_state = True
args, kw = from_kw(func, args, kw)
return func(*args, **kw)
return decorated(func_wrapper, func)
return validate_wrapper
def error_handler(handler):
def wrapper(func):
def func_wrapper(*args, **kw):
tg_errors = getattr(cherrypy.request, 'validation_errors', None)
if tg_errors:
kw['tg_errors'] = tg_errors
return handler(*args, **kw)
return func(*args, **kw)
return decorated(func_wrapper, func)
return wrapper
def exception_handler(*args, **kw):
return lambda f: f
|