diff --git a/src/database/__init__.py b/core/database/__init__.py similarity index 100% rename from src/database/__init__.py rename to core/database/__init__.py diff --git a/src/database/classes.py b/core/database/classes.py similarity index 100% rename from src/database/classes.py rename to core/database/classes.py diff --git a/src/database/database.py b/core/database/database.py similarity index 51% rename from src/database/database.py rename to core/database/database.py index 8f10ac8..3d73964 100644 --- a/src/database/database.py +++ b/core/database/database.py @@ -1,24 +1,27 @@ -from pymongo import MongoClient -from dotenv import load_dotenv +from __future__ import annotations + 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_DB", - "MONGO_COLLECTION" + 'MONGO_HOST', + 'MONGO_DB', + '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")) - db = client[os.getenv("MONGO_DB")] - collection = db[os.getenv("MONGO_COLLECTION")] - collection.create_index("name", unique=True) - logging.info("connected to database") + client = MongoClient(os.getenv('MONGO_HOST')) + db = client[os.getenv('MONGO_DB')] + collection = db[os.getenv('MONGO_COLLECTION')] + collection.create_index('name', unique=True) + logging.info('connected to database') return collection, db, client diff --git a/src/database/utils.py b/core/database/utils.py similarity index 100% rename from src/database/utils.py rename to core/database/utils.py