1 | import os, sys |
---|
2 | |
---|
3 | import cherrypy |
---|
4 | from mako.template import Template |
---|
5 | from mako.lookup import TemplateLookup |
---|
6 | import simplejson |
---|
7 | import datetime, decimal |
---|
8 | from StringIO import StringIO |
---|
9 | from invirt.config import structs as config |
---|
10 | from webcommon import State |
---|
11 | |
---|
12 | class MakoHandler(cherrypy.dispatch.LateParamPageHandler): |
---|
13 | """Callable which processes a dictionary, returning the rendered body.""" |
---|
14 | |
---|
15 | def __init__(self, template, next_handler, content_type='text/html; charset=utf-8'): |
---|
16 | self.template = template |
---|
17 | self.next_handler = next_handler |
---|
18 | self.content_type = content_type |
---|
19 | |
---|
20 | def __call__(self): |
---|
21 | env = globals().copy() |
---|
22 | env.update(self.next_handler()) |
---|
23 | cherrypy.response.headers['Content-Type'] = self.content_type |
---|
24 | return self.template.render(**env) |
---|
25 | |
---|
26 | |
---|
27 | class MakoLoader(object): |
---|
28 | |
---|
29 | def __init__(self): |
---|
30 | self.lookups = {} |
---|
31 | |
---|
32 | def get_lookup(self, directories, module_directory=None, |
---|
33 | collection_size=-1, imports=[], **kwargs): |
---|
34 | # Find the appropriate template lookup. |
---|
35 | key = (tuple(directories), module_directory) |
---|
36 | try: |
---|
37 | lookup = self.lookups[key] |
---|
38 | except KeyError: |
---|
39 | lookup = TemplateLookup(directories=directories, |
---|
40 | module_directory=module_directory, |
---|
41 | collection_size=collection_size, |
---|
42 | default_filters=['decode.utf8'], |
---|
43 | input_encoding='utf-8', |
---|
44 | output_encoding='utf-8', |
---|
45 | imports=imports, |
---|
46 | ) |
---|
47 | self.lookups[key] = lookup |
---|
48 | return lookup |
---|
49 | |
---|
50 | def __call__(self, filename, directories, module_directory=None, |
---|
51 | collection_size=-1, content_type='text/html; charset=utf-8', |
---|
52 | imports=[]): |
---|
53 | cherrypy.request.lookup = lookup = self.get_lookup(directories, module_directory, |
---|
54 | collection_size, imports) |
---|
55 | cherrypy.request.template = t = lookup.get_template(filename) |
---|
56 | cherrypy.request.handler = MakoHandler(t, cherrypy.request.handler, content_type) |
---|
57 | |
---|
58 | cherrypy.tools.mako = cherrypy.Tool('on_start_resource', MakoLoader()) |
---|
59 | |
---|
60 | def revertStandardError(): |
---|
61 | """Move stderr to stdout, and return the contents of the old stderr.""" |
---|
62 | errio = sys.stderr |
---|
63 | if not isinstance(errio, StringIO): |
---|
64 | return '' |
---|
65 | sys.stderr = sys.stdout |
---|
66 | errio.seek(0) |
---|
67 | return errio.read() |
---|
68 | |
---|
69 | def catchStderr(): |
---|
70 | old_handler = cherrypy.request.handler |
---|
71 | def wrapper(*args, **kwargs): |
---|
72 | sys.stderr = StringIO() |
---|
73 | ret = old_handler(*args, **kwargs) |
---|
74 | e = revertStandardError() |
---|
75 | if e: |
---|
76 | if isinstance(ret, dict): |
---|
77 | ret["error_text"] = e |
---|
78 | return ret |
---|
79 | if old_handler: |
---|
80 | cherrypy.request.handler = wrapper |
---|
81 | |
---|
82 | cherrypy.tools.catch_stderr = cherrypy.Tool('before_handler', catchStderr) |
---|
83 | |
---|
84 | class JSONEncoder(simplejson.JSONEncoder): |
---|
85 | def default(self, obj): |
---|
86 | if isinstance(obj, datetime.datetime): |
---|
87 | return str(obj) |
---|
88 | elif isinstance(obj, decimal.Decimal): |
---|
89 | return float(obj) |
---|
90 | else: |
---|
91 | return simplejson.JSONEncoder.default(self, obj) |
---|
92 | |
---|
93 | def jsonify_tool_callback(*args, **kwargs): |
---|
94 | if not cherrypy.request.cached: |
---|
95 | response = cherrypy.response |
---|
96 | response.headers['Content-Type'] = 'text/javascript' |
---|
97 | response.body = JSONEncoder().iterencode(response.body) |
---|
98 | |
---|
99 | cherrypy.tools.jsonify = cherrypy.Tool('before_finalize', jsonify_tool_callback, priority=30) |
---|
100 | |
---|
101 | def require_login(): |
---|
102 | """If the user isn't logged in, raise 403 with an error.""" |
---|
103 | if cherrypy.request.login is False: |
---|
104 | raise cherrypy.HTTPError(403, |
---|
105 | "You are not authorized to access that resource") |
---|
106 | |
---|
107 | cherrypy.tools.require_login = cherrypy.Tool('on_start_resource', require_login, priority=150) |
---|
108 | |
---|
109 | def require_POST(): |
---|
110 | """If the request isn't a POST request, raise 405 Method Not Allowed""" |
---|
111 | if cherrypy.request.method != "POST": |
---|
112 | raise cherrypy.HTTPError(405, |
---|
113 | "You must submit this request with POST") |
---|
114 | |
---|
115 | cherrypy.tools.require_POST = cherrypy.Tool('on_start_resource', require_POST, priority=150) |
---|
116 | |
---|
117 | def remote_user_login(): |
---|
118 | """Get the current user based on the SSL or GSSAPI environment |
---|
119 | variables and store it in the request object's login variable. This |
---|
120 | conforms to the CherryPy API: |
---|
121 | http://www.cherrypy.org/wiki/RequestObject#login |
---|
122 | |
---|
123 | If the user is logged in successfully, cherrypy.request.login is set |
---|
124 | to the username. If the user failed to log in, cherrypy.request.login |
---|
125 | is set to False. If the user did not attempt authentication, |
---|
126 | cherrypy.request.login is set to None.""" |
---|
127 | environ = cherrypy.request.wsgi_environ |
---|
128 | user = environ.get('REMOTE_USER') |
---|
129 | if user is None: |
---|
130 | return |
---|
131 | else: |
---|
132 | cherrypy.request.login = None # clear what cherrypy put there |
---|
133 | |
---|
134 | if environ.get('AUTH_TYPE') == 'Negotiate': |
---|
135 | # Convert the krb5 principal into a krb4 username |
---|
136 | if not user.endswith('@%s' % config.kerberos.realm): |
---|
137 | cherrypy.request.login = False # failed to login |
---|
138 | else: |
---|
139 | cherrypy.request.login = user.split('@')[0].replace('/', '.') |
---|
140 | else: |
---|
141 | cherrypy.request.login = user |
---|
142 | |
---|
143 | cherrypy.tools.remote_user_login = cherrypy.Tool('on_start_resource', remote_user_login, priority=50) |
---|
144 | |
---|
145 | def invirtwebstate_init(): |
---|
146 | """Initialize the cherrypy.request.state object from Invirt""" |
---|
147 | if not hasattr(cherrypy.request, "state"): |
---|
148 | cherrypy.request.state = State(cherrypy.request.login) |
---|
149 | |
---|
150 | cherrypy.tools.invirtwebstate = cherrypy.Tool('on_start_resource', invirtwebstate_init, priority=100) |
---|
151 | |
---|
152 | class View(object): |
---|
153 | _cp_config = {'tools.mako.directories': [os.path.join(os.path.dirname(__file__),'templates')]} |
---|