entidi2608 commited on
Commit
0bf745a
·
1 Parent(s): 57f5538
Files changed (2) hide show
  1. config.py +4 -1
  2. db/mongoDB.py +3 -7
config.py CHANGED
@@ -51,4 +51,7 @@ FRONTEND_URL = os.environ.get("FRONTEND_URL")
51
 
52
  APP_ENVIRONMENT = os.environ.get("APP_ENVIRONMENT")
53
 
54
- CHECKPOINT_FILE = "processed_files.log"
 
 
 
 
51
 
52
  APP_ENVIRONMENT = os.environ.get("APP_ENVIRONMENT")
53
 
54
+ CHECKPOINT_FILE = "processed_files.log"
55
+
56
+ MONGODB_CLOUD_URI= os.environ.get("MONGODB_CLOUD_URI")
57
+ DB_NAME= os.environ.get("DB_NAME")
db/mongoDB.py CHANGED
@@ -1,21 +1,17 @@
1
  from pymongo import MongoClient, errors
2
- from dotenv import load_dotenv
3
- import os
4
  import logging
5
  logger = logging.getLogger(__name__)
6
- load_dotenv()
7
 
8
- MONGO_URI = os.getenv("MONGODB_CLOUD_URI")
9
- DB_NAME = os.getenv("DB_NAME")
10
 
11
  try:
12
  # Kết nối MongoDB
13
- client = MongoClient(MONGO_URI, serverSelectionTimeoutMS=30000)
14
 
15
  # Trigger kết nối thử
16
  client.server_info() # Gây lỗi nếu không kết nối được
17
 
18
- db = client[DB_NAME]
19
  user_collection = db["users"]
20
  blacklist_collection = db["token_blacklist"]
21
  conversations_collection = db["conversations"]
 
1
  from pymongo import MongoClient, errors
2
+ import config
 
3
  import logging
4
  logger = logging.getLogger(__name__)
 
5
 
 
 
6
 
7
  try:
8
  # Kết nối MongoDB
9
+ client = MongoClient(config.MONGODB_CLOUD_URI, serverSelectionTimeoutMS=30000)
10
 
11
  # Trigger kết nối thử
12
  client.server_info() # Gây lỗi nếu không kết nối được
13
 
14
+ db = client[config.DB_NAME]
15
  user_collection = db["users"]
16
  blacklist_collection = db["token_blacklist"]
17
  conversations_collection = db["conversations"]