tidy up & black

pyinfra/callback.py

@@ -65,8 +65,10 @@ def make_callback_for_output_queue(json_wrapped_body_processor, output_queue_nam
         if not CONFIG.service.response.save:
             channel.basic_publish(exchange="", routing_key=output_queue_name, body=result)
         else:
-            upload_compressed_response(get_storage(CONFIG.storage.backend), CONFIG.storage.bucket, dossier_id, file_id, result)
+            upload_compressed_response(
-            result = json.dumps({"dossierId" : dossier_id, "fileId" : file_id})
+                get_storage(CONFIG.storage.backend), CONFIG.storage.bucket, dossier_id, file_id, result
+            )
+            result = json.dumps({"dossierId": dossier_id, "fileId": file_id})
             channel.basic_publish(exchange="", routing_key=output_queue_name, body=result)
 
         channel.basic_ack(delivery_tag=method.delivery_tag)

pyinfra/storage/storages.py

@@ -13,6 +13,7 @@ def get_azure_storage(config=None):
 def get_s3_storage(config=None):
     return Storage(S3StorageAdapter(get_s3_client(config)))
 
+
 def get_storage(storage_backend):
 
     if storage_backend == "s3":

pyinfra/utils/file.py

@@ -1,12 +1,7 @@
 """Defines utilities for different operations on files."""
 
 import gzip
-from io import BytesIO, StringIO
-import json
 import os
-from pathlib import Path
-import sys
-import tempfile
 
 from pyinfra.config import CONFIG