ztfy/utils/request.py
changeset 0 712d20d2751e
child 2 20f3c0eb8fdf
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/ztfy/utils/request.py	Sat Jan 10 00:35:35 2009 +0100
@@ -0,0 +1,74 @@
+### -*- coding: utf-8 -*- ####################################################
+##############################################################################
+#
+# Copyright (c) 2008 Thierry Florac <tflorac AT ulthar.net>
+# All Rights Reserved.
+#
+# This software is subject to the provisions of the Zope Public License,
+# Version 2.1 (ZPL).  A copy of the ZPL should accompany this distribution.
+# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
+# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
+# FOR A PARTICULAR PURPOSE.
+#
+##############################################################################
+"""
+
+$Id: $
+"""
+
+__version__   = "$Revision: $"
+__release__   = "$Id: $"
+__docformat__ = "restructuredtext"
+ 
+# import standard packages
+
+# import Zope3 interfaces
+from zope.publisher.interfaces import IRequest
+from zope.annotation.interfaces import IAnnotations
+
+# import local interfaces
+
+# import Zope3 packages
+from zope.security.management import getInteraction
+
+# import local packages
+
+from ztfy.utils import _
+
+
+def getRequest():
+    """Extract request from current interaction"""
+    interaction = getInteraction()
+    for participation in interaction.participations:
+        if IRequest.providedBy(participation):
+            return participation
+    raise RuntimeError, _("No Request in interaction !")
+
+
+def getRequestPrincipal(request=None):
+    """Get principal from given request"""
+    if request is None:
+        request = getRequest()
+    return request.principal
+
+
+def getRequestAnnotations(request=None):
+    """Get annotations from given request"""
+    if request is None:
+        request = getRequest()
+    return IAnnotations(request)
+
+
+def getRequestData(key, request=None, default=None):
+    """Get request data, stored into request annotations"""
+    try:
+        annotations = getRequestAnnotations(request)
+        return annotations.get(key, default)
+    except:
+        return default
+
+
+def setRequestData(key, data, request=None):
+    annotations = getRequestAnnotations(request)
+    annotations[key] = data