moved shared functions

core/database/database.py

@@ -1,24 +1,27 @@
-from pymongo import MongoClient
+from __future__ import annotations
-from dotenv import load_dotenv
+
 import logging
 import os
 
+from dotenv import load_dotenv
+from pymongo import MongoClient
+
 
 def connect():
     """Connect to MongoDB."""
     # load env vars
     load_dotenv()
     necessary_env_vars = [
-        "MONGO_HOST",
+        'MONGO_HOST',
-        "MONGO_DB",
+        'MONGO_DB',
-        "MONGO_COLLECTION"
+        'MONGO_COLLECTION',
     ]
     for env_var in necessary_env_vars:
         assert env_var in os.environ, f"{env_var} not found"
     # connect to database
-    client = MongoClient(os.getenv("MONGO_HOST"))
+    client = MongoClient(os.getenv('MONGO_HOST'))
-    db = client[os.getenv("MONGO_DB")]
+    db = client[os.getenv('MONGO_DB')]
-    collection = db[os.getenv("MONGO_COLLECTION")]
+    collection = db[os.getenv('MONGO_COLLECTION')]
-    collection.create_index("name", unique=True)
+    collection.create_index('name', unique=True)
-    logging.info("connected to database")
+    logging.info('connected to database')
     return collection, db, client