diff --git a/src/server.py b/src/server.py index f54bcfe..cda6345 100644 --- a/src/server.py +++ b/src/server.py @@ -105,12 +105,12 @@ class Server: """ client_hello = await reader.readline() peer_is_sender = client_hello.decode('utf-8') == 'sender\n' - await writer.drain() if peer_is_sender: self._working = True logging.info("Sender is connecting...") # Send start signal to client writer.write("Start!\n".encode('utf-8')) + await writer.drain() await self.run_reader(reader=reader) logging.info("Incoming transmission ended") else: @@ -119,6 +119,7 @@ class Server: await asyncio.sleep(.5) # Send start signal to client writer.write("Start!\n".encode('utf-8')) + await writer.drain() await self.run_writer(writer=writer) logging.info("Outgoing transmission ended") self._working = False