app/soc/middleware/value_store.py
author Sverre Rabbelier <srabbelier@gmail.com>
Sat, 12 Sep 2009 21:27:17 +0200
changeset 2910 60d56cf01b54
parent 2906 6fb53ed7aff4
permissions -rw-r--r--
Use the new optional argument to endRequest in the middleware Currently endRequest is called twice on an exception if Django intercepts the exception and returns it's own result. By setting optional=True we do not crash on 'assert self.in_request'.

#!/usr/bin/python2.5
#
# Copyright 2009 the Melange authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Middleware to set up and empty the value store.
"""

__authors__ = [
  '"Sverre Rabbelier" <sverre@rabbelier.nl>',
  ]


from soc.modules import callback


class ValueStoreMiddleware(object):
  """Middleware class to set up and empty the value store.
  """

  def start(self, request):
    """Sets up the value store.

    Args:
      request: a Django HttpRequest object
    """

    core = callback.getCore()
    core.startNewRequest(request)

  def end(self, request, optional):
    """Empties the value store.

    Args:
      request: a Django HttpRequest object
    """

    core = callback.getCore()
    core.endRequest(request, optional)

  def process_request(self, request):
    """Called when a request is made.

    See the Django middleware documentation for an explanation of
    the method signature.
    """
    self.start(request)

  def process_response(self, request, response):
    """Called when a response is returned.

    See the Django middleware documentation for an explanation of
    the method signature.
    """

    self.end(request, True)
    return response

  def process_exception(self, request, exception):
    """Called when an uncaught exception is raised.

    See the Django middleware documentation for an explanation of
    the method signature.
    """

    self.end(request, False)