From aeb60a97a9432c0911f106653f17c28a62ec1ab6 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Mon, 20 Jun 2022 10:09:58 +0200 Subject: [PATCH] refactoring --- pyinfra/visitor/strategies/response/aggregation.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pyinfra/visitor/strategies/response/aggregation.py b/pyinfra/visitor/strategies/response/aggregation.py index b9489a5..d3845f9 100644 --- a/pyinfra/visitor/strategies/response/aggregation.py +++ b/pyinfra/visitor/strategies/response/aggregation.py @@ -50,11 +50,11 @@ class AggregationStorageStrategy(ResponseStrategy): self.storage.put_object(**object_descriptor, data=compress(data)) return {**storage_upload_info, "responseFile": object_descriptor["object_name"]} + def upload_queue_items(self, storage_upload_info): + data = json.dumps(self.merge_queue_items()).encode() + return self.put_object(data, storage_upload_info) + def merge_queue_items(self): merged_buffer_content = self.merger(self.buffer) self.buffer.clear() return merged_buffer_content - - def upload_queue_items(self, storage_upload_info): - data = json.dumps(self.merge_queue_items()).encode() - return self.put_object(data, storage_upload_info)