Expose GzipEncoder through new encoders module

alekstorm · Nov 17, 2011 · 17c0415 · 17c0415
1 parent b98d573
commit 17c0415
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 7 deletions.
diff --git a/vortex/__init__.py b/vortex/__init__.py
@@ -192,7 +192,7 @@ def __str__(self):
         return b'\r\n'.join(lines) + b'\r\n\r\n'
 
 
-class GzipEncoder(object):
+class _GzipEncoder(object):
     def __init__(self, request, response):
         response.headers['Vary'] = response.headers.get('Vary', '') + ',Accept-Encoding'
         self._accepted = 'gzip' in request.headers.get('Accept-Encoding', '').replace(' ','').split(',')
@@ -218,7 +218,7 @@ def finish(self, data):
 
 
 class HTTPStream(object):
-    def __init__(self, request, response, encoders=[GzipEncoder]):
+    def __init__(self, request, response, encoders=[_GzipEncoder]):
         self._request = request
         self._response = response
         self._buffer = []

diff --git a/vortex/encoders.py b/vortex/encoders.py
@@ -0,0 +1,4 @@
+# All encoders should technically be defined here, but vortex.HTTPStream uses
+# GzipEncoder by default, which would create a circular dependency. Therefore,
+# it's defined in the root module, but exposed publicly here.
+from vortex import _GzipEncoder as GzipEncoder
diff --git a/vortex/streams.py b/vortex/streams.py