Extract method .create_compressor x2 (#207)

evansd · Nov 19, 2018 · 7000c93 · 7000c93
1 parent 4cdbe2a
commit 7000c93
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/whitenoise/storage.py b/whitenoise/storage.py
@@ -34,10 +34,13 @@ def fallback_post_process(self, paths, dry_run=False, **options):
             for path in paths:
                 yield path, None, False
 
+    def create_compressor(self, **kwargs):
+        return Compressor(**kwargs)
+
     def post_process_with_compression(self, files):
         extensions = getattr(settings,
                              'WHITENOISE_SKIP_COMPRESS_EXTENSIONS', None)
-        compressor = Compressor(extensions=extensions, quiet=True)
+        compressor = self.create_compressor(extensions=extensions, quiet=True)
         for name, hashed_name, processed in files:
             yield name, hashed_name, processed
             if isinstance(processed, Exception):
@@ -171,10 +174,13 @@ def delete_files(self, files_to_delete):
                 if e.errno != errno.ENOENT:
                     raise
 
+    def create_compressor(self, **kwargs):
+        return Compressor(**kwargs)
+
     def compress_files(self, names):
         extensions = getattr(settings,
                              'WHITENOISE_SKIP_COMPRESS_EXTENSIONS', None)
-        compressor = Compressor(extensions=extensions, quiet=True)
+        compressor = self.create_compressor(extensions=extensions, quiet=True)
         for name in names:
             if compressor.should_compress(name):
                 path = self.path(name)