Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

applied black

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-14 14:49:21 +01:00
parent f78a6894bc
commit 59d0de97e3
2 changed files with 4 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pyinfra/flask.py
View File

@ -46,6 +46,7 @@ def set_up_probing_webserver():
metric = requests.get(prom_endpoint) metric = requests.get(prom_endpoint)
metric.raise_for_status() metric.raise_for_status()
return metric.text return metric.text
return inner() return inner()
return app return app

16
pyinfra/visitor.py
View File

@ -23,17 +23,11 @@ def get_response_object_name(body):
def get_object_descriptor(body): def get_object_descriptor(body):
return { return {"bucket_name": CONFIG.storage.bucket, "object_name": get_object_name(body)}
"bucket_name": CONFIG.storage.bucket,
"object_name": get_object_name(body)
}
def get_response_object_descriptor(body): def get_response_object_descriptor(body):
return { return {"bucket_name": CONFIG.storage.bucket, "object_name": get_response_object_name(body)}
"bucket_name": CONFIG.storage.bucket,
"object_name": get_response_object_name(body)
}
class ResponseStrategy(abc.ABC): class ResponseStrategy(abc.ABC):
@ -47,9 +41,7 @@ class ResponseStrategy(abc.ABC):
class StorageStrategy(ResponseStrategy): class StorageStrategy(ResponseStrategy):
def handle_response(self, body): def handle_response(self, body):
self.storage.put_object( self.storage.put_object(**get_response_object_descriptor(body), data=gzip.compress(json.dumps(body).encode()))
**get_response_object_descriptor(body), data=gzip.compress(json.dumps(body).encode())
)
body.pop("data") body.pop("data")
return body return body
@ -60,7 +52,6 @@ class ForwardingStrategy(ResponseStrategy):
class QueueVisitor: class QueueVisitor:
def __init__(self, storage: Storage, callback: Callable, response_strategy): def __init__(self, storage: Storage, callback: Callable, response_strategy):
self.storage = storage self.storage = storage
self.callback = callback self.callback = callback
@ -69,7 +60,6 @@ class QueueVisitor:
self.response_strategy.storage = self.storage self.response_strategy.storage = self.storage
def load_data(self, body): def load_data(self, body):
def download(): def download():
logging.debug(f"Downloading {object_descriptor}...") logging.debug(f"Downloading {object_descriptor}...")
data = self.storage.get_object(**object_descriptor) data = self.storage.get_object(**object_descriptor)