From 595b68ba03ccb1535b4d07ecc25cb4c7eea5e854 Mon Sep 17 00:00:00 2001 From: Stefan Allius Date: Sun, 6 Oct 2024 21:20:53 +0200 Subject: [PATCH] reduce cognitive complexity --- app/src/async_stream.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/app/src/async_stream.py b/app/src/async_stream.py index 21bb6b9..559e48a 100644 --- a/app/src/async_stream.py +++ b/app/src/async_stream.py @@ -179,11 +179,7 @@ class AsyncStream(AsyncIfcImpl): self.proc_start = time.time() while True: try: - if self.proc_start: - proc = time.time() - self.proc_start - if proc > self.proc_max: - self.proc_max = proc - self.proc_start = None + self.__calc_proc_time() dead_conn_to = self.__timeout() await asyncio.wait_for(self.__async_read(), dead_conn_to) @@ -220,6 +216,13 @@ class AsyncStream(AsyncIfcImpl): f"{traceback.format_exc()}") await asyncio.sleep(0) # be cooperative to other task + def __calc_proc_time(self): + if self.proc_start: + proc = time.time() - self.proc_start + if proc > self.proc_max: + self.proc_max = proc + self.proc_start = None + async def disc(self) -> None: """Async disc handler for graceful disconnect""" self.remote = None