add mongodb storage apis

Author: Jack Li

storage/base.py

@@ -0,0 +1,25 @@
+from pymongo import collection
+
+def __insert_one__(coll: collection.Collection, doc: dict):
+   return coll.insert_one(doc)
+
+def __insert_many__(coll: collection.Collection, docs: list):
+   return coll.insert_many(docs)
+
+def __delete_one__(coll: collection.Collection, filter: dict):
+   return coll.delete_one(filter)
+
+def __delete_many__(coll: collection.Collection, filter: dict):
+   return coll.delete_many(filter)
+
+def __find_one_and_replace__(coll: collection.Collection, filter: dict, new_doc: dict):
+   return coll.find_one_and_replace(filter, new_doc)
+
+def __find_one_and_delete__(coll: collection.Collection, filter: dict):
+   return coll.find_one_and_delete(filter)
+
+def __find_one__(coll: collection.Collection, filter: dict, projection):
+   return coll.find_one(filter=filter, projection=projection)
+
+def __find_many__(coll: collection.Collection, filter: dict, projection, **kwargs):
+   return coll.find(filter=filter, projection=projection, **kwargs)

storage/storage.py

@@ -0,0 +1,49 @@
+from ..utils import get_db_handle
+from .base import *
+
+# insert one
+def insert_one(collection_name, doc: dict, db_name="tickets"):
+   db = get_db_handle(db_name)
+   coll = db[collection_name]
+   return __insert_one__(coll, doc)
+
+# insert many
+def insert_many(collection_name, docs: list, db_name="tickets"):
+   db = get_db_handle(db_name)
+   coll = db[collection_name]
+   return __insert_many__(coll, docs)
+
+# delete one
+def delete_one(collection_name, filter: dict, db_name="tickets"):
+   db = get_db_handle(db_name)
+   coll = db[collection_name]
+   return __delete_one__(coll, filter)
+
+# delete many
+def delete_many(collection_name, filter: dict, db_name="tickets"):
+   db = get_db_handle(db_name)
+   coll = db[collection_name]
+   return __delete_many__(coll, filter)
+
+# find one and replace
+def find_one_and_replace(collection_name, filter: dict, new_doc: dict, db_name="tickets"):
+   db = get_db_handle(db_name)
+   coll = db[collection_name]
+   return __find_one_and_replace__(coll, filter, new_doc)
+
+def find_one_and_delete(collection_name, filter: dict, db_name="tickets"):
+   db = get_db_handle(db_name)
+   coll = db[collection_name]
+   return __find_one_and_delete__(coll, filter)
+
+# find one
+def find_one(collection_name, filter: dict, projection=None, db_name="tickets"):
+   db = get_db_handle(db_name)
+   coll = db[collection_name]
+   return __find_one__(coll, filter, projection)
+
+# find many
+def find_many(collection_name, filter: dict, projection=None, db_name="tickets", **kwargs):
+   db = get_db_handle(db_name)
+   coll = db[collection_name]
+   return __find_many__(coll, filter, projection, **kwargs)

tmtracker/app.py

@@ -9,6 +9,5 @@ class MyAppConfig(AppConfig):
 
    def ready(self):
       print(f"server started at {datetime.now().strftime('%d/%m/%Y %H:%M:%S')}")
-      get_db_handle('tickets')
       print("=== database connection is established ===")
       start()