From 19fd3032fc36cce09be4db89fc6580f7e124bb63 Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Sat, 29 May 2021 18:51:09 +0300 Subject: chore: reformat for consistency (placate flake8) Sponsored-by: author --- max-client.py | 20 ++++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) diff --git a/max-client.py b/max-client.py index d691909..488fc6b 100644 --- a/max-client.py +++ b/max-client.py @@ -10,27 +10,32 @@ class Compare: right: int type: str = "compare" + @dataclass class ComparisonResult: request_id: int answer: bool type: str = "comp_result" + @dataclass class ComputeMax: length: int type: str = "compute_max" + @dataclass class ComputeMin: length: int type: str = "compute_min" + @dataclass class Done: result: int type: str = "done" + def message_to_struct(message): if message["type"] == "compare": return Compare(**message) @@ -43,6 +48,7 @@ def message_to_struct(message): elif message["type"] == "done": return Done(**message) + class Client: def __init__(self, address, log=False): self.address = address if address else "http://localhost:5000" @@ -72,15 +78,21 @@ class Client: request_id = next_message.request_id left = next_message.left right = next_message.right - next_message = self.send(ComparisonResult(request_id, values[left] < values[right])) + next_message = self.send( + ComparisonResult(request_id, values[left] < values[right]) + ) else: raise Exception("Unexpected message: ", next_message) if __name__ == "__main__": - parser = argparse.ArgumentParser(description='Run the max computer') - parser.add_argument('--address', type=str, required=False, - help='address of the max computer (defaults to http://localhost:5000)') + parser = argparse.ArgumentParser(description="Run the max computer") + parser.add_argument( + "--address", + type=str, + required=False, + help="address of the max computer (defaults to http://localhost:5000)", + ) args = parser.parse_args() client = Client(args.address, log=True) -- cgit v1.2.1