🎨 Change stub format

app/stub.py

@@ -44,6 +44,39 @@ def populate_data():
         "password": get_password_hash("test")
     }).inserted_id
 
+    # Créer d'abord les pins
+    pin_a = pins_collection.insert_one({
+        "title": "Tour Eiffel",
+        "description": "Description A",
+        "location": [48.858296, 2.294526],
+        "files": [],
+        "user_id": str(user1_id)
+    })
+
+    pin_b = pins_collection.insert_one({
+        "title": "Mont St Michel",
+        "description": "Description B",
+        "location": [48.636111, -1.511389],
+        "files": [],
+        "user_id": str(user1_id)
+    })
+
+    pin_x = pins_collection.insert_one({
+        "title": "Eiffel Tower",
+        "description": "Description X",
+        "location": [48.858296, 2.294526],
+        "files": [],
+        "user_id": str(user2_id)
+    })
+
+    pin_y = pins_collection.insert_one({
+        "title": "Mont Saint Michel",
+        "description": "Description Y",
+        "location": [48.636111, -1.511389],
+        "files": [],
+        "user_id": str(user2_id)
+    })
+
     # Créer les images de test
     image_a = create_test_image(color='red')
     image_b = create_test_image(color='blue')
@@ -56,13 +89,14 @@ def populate_data():
     image_x_path = save_image(image_x, f"{uuid.uuid4()}.jpg")
     image_y_path = save_image(image_y, f"{uuid.uuid4()}.jpg")
 
-    # Insérer les métadonnées des images dans la base de données
+    # Insérer les métadonnées des images dans la base de données avec leur pin_id
     image_a_id = images_collection.insert_one({
         "path": image_a_path,
         "content_type": "image/jpeg",
         "exif_date": datetime.now(),
         "caption": "Tour Eiffel",
-        "user_id": str(user1_id)
+        "user_id": str(user1_id),
+        "pin_id": str(pin_a.inserted_id)
     }).inserted_id
 
     image_b_id = images_collection.insert_one({
@@ -70,7 +104,8 @@ def populate_data():
         "content_type": "image/jpeg",
         "exif_date": datetime.now(),
         "caption": "Mont St Michel",
-        "user_id": str(user1_id)
+        "user_id": str(user1_id),
+        "pin_id": str(pin_b.inserted_id)
     }).inserted_id
 
     image_x_id = images_collection.insert_one({
@@ -78,7 +113,8 @@ def populate_data():
         "content_type": "image/jpeg",
         "exif_date": datetime.now(),
         "caption": "Eiffel Tower",
-        "user_id": str(user2_id)
+        "user_id": str(user2_id),
+        "pin_id": str(pin_x.inserted_id)
     }).inserted_id
 
     image_y_id = images_collection.insert_one({
@@ -86,40 +122,27 @@ def populate_data():
         "content_type": "image/jpeg",
         "exif_date": datetime.now(),
         "caption": "Mont Saint Michel",
-        "user_id": str(user2_id)
+        "user_id": str(user2_id),
+        "pin_id": str(pin_y.inserted_id)
     }).inserted_id
 
-    # Créer les pins avec les vraies images
-    pins_collection.insert_many([
-        {
-            "title": "Tour Eiffel",
-            "description": "Description A",
-            "location": [48.858296, 2.294526],
-            "files": [str(image_a_id)],
-            "user_id": str(user1_id)
-        },
-        {
-            "title": "Mont St Michel",
-            "description": "Description B",
-            "location": [48.636111, -1.511389],
-            "files": [str(image_b_id)],
-            "user_id": str(user1_id)
-        },
-        {
-            "title": "Eiffel Tower",
-            "description": "Description X",
-            "location": [48.858296, 2.294526],
-            "files": [str(image_x_id)],
-            "user_id": str(user2_id)
-        },
-        {
-            "title": "Mont Saint Michel",
-            "description": "Description Y",
-            "location": [48.636111, -1.511389],
-            "files": [str(image_y_id)],
-            "user_id": str(user2_id)
-        }
-    ])
+    # Mettre à jour les pins avec les IDs des images
+    pins_collection.update_one(
+        {"_id": pin_a.inserted_id},
+        {"$set": {"files": [str(image_a_id)]}}
+    )
+    pins_collection.update_one(
+        {"_id": pin_b.inserted_id},
+        {"$set": {"files": [str(image_b_id)]}}
+    )
+    pins_collection.update_one(
+        {"_id": pin_x.inserted_id},
+        {"$set": {"files": [str(image_x_id)]}}
+    )
+    pins_collection.update_one(
+        {"_id": pin_y.inserted_id},
+        {"$set": {"files": [str(image_y_id)]}}
+    )
     
     friends_collection = db["friends"]
     friends_collection.insert_one({