From 8238e451ded2965164883d450e12b7a3d20fc63e Mon Sep 17 00:00:00 2001 From: PAN Lei Date: Wed, 17 Dec 2025 08:42:56 +0100 Subject: [PATCH] use mongo2 db --- src/db/mongo_manager.py | 20 ++++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) diff --git a/src/db/mongo_manager.py b/src/db/mongo_manager.py index 37350e4..1680d5b 100755 --- a/src/db/mongo_manager.py +++ b/src/db/mongo_manager.py @@ -1,10 +1,10 @@ import datetime import logging import time +from pymongo import MongoClient from typing import Union -from pymongo import MongoClient - +from src.pojo.LinkPojo import LinkPojo from src.pojo.ReserveResultPojo import ReserveResultPojo from src.pojo.ResultEnum import ResultEnum from src.pojo.accepted_appointment_pojo import AcceptedAppointmentPojo @@ -14,7 +14,7 @@ from src.pojo.contact_pojo import ContactPojo from src.pojo.mail.mail_pojo import MailAddress from src.pojo.users.regisered_user_pojo import RegisteredUserPojo -MONGO_DB_URL = "mongo.lpaconsulting.fr" +MONGO_DB_URL = "mongodb://mongo2.lpaconsulting.fr" # MONGO_DB_URL = "clusterfr.zlmpq.mongodb.net" CAPTCHA_ERROR_COLLECTION_PREFIX = "CAPTCHA_ERROR_" BLACK_LIST = "BLACK_LIST" @@ -30,7 +30,8 @@ CONTACT_LIST_SERIAL_MAP = "CONTACT_LIST_SERIAL_MAP" class MongoDbManager: def __init__(self): - client = MongoClient(MONGO_DB_URL, username='appointment', password='Rdv@2022', authSource='appointment') + client = MongoClient(MONGO_DB_URL, username='appointment', password='Rdv@20222021', authSource='appointment', + connectTimeoutMS=60 * 1000) self.db = client.appointment self.logger = logging.getLogger("mongoDb") @@ -78,6 +79,17 @@ class MongoDbManager: except Exception as Error: self.logger.info(Error) + def get_links_to_validate(self) -> list: + collection_name = LINKS_TO_VALIDATE + link_list = [] + try: + collection_to_use = self.db[collection_name] + for document in collection_to_use.find(): + link_list.append(LinkPojo.from_firestore_dict(document)) + except Exception as error: + self.logger.info(error) + return link_list + def upload_contact_list(self, contact_list: list): try: