orbit-oss/flask
4
4
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Preserve the request context in debug mode.

Browse source 
This makes it possible to access request information in the interactive
debugger.  Closes #8.
This commit is contained in:
Armin Ronacher 2010-04-16 02:27:37 +02:00
parent 40e0024d7b
commit 7b5015010b
1 changed files with 11 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

18
flask.py
View file

@ -14,7 +14,6 @@ import os
import sys import sys
from threading import local from threading import local
from contextlib import contextmanager
from jinja2 import Environment, PackageLoader, FileSystemLoader from jinja2 import Environment, PackageLoader, FileSystemLoader
from werkzeug import Request as RequestBase, Response as ResponseBase, \ from werkzeug import Request as RequestBase, Response as ResponseBase, \
LocalStack, LocalProxy, create_environ, cached_property, \ LocalStack, LocalProxy, create_environ, cached_property, \
@ -84,6 +83,16 @@ class _RequestContext(object):
self.g = _RequestGlobals() self.g = _RequestGlobals()
self.flashes = None self.flashes = None
def __enter__(self):
_request_ctx_stack.push(self)
def __exit__(self, exc_type, exc_value, tb):
# do not pop the request stack if we are in debug mode and an
# exception happened. This will allow the debugger to still
# access the request object in the interactive shell.
if tb is None or not self.app.debug:
_request_ctx_stack.pop()
def url_for(endpoint, **values): def url_for(endpoint, **values):
"""Generates a URL to the given endpoint with the method provided. """Generates a URL to the given endpoint with the method provided.
@ -618,7 +627,6 @@ class Flask(object):
response = self.process_response(response) response = self.process_response(response)
return response(environ, start_response) return response(environ, start_response)
@contextmanager
def request_context(self, environ): def request_context(self, environ):
"""Creates a request context from the given environment and binds """Creates a request context from the given environment and binds
it to the current context. This must be used in combination with it to the current context. This must be used in combination with
@ -632,11 +640,7 @@ class Flask(object):
:params environ: a WSGI environment :params environ: a WSGI environment
""" """
_request_ctx_stack.push(_RequestContext(self, environ)) return _RequestContext(self, environ)
try:
yield
finally:
_request_ctx_stack.pop()
def test_request_context(self, *args, **kwargs): def test_request_context(self, *args, **kwargs):
"""Creates a WSGI environment from the given values (see """Creates a WSGI environment from the given values (see