Spaces:
Sleeping
Sleeping
File size: 998 Bytes
d05d972 9a82703 d05d972 | 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 | 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)
|