From 8226534e173df938c533ebab6db8cd08a60901b9 Mon Sep 17 00:00:00 2001
From: Michael Merickel <michael@merickel.org>
Date: Sun, 18 Jun 2017 06:53:01 +0200
Subject: [PATCH] add a router.request_context context manager
---
pyramid/threadlocal.py | 27 +++++++++++++++++++++++++--
1 files changed, 25 insertions(+), 2 deletions(-)
diff --git a/pyramid/threadlocal.py b/pyramid/threadlocal.py
index 9429fe9..e8f8257 100644
--- a/pyramid/threadlocal.py
+++ b/pyramid/threadlocal.py
@@ -36,7 +36,8 @@
manager = ThreadLocalManager(default=defaults)
def get_current_request():
- """Return the currently active request or ``None`` if no request
+ """
+ Return the currently active request or ``None`` if no request
is currently active.
This function should be used *extremely sparingly*, usually only
@@ -44,11 +45,13 @@
``get_current_request`` outside a testing context because its
usage makes it possible to write code that can be neither easily
tested nor scripted.
+
"""
return manager.get()['request']
def get_current_registry(context=None): # context required by getSiteManager API
- """Return the currently active :term:`application registry` or the
+ """
+ Return the currently active :term:`application registry` or the
global application registry if no request is currently active.
This function should be used *extremely sparingly*, usually only
@@ -56,5 +59,25 @@
``get_current_registry`` outside a testing context because its
usage makes it possible to write code that can be neither easily
tested nor scripted.
+
"""
return manager.get()['registry']
+
+class RequestContext(object):
+ def __init__(self, request):
+ self.request = request
+
+ def begin(self):
+ request = self.request
+ registry = request.registry
+ manager.push({'registry': registry, 'request': request})
+ return request
+
+ def end(self):
+ manager.pop()
+
+ def __enter__(self):
+ return self.begin()
+
+ def __exit__(self, *args):
+ self.end()
--
Gitblit v1.9.3