From d90b280d4149fb4d48ce037bc6291d40ec3b454d Mon Sep 17 00:00:00 2001
From: KKlochko <KKlochko@protonmail.com>
Date: Tue, 12 Dec 2023 18:01:45 +0200
Subject: [PATCH] Add a function to receive a message.

---
 notification_client/client.py | 11 ++++++++---
 notification_client/main.py   |  2 +-
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/notification_client/client.py b/notification_client/client.py
index 22a8196..de86e33 100644
--- a/notification_client/client.py
+++ b/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')
diff --git a/notification_client/main.py b/notification_client/main.py
index f5ed8f1..162199b 100644
--- a/notification_client/main.py
+++ b/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)