Add a function to receive a message.

notification_client/client.py

@@ -32,12 +32,17 @@ class Client:
         reader, writer = await asyncio.open_connection(self.__ip, self.__port)
         await self.handle(reader, writer)
 
+    @staticmethod
+    async def receive_message(reader) -> str:
+        size, = struct.unpack('<L', await reader.readexactly(4))
+        message = await reader.readexactly(size)
+        return message.decode('utf-8')
+
     async def handle(self, reader, writer):
         try:
             while True:
-                size, = struct.unpack('<L', await reader.readexactly(4))
-                message = await reader.readexactly(size)
-                print(f"{message.decode('utf-8') =}")
+                title = await self.receive_message(reader)
+                message = await self.receive_message(reader)
 
         except asyncio.CancelledError:
             print('Something went wrong')

notification_client/main.py

@@ -19,7 +19,7 @@
 
 import asyncio
 
-from .client import Client
+from notification_client.client import Client
 
 if __name__ == '__main__':
     client = Client('127.0.0.1', 5555)