fix: update import paths to streamline database collection access

app/collectors/finfast/article.py

@@ -2,7 +2,7 @@
 from venv import logger
 from datetime import datetime, timedelta
 from pymongo.errors import PyMongoError
-from database.mongodb import article_collection
+from database import article_collection
 
 from .utils import scan_dynamodb_table, delete_old_documents, upsert_item
 

app/collectors/finfast/category.py

@@ -11,7 +11,7 @@ from typing import Dict, List, Tuple
 from collections import defaultdict
 from dataclasses import dataclass
 from botocore.exceptions import ClientError
-from database.mongodb import category_collection
+from database import category_collection
 
 from ..utils import get_client_connection
 

app/collectors/finfast/entity.py

@@ -1,7 +1,7 @@
 """Module for collecting and managing entity data from DynamoDB to MongoDB."""
 from datetime import datetime, timedelta
 from pymongo.errors import PyMongoError
-from database.mongodb import entity_collection
+from database import entity_collection
 
 from .utils import scan_dynamodb_table, delete_old_documents, upsert_item
 

app/controllers/category.py

@@ -4,7 +4,7 @@ Category Controller - Business logic for handling category data.
 This module contains functions that interact with the database
 to fetch and process data sorted by category
 """
-from database.mongodb import category_collection
+from database import category_collection
 def get_categories():
 
     """

app/controllers/summary/utils.py

@@ -6,7 +6,7 @@ from datetime import datetime, timedelta
 from typing import Dict, Any, List, DefaultDict
 from collections import defaultdict
 
-from database.mongodb import article_collection, entity_collection
+from database import article_collection, entity_collection
 
 
 def _get_latest_publish_date_from_collection(collection) -> datetime:

app/database/__init__.py

@@ -1,10 +1,8 @@
 """Module for Mongodb database"""
-from .mongodb import MongodbClient, FinFastMongodbClient
-
-MongoClient = MongodbClient
-FinFastMongoClient = FinFastMongodbClient
+from .mongodb import article_collection, entity_collection, category_collection
 
 __all__ = [
-    "MongoClient",
-    "FinFastMongoClient"
+    "article_collection",
+    "entity_collection",
+    "category_collection"
 ]

app/database/mongodb.py

@@ -3,7 +3,6 @@ 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"]
+article_collection = MongodbClient["FinFAST_China"]["Article"]
+category_collection = MongodbClient["FinFAST_China"]["Category"]
+entity_collection = MongodbClient["FinFAST_China"]["Entity"]