use logging instead of print

yoheinakajima · Dec 21, 2023 · 37c910f · 37c910f
1 parent 6eac7f9
commit 37c910f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/drivers/falkordb.py b/drivers/falkordb.py
@@ -1,3 +1,4 @@
+import logging
 import os
 from typing import Any
 
@@ -14,9 +15,9 @@ def __init__(self):
 
         # Check if connection is successful
         try:
-            print("FalkorDB database connected successfully!")
+            logging.info("FalkorDB database connected successfully!")
         except ValueError as ve:
-            print("FalkorDB database: {}".format(ve))
+            logging.error("FalkorDB database: {}".format(ve))
             raise
 
     def get_graph_data(self) -> tuple[list[dict[str, Any]], list[dict[str, Any]]]: