fix: replace print() with logger in responses, reloader, and __init__

robyn/logger.py

@@ -19,6 +19,7 @@ class Logger:
 
     def __init__(self):
         self.logger = logging.getLogger(__name__)
+        self.logger.addHandler(logging.NullHandler())
 
     def _format_msg(
         self,

robyn/reloader.py

@@ -17,7 +17,7 @@ def compile_rust_files(directory_path: str) -> List[str]:
     rust_binaries: list[str] = []
 
     for rust_file in rust_files:
-        print(f"Compiling rust file: {rust_file}")
+        logger.info("Compiling rust file: %s", rust_file)
 
         result = subprocess.run(
             [sys.executable, "-m", "rustimport", "build", rust_file],
@@ -26,9 +26,9 @@ def compile_rust_files(directory_path: str) -> List[str]:
             start_new_session=False,
         )
         if result.returncode != 0:
-            print(f"Error compiling rust file: {rust_file} \n {result.stderr.decode('utf-8')} \n {result.stdout.decode('utf-8')}")
+            logger.error("Error compiling rust file: %s\n%s\n%s", rust_file, result.stderr.decode("utf-8"), result.stdout.decode("utf-8"))
         else:
-            print(f"Compiled rust file: {rust_file}")
+            logger.info("Compiled rust file: %s", rust_file)
             rust_file_base = rust_file.removesuffix(".rs")
 
             # Define the search pattern for the binary file
@@ -63,18 +63,18 @@ def create_rust_file(file_name: str) -> None:
     )
 
     if result.returncode != 0:
-        print(
-            "Error creating rust file : %s %s",
+        logger.error(
+            "Error creating rust file: %s %s",
             result.stderr.decode("utf-8"),
             result.stdout.decode("utf-8"),
         )
     else:
-        print("Created rust file : %s", rust_file)
+        logger.info("Created rust file: %s", rust_file)
 
 
 def clean_rust_binaries(rust_binaries: List[str]) -> None:
     for file in rust_binaries:
-        print("Cleaning rust file : %s", file)
+        logger.info("Cleaning rust file: %s", file)
         os.remove(file)
 
 
@@ -127,7 +127,7 @@ def stop_server(self) -> None:
 
     def reload(self) -> None:
         self.stop_server()
-        print("Reloading the server")
+        logger.info("Reloading the server")
 
         new_env = os.environ.copy()
         new_env["IS_RELOADER_RUNNING"] = "True"  # This is used to check if a reloader is already running

robyn/responses.py

@@ -1,10 +1,13 @@
 import asyncio
+import logging
 import mimetypes
 import os
 from typing import AsyncGenerator, Generator, Optional, Union
 
 from robyn.robyn import Headers, Response
 
+logger = logging.getLogger(__name__)
+
 
 class FileResponse:
     def __init__(
@@ -117,7 +120,7 @@ async def get_next():
             raise StopIteration
         except Exception as e:
             # Log error and stop iteration
-            print(f"Error in async generator: {e}")
+            logger.exception("Error in async generator: %s", e)
             raise StopIteration