diff --git a/pyinfra/flask.py b/pyinfra/flask.py index 0fc294c..81e8b94 100644 --- a/pyinfra/flask.py +++ b/pyinfra/flask.py @@ -46,6 +46,7 @@ def set_up_probing_webserver(): metric = requests.get(prom_endpoint) metric.raise_for_status() return metric.text + return inner() return app diff --git a/pyinfra/visitor.py b/pyinfra/visitor.py index 69f75e0..1bff413 100644 --- a/pyinfra/visitor.py +++ b/pyinfra/visitor.py @@ -23,17 +23,11 @@ def get_response_object_name(body): def get_object_descriptor(body): - return { - "bucket_name": CONFIG.storage.bucket, - "object_name": get_object_name(body) - } + return {"bucket_name": CONFIG.storage.bucket, "object_name": get_object_name(body)} def get_response_object_descriptor(body): - return { - "bucket_name": CONFIG.storage.bucket, - "object_name": get_response_object_name(body) - } + return {"bucket_name": CONFIG.storage.bucket, "object_name": get_response_object_name(body)} class ResponseStrategy(abc.ABC): @@ -47,9 +41,7 @@ class ResponseStrategy(abc.ABC): class StorageStrategy(ResponseStrategy): def handle_response(self, body): - self.storage.put_object( - **get_response_object_descriptor(body), data=gzip.compress(json.dumps(body).encode()) - ) + self.storage.put_object(**get_response_object_descriptor(body), data=gzip.compress(json.dumps(body).encode())) body.pop("data") return body @@ -60,7 +52,6 @@ class ForwardingStrategy(ResponseStrategy): class QueueVisitor: - def __init__(self, storage: Storage, callback: Callable, response_strategy): self.storage = storage self.callback = callback @@ -69,7 +60,6 @@ class QueueVisitor: self.response_strategy.storage = self.storage def load_data(self, body): - def download(): logging.debug(f"Downloading {object_descriptor}...") data = self.storage.get_object(**object_descriptor)