File size: 1,450 Bytes
f827a8b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | import os
import logging
from typing import Optional
from bson import ObjectId
from motor.motor_asyncio import AsyncIOMotorClient
logger = logging.getLogger(__name__)
_client: Optional[AsyncIOMotorClient] = None
_db = None
async def connect():
global _client, _db
uri = os.getenv("MONGODB_URI")
if not uri:
logger.warning("MONGODB_URI not set — MongoDB writes disabled")
return
db_name = os.getenv("MONGODB_DB_NAME", "myapp")
_client = AsyncIOMotorClient(uri)
_db = _client[db_name]
logger.info("Connected to MongoDB database '%s'", db_name)
async def disconnect():
global _client
if _client:
_client.close()
async def update_deepfake_result(post_id: str, image_index: int, is_deepfake: bool) -> None:
"""Write the boolean deepfake result back into the post's image_deepfake_results array."""
if _db is None:
logger.debug("MongoDB not connected; skipping write for post %s[%d]", post_id, image_index)
return
try:
field = f"image_deepfake_results.{image_index}"
result = await _db.posts.update_one(
{"_id": ObjectId(post_id)},
{"$set": {field: is_deepfake}},
)
if result.matched_count == 0:
logger.warning("Post %s not found; deepfake result not saved", post_id)
except Exception:
logger.exception("Failed to update deepfake result for post %s[%d]", post_id, image_index)
|