chat-ai-api/app/db/souyue_mongo.py

from pymongo import MongoClient
from bson import ObjectId
from dotenv import load_dotenv
import os

load_dotenv()

_client = MongoClient(os.getenv("SOUYUE_MONGO_URI"), serverSelectionTimeoutMS=5000)
mblog = _client["souyue"]["mblog"]


def get_mblog_by_id(post_id: str) -> dict | None:
    try:
        return mblog.find_one({"_id": ObjectId(post_id)})
    except Exception:
        return None