refactored logging with json_logs func

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)

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()