updated

scripts/constants/db_connection.py

@@ -11,9 +11,7 @@ def db_connect():
         # Create a database
         db_name = 'db_person'
         db = connection[db_name]
-        collection_name = 'personal_details'
-        collection = db[collection_name]
-        print("Database Connected Successfully !")
-        return collection
+        print("Database connected successfully!")
+        return db
     except Exception as e:
         print("Error ", e)

scripts/core/handlers/api_functions.py

@@ -11,7 +11,9 @@ connect(db='db_person', host='localhost', port=27017)
 
 class operations_on_api:
     def __init__(self):
-        self.collection = db_connect()
+        self.db = db_connect()
+        collection_name = 'personal_details'
+        self.collection = self.db[collection_name]
 
     def insert_all_data(self):
         try:
@@ -43,7 +45,7 @@ class operations_on_api:
                                              phone_no=details.phone_no)
             result = self.collection.insert_one(dict(person_details))
             if result:
-                print("Data inserted to the document", dict(person_details))
+                print("Data inserted to the document\n", dict(person_details))
             return result
         except Exception as e:
             print("Data insertion error-", e)
@@ -61,6 +63,7 @@ class operations_on_api:
             if update_details.phone_no is not None:
                 set_data["$set"]["phone_no"] = update_details.phone_no
             result = self.collection.update_one(condition_data, set_data)
+            print("Updated Data:\n", set_data["$set"])
             return result
         except Exception as e:
             print("Data updation error-", e)
@@ -68,7 +71,9 @@ class operations_on_api:
     def delete_one_document(self, document_id):
         try:
             condition_data = {"id": document_id}
+            document = self.collection.find_one(condition_data)
             result = self.collection.delete_one(condition_data)
+            print("Deleted document details:\n", document)
             return result
         except Exception as e:
             print("Data deletion error-", e)

scripts/core/handlers/insert_data_into_db.py

-from scripts.constants.app_configuration import csv_path
-from scripts.core.handlers.read_file import extract_data
-
-
 def insert_into_db(collection, data):
-    status = collection.insert_many(data)
-    if status:
-        return True
-    else:
-        return False
-
+    try:
+        status = collection.insert_many(data)
+        if status:
+            return True
+        else:
+            return False
+    except Exception as e:
+        print("Data insertion to db error-", e)