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

MongodbClient = MongoClient(os.getenv('MONGODB_URI'))
FinFastMongodbClient = MongoClient(os.getenv("MONGODB_FINFAST_URI"))
article_collection = FinFastMongodbClient["FinFAST_China"]["Article"]
category_collection = FinFastMongodbClient["FinFAST_China"]["Category"]
entity_collection = FinFastMongodbClient["FinFAST_China"]["Entity"]