"""MongoDB database interaction module."""
import os
from pymongo import MongoClient

from dotenv import load_dotenv

load_dotenv()

MongodbClient = MongoClient(os.getenv("MONGODB_URI"))
article_collection = MongodbClient["FinFAST_China"]["Article"]
category_collection = MongodbClient["FinFAST_China"]["Category"]
entity_collection = MongodbClient["FinFAST_China"]["Entity"]
keywords_collection = MongodbClient["FinFAST_China"]["Keywords"]
summary_collection = MongodbClient["FinFAST_China"]["Summary"]
lda_collection = MongodbClient["FinFAST_China"]["LDA"]