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