From 4cdaa84c6552e3ff51ac5d15b08aece3383e136f Mon Sep 17 00:00:00 2001 From: Stefan Allius Date: Fri, 3 Nov 2023 18:28:04 +0100 Subject: [PATCH] move self.server_side to Message class --- app/src/async_stream.py | 3 +-- app/src/messages.py | 3 ++- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/app/src/async_stream.py b/app/src/async_stream.py index cb4f115..96a0ff5 100644 --- a/app/src/async_stream.py +++ b/app/src/async_stream.py @@ -11,11 +11,10 @@ class AsyncStream(Message): def __init__(self, reader, writer, addr, remote_stream, server_side: bool ) -> None: - super().__init__() + super().__init__(server_side) self.reader = reader self.writer = writer self.remoteStream = remote_stream - self.server_side = server_side self.addr = addr ''' diff --git a/app/src/messages.py b/app/src/messages.py index b80da16..77d2742 100644 --- a/app/src/messages.py +++ b/app/src/messages.py @@ -74,8 +74,9 @@ class Message(metaclass=IterRegistry): _registry = [] new_stat_data = {} - def __init__(self): + def __init__(self, server_side: bool): self._registry.append(weakref.ref(self)) + self.server_side = server_side self.header_valid = False self.header_len = 0 self.data_len = 0