From d984566a881e7597af713b1bf171b043b85bd831 Mon Sep 17 00:00:00 2001
From: Dantenerosas <solleo55@gmail.com>
Date: Wed, 29 Nov 2023 16:15:48 +0300
Subject: [PATCH] refactored logging with json_logs func

---
 main.py                 | 22 ++++++++++++++++++++++
 src/core/link_parser.py |  4 +++-
 src/core/rabbitmq.py    |  6 +++---
 3 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index 91f82c3..e5b6589 100644
--- a/main.py
+++ b/main.py
@@ -1,8 +1,30 @@
 import asyncio
+import json
+from typing import Any
 from multiprocessing import freeze_support
 
 from src.core.master_service import MasterService
 
+from loguru import logger
+
+
+def json_logs(message: Any) -> None:
+    record = message.record
+    data = {
+        "timestamp": record["time"].strftime("%d.%m.%y %H.%M.%S %Z%z"),
+        "level": record["level"].name,
+        "message": record["message"],
+        "path": record["file"].path,
+        "function": record["function"],
+        "line": record["line"],
+    }
+    print(json.dumps(data))
+
+
+logger.remove(0)
+logger.add(json_logs)
+
+
 if __name__ == '__main__':
     freeze_support()
     loop = asyncio.new_event_loop()
diff --git a/src/core/link_parser.py b/src/core/link_parser.py
index 6dc205d..3caa6f3 100644
--- a/src/core/link_parser.py
+++ b/src/core/link_parser.py
@@ -1,6 +1,7 @@
 import asyncio
 import json
 
+from loguru import logger
 from playwright.async_api import async_playwright
 from playwright.async_api import Playwright
 from aio_pika import Message, connect, DeliveryMode
@@ -38,12 +39,13 @@ async def run(playwright: Playwright):
                 routing_key='hello',
             )
 
-            print(f" [x] Sent '{body}'")
+            logger.info(f" [x] Sent '{body}'")
             await page.keyboard.press("ArrowDown")
 
             while title == await page.title():
                 await page.title()
 
+
 async def main():
     async with async_playwright() as playwright:
         await run(playwright)
diff --git a/src/core/rabbitmq.py b/src/core/rabbitmq.py
index d940e99..4b9ef45 100644
--- a/src/core/rabbitmq.py
+++ b/src/core/rabbitmq.py
@@ -4,12 +4,13 @@ from functools import partial
 
 from aio_pika import connect, Message, DeliveryMode
 from aio_pika.abc import AbstractIncomingMessage
+from loguru import logger
 
 
 async def on_message(message: AbstractIncomingMessage, queue) -> None:
     async with message.process():
         await queue.put(json.loads(message.body))
-        print(f"     Message body is: {message.body!r}")
+        logger.info(f"     Message body is: {message.body!r}")
 
 
 async def get_messages(inner_queue) -> None:
@@ -23,8 +24,7 @@ async def get_messages(inner_queue) -> None:
         )
 
         await queue.consume(partial(on_message, queue=inner_queue))
-
-        print(" [*] Waiting for messages. To exit press CTRL+C")
+        logger.info("[*] Waiting for messages. To exit press CTRL+C")
         await asyncio.Future()