Fix memory leak from holding reference to toolbar (fixes #10)

This commit is contained in:
Matt Good
2012-02-14 23:56:36 -08:00
parent 17b629742d
commit bdbc570c91

View File

@@ -57,6 +57,7 @@ class DebugToolbarExtension(object):
self.app.before_request(self.process_request)
self.app.after_request(self.process_response)
self.app.teardown_request(self.teardown_request)
# Monkey-patch the Flask.dispatch_request method
app.dispatch_request = self.dispatch_request
@@ -175,6 +176,9 @@ class DebugToolbarExtension(object):
return response
def teardown_request(self, exc):
self.debug_toolbars.pop(request._get_current_object(), None)
def render(self, template_name, context):
template = self.jinja_env.get_template(template_name)
return template.render(**context)