summaryrefslogtreecommitdiff
path: root/repoze/bfg/wsgi.py
blob: 1a2c0460bc1b7dd0b09956ae882d4e6cf80d140e (plain)
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
from cgi import escape

try:
    from functools import wraps
except ImportError:
    # < 2.5
    from repoze.bfg.functional import wraps

def wsgiapp(wrapped):
    """ Decorator to turn a WSGI application into a repoze.bfg view callable.

    E.g.::

      @wsgiapp
      def hello_world(environ, start_response):
          body = 'Hello world'
          start_response('200 OK', [ ('Content-Type', 'text/plain'),
                                     ('Content-Length', len(body)) ] )
          return [body]

    Allows the following view declaration to be made::

       <view
          view=".views.hello_world"
          name="hello_world.txt"
          context="*"
        />

    The wsgiapp decorator will convert the result of the WSGI
    application to a Response and return it to repoze.bfg as if the
    WSGI app were a repoze.bfg view.
    """
    def decorator(context, request):
        return request.get_response(wrapped)
    return wraps(wrapped)(decorator) # pickleability

class HTTPException(object):
    def __call__(self, environ, start_response, exc_info=False):
        try:
            msg = escape(environ['message'])
        except KeyError:
            msg = ''
        html = """<body>
        <html><title>%s</title><body><h1>%s</h1>
        <code>%s</code>
        """ % (self.status, self.status, msg)
        headers = [('Content-Length', len(html)), ('Content-Type', 'text/html')]
        start_response(self.status, headers)
        return [html]

class NotFound(HTTPException):
    """ The default NotFound WSGI application """
    status = '404 Not Found'

class Unauthorized(HTTPException):
    """ The default Unauthorized WSGI application """
    status = '401 Unauthorized'