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

refactoring

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-06-14 11:26:46 +02:00
parent 2954bbc1ad
commit d8b5be9e72
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pyinfra/visitor.py
View File

@ -81,8 +81,8 @@ class ResponseStrategy(abc.ABC):
def handle_response(self, analysis_response: dict):
pass
def __call__(self, analysis_payload: dict):
return self.handle_response(analysis_payload)
def __call__(self, analysis_response: dict):
return self.handle_response(analysis_response)
class StorageStrategy(ResponseStrategy):
@ -98,8 +98,8 @@ class StorageStrategy(ResponseStrategy):
class ForwardingStrategy(ResponseStrategy):
def handle_response(self, body):
return body
def handle_response(self, analysis_response):
return analysis_response
class DispatchCallback(abc.ABC):
@ -163,12 +163,12 @@ class AggregationStorageStrategy(ResponseStrategy):
else:
return Nothing
def handle_response(self, analysis_payload, final=False):
def handle_response(self, analysis_response, final=False):
def upload_or_aggregate(analysis_payload):
return self.upload_or_aggregate(analysis_payload, request_metadata, last=not result_data.peek(False))
request_metadata = omit(analysis_payload, ["data"])
result_data = peekable(analysis_payload["data"])
request_metadata = omit(analysis_response, ["data"])
result_data = peekable(analysis_response["data"])
yield from filter(is_not_nothing, map(upload_or_aggregate, result_data))