Update db_operations/db_write.py

db_operations/db_write.py

@@ -1,7 +1,9 @@
 import pymongo
 import os
 from config import DATABASE, COLLECTION
+from logger import get_logger
 
+logger = get_logger()
 
 class DBWrite:
     """
@@ -42,6 +44,7 @@ class DBWrite:
             self.__client = None
 
     def insert_news_into_db(self, documents: list):
+        logger.warning('Entering insert_news_into_db()')
         if self.url is not None:
             if self.__error == 0:
                 self.__connect()
@@ -51,3 +54,4 @@ class DBWrite:
                 print("Insertion Successful")
             if self.__client is not None:
                 self.__close_connection()
+        logger.warning('Exiting insert_news_into_db()')