diff --git a/app/routes/pins.py b/app/routes/pins.py index 6570cd4..e4c58ff 100644 --- a/app/routes/pins.py +++ b/app/routes/pins.py @@ -20,6 +20,7 @@ db = client[config.MONGODB_DATABASE] pins_collection = db["pins"] friends_collection = db["friends"] pin_permissions_collection = db["pin_permissions"] +images_collection = db["images"] pins_router = APIRouter( prefix="/pin", @@ -85,6 +86,14 @@ async def update_pin(id: str, pin: PinDTO, current_user: User = Depends(get_curr async def add_pin(pin: PinDTO, current_user: User = Depends(get_current_user)): pin.user_id = current_user.uid pin_id = pins_collection.insert_one(pin.model_dump()).inserted_id + + # Mettre à jour les images avec le pin_id + if pin.files: + images_collection.update_many( + {"_id": {"$in": [ObjectId(img_id) for img_id in pin.files]}}, + {"$set": {"pin_id": pin_id}} + ) + return {"id": str(pin_id)} @pins_router.get(