source: package_branches/invirt-web/cherrypy-rebased/code/view.py @ 2697

Last change on this file since 2697 was 2697, checked in by broder, 14 years ago

Remove unused external_remote_user_login function

File size: 5.4 KB
RevLine 
[2693]1import os, sys
[2659]2
3import cherrypy
4from mako.template import Template
5from mako.lookup import TemplateLookup
6import simplejson
7import datetime, decimal
[2693]8from StringIO import StringIO
[2669]9from invirt.config import structs as config
10from webcommon import State
[2659]11
12class MakoHandler(cherrypy.dispatch.LateParamPageHandler):
13    """Callable which sets response.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
27class MakoLoader(object):
28   
29    def __init__(self):
30        self.lookups = {}
[2693]31
32    def get_lookup(self, directories, module_directory=None,
33                     collection_size=-1, imports=[], **kwargs):
[2659]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',
[2673]45                                    imports=imports,
[2659]46                                    )
47            self.lookups[key] = lookup
[2693]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)
[2659]55       
56        # Replace the current handler.
57        cherrypy.request.template = t = lookup.get_template(filename)
58        cherrypy.request.handler = MakoHandler(t, cherrypy.request.handler, content_type)
59
60main = MakoLoader()
61cherrypy.tools.mako = cherrypy.Tool('on_start_resource', main)
62
[2693]63def revertStandardError():
64    """Move stderr to stdout, and return the contents of the old stderr."""
65    errio = sys.stderr
66    if not isinstance(errio, StringIO):
67        return ''
68    sys.stderr = sys.stdout
69    errio.seek(0)
70    return errio.read()
71
72def catchStderr():
73    old_handler = cherrypy.request.handler
74    def wrapper(*args, **kwargs):
75        sys.stderr = StringIO()
76        ret = old_handler(*args, **kwargs)
77        e = revertStandardError()
78        if e:
79            if isinstance(ret, dict):
80                ret["error_text"] = e
81        return ret
82    if old_handler:
83        cherrypy.request.handler = wrapper
84
85cherrypy.tools.catch_stderr = cherrypy.Tool('before_handler', catchStderr)
86
[2659]87class JSONEncoder(simplejson.JSONEncoder):
88        def default(self, obj):
89                if isinstance(obj, datetime.datetime):
90                        return str(obj)
91                elif isinstance(obj, decimal.Decimal):
92                        return float(obj)
93                else:
94                        return simplejson.JSONEncoder.default(self, obj)
95
96def jsonify_tool_callback(*args, **kwargs):
97    if not cherrypy.request.cached:
98        response = cherrypy.response
99        response.headers['Content-Type'] = 'text/javascript'
100        response.body = JSONEncoder().iterencode(response.body)
101
102cherrypy.tools.jsonify = cherrypy.Tool('before_finalize', jsonify_tool_callback, priority=30)
103
[2665]104def require_login():
105    """If the user isn't logged in, raise 403 with an error."""
106    if cherrypy.request.login is False:
107        raise cherrypy.HTTPError(403,
108            "You are not authorized to access that resource")
109
[2670]110cherrypy.tools.require_login = cherrypy.Tool('on_start_resource', require_login, priority=150)
[2665]111
[2685]112def require_POST():
113    """If the request isn't a POST request, raise 405 Method Not Allowed"""
114    if cherrypy.request.method != "POST":
115        raise cherrypy.HTTPError(405,
116                                 "You must submit this request with POST")
117
118cherrypy.tools.require_POST = cherrypy.Tool('on_start_resource', require_POST, priority=150)
119
[2670]120def remote_user_login():
121    """Get the current user based on the SSL or GSSAPI environment variables"""
122    environ = cherrypy.request.wsgi_environ
123    user = environ.get('REMOTE_USER')
124    if user is None:
125        return
126    else:
127        cherrypy.request.login = None # clear what cherrypy put there
128
129    if environ.get('AUTH_TYPE') == 'Negotiate':
130        # Convert the krb5 principal into a krb4 username
131        if not user.endswith('@%s' % config.kerberos.realm):
132            cherrypy.request.login = False # failed to login
133        else:
134            cherrypy.request.login = user.split('@')[0].replace('/', '.')
135    else:
136        cherrypy.request.login = user
137
138cherrypy.tools.remote_user_login = cherrypy.Tool('on_start_resource', remote_user_login, priority=50)
139
[2669]140def invirtwebstate_init():
141    """Initialize the cherrypy.request.state object from Invirt"""
[2690]142    if not hasattr(cherrypy.request, "state"):
143        cherrypy.request.state = State(cherrypy.request.login)
[2669]144
145cherrypy.tools.invirtwebstate = cherrypy.Tool('on_start_resource', invirtwebstate_init, priority=100)
146
[2659]147class View(object):
148    _cp_config = {'tools.mako.directories': [os.path.join(os.path.dirname(__file__),'templates')]}
Note: See TracBrowser for help on using the repository browser.