summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorHelmut Grohne <helmut@subdivi.de>2008-09-19 12:42:53 +0200
committerHelmut Grohne <helmut@subdivi.de>2008-09-19 12:42:53 +0200
commit863a8fd236b57fdb70e8008faea6f3e25bf025bb (patch)
tree69cdea7f486680aa2972f2afb4192bfcb15defa8
parentc02db04f584785adf9e00f9385eaf3aa46ef6436 (diff)
downloadwsgitools-863a8fd236b57fdb70e8008faea6f3e25bf025bb.tar.gz
added ReusableWSGIInputFilter
-rw-r--r--wsgitools/filters.py42
1 files changed, 42 insertions, 0 deletions
diff --git a/wsgitools/filters.py b/wsgitools/filters.py
index 1c7d03b..07675ba 100644
--- a/wsgitools/filters.py
+++ b/wsgitools/filters.py
@@ -388,3 +388,45 @@ class GzipWSGIFilter(BaseWSGIFilter):
self.gzip.close()
data = self.sio.getvalue()
return [data]
+
+class ReusableWSGIInputFilter(BaseWSGIFilter):
+ """Make environ["wsgi.input"] readable multiple times. Although this is not
+ required by the standard it is sometimes desirable to read wsgi.input
+ multiple times. This filter will therefore replace that variable with a
+ StringIO instance which provides a seek method.
+ """
+ @classmethod
+ def creator(cls, maxrequestsize):
+ """
+ Returns a function creating ReusableWSGIInputFilters with desired
+ maxrequestsize being set. If there is more data than maxrequestsize is
+ available in wsgi.input the rest will be ignored. (It is up to the
+ adapter to eat this data.)
+ @type maxrequestsize int
+ @param maxrequestsize is the maximum number of bytes to store in the
+ StringIO
+ """
+ return lambda:cls(maxrequestsize)
+ def __init__(self, maxrequestsize=65536):
+ """ReusableWSGIInputFilters constructor.
+ @type maxrequestsize int
+ @param maxrequestsize is the maximum number of bytes to store in the
+ StringIO, see creator
+ """
+ BaseWSGIFilter.__init__(self)
+ self.maxrequestsize = maxrequestsize
+
+ def filter_environ(self, environ):
+ """BaseWSGIFilter interface
+ @type environ: {str: str}
+ """
+
+ if isinstance(environ["wsgi.input"], StringIO):
+ return environ # nothing to be done
+
+ # XXX: is this really a good idea? use with care
+ environ["wsgitools.oldinput"] = environ["wsgi.input"]
+ data = StringIO(environ["wsgi.input"].read(self.maxrequestsize))
+ environ["wsgi.input"] = data
+
+ return environ