Added some more hooks for capturing output

1 files changed, 12 insertions, 5 deletions

diff --git a/paste/debug/prints.py b/paste/debug/prints.py
index f69b95a..c2cfaa6 100644
--- a/paste/debug/prints.py
+++ b/paste/debug/prints.py
@@ -19,7 +19,7 @@ __all__ = ['PrintDebugMiddleware']
 
 class TeeFile(object):
 
-    def __init__(self, *files):
+    def __init__(self, files):
         self.files = files
 
     def write(self, v):
@@ -35,6 +35,13 @@ class PrintDebugMiddleware(object):
     This middleware captures all the printed statements, and inlines
     them in HTML pages, so that you can see all the (debug-intended)
     print statements in the page itself.
+
+    There are two keys added to the environment to control this:
+    ``environ['paste.printdebug_listeners']`` is a list of functions
+    that will be called everytime something is printed.
+
+    ``environ['paste.remove_printdebug']`` is a function that, if
+    called, will disable printing of output for that request.
     """
 
     log_template = (
@@ -64,11 +71,11 @@ class PrintDebugMiddleware(object):
             removed.append(None)
         environ['paste.remove_printdebug'] = remove_printdebug
         logged = StringIO()
+        listeners = [logged]
+        environ['paste.printdebug_listeners'] = listeners
         if self.print_wsgi_errors:
-            replacement_stdout = TeeFile(environ['wsgi.errors'], logged)
-        else:
-            replacement_stdout = logged
-        output = StringIO()
+            listeners.append(environ['wsgi.errors'])
+        replacement_stdout = TeeFile(listeners)
         try:
             threadedprint.register(replacement_stdout)
             status, headers, body = wsgilib.intercept_output(