src/mongo_logger.py

import os, uuid
from pymongo import MongoClient
from datetime import datetime, timezone

client = MongoClient(os.getenv("MONGO_URI"))
collection = client.adgenesis.records

def create_log(category, file_name, prompt):
    log_id = str(uuid.uuid4())
    collection.insert_one({
        "_id": log_id,
        "category": category,
        "prompt": prompt,
        "file_name": file_name,
        "status": "in_progress",
        "urls": [],
        "message": "Processing started",
        "created_at": datetime.now(timezone.utc),
        "updated_at": datetime.now(timezone.utc),
        "lob": "leadgen_vivek"
    })
    return log_id

def update_log_status(log_id, status, urls=None, message=None):
    update = {
        "$set": {
            "status": status,
            "updated_at": datetime.now(timezone.utc)
        }
    }
    if urls:
        update["$set"]["urls"] = urls
    if message:
        update["$set"]["message"] = message

    collection.update_one({"_id": log_id}, update)