diff --git a/core/database/base.py b/core/database/base.py deleted file mode 100644 index 3d73964..0000000 --- a/core/database/base.py +++ /dev/null @@ -1,27 +0,0 @@ -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', - ] - 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') - return collection, db, client