From 302791b15e81918c7443ee1730af858b38f8d3d2 Mon Sep 17 00:00:00 2001
From: Jason Heppler <jason.heppler@gmail.com>
Date: Fri, 26 Apr 2024 10:56:54 -0500
Subject: [PATCH] refactor: populate_images.py to use filter instead of get for
 Object queries

---
 .../management/commands/populate_images.py    | 35 ++++++++++---------
 1 file changed, 18 insertions(+), 17 deletions(-)

diff --git a/postcards/management/commands/populate_images.py b/postcards/management/commands/populate_images.py
index 4e9dfd0..7a856d0 100644
--- a/postcards/management/commands/populate_images.py
+++ b/postcards/management/commands/populate_images.py
@@ -34,24 +34,25 @@ def handle(self, *args, **options):
                 ).strip()  # Join the parts of the item_id back into a string
                 caption = parts[-1].strip()  # The last part is the caption
                 try:
-                    obj = Object.objects.get(item_id=item_id)
+                    obj = Object.objects.filter(item_id=item_id)
                     with open(os.path.join(filepath, filename), "rb") as f:
-                        if caption == "Front":
-                            img = Image(
-                                postcard=obj, image_caption="Front"
-                            )  # Set the image_caption field to 'Front'
-                        elif caption == "Reverse":
-                            img = Image(
-                                postcard=obj, image_caption="Reverse"
-                            )  # Set the image_caption field to 'Reverse'
-                        else:
-                            img = Image(
-                                postcard=obj
-                            )  # If neither 'Front' nor 'Reverse' is in the filename, don't set the image_caption field
-                        img.image.save(
-                            filename, File(f)
-                        )  # Save the image file to the Image instance
-                        img.save()  # Save the Image instance
+                        for mail in obj:
+                            if caption == "Front":
+                                img = Image(
+                                    postcard=mail, image_caption="Front"
+                                )  # Set the image_caption field to 'Front'
+                            elif caption == "Reverse":
+                                img = Image(
+                                    postcard=mail, image_caption="Reverse"
+                                )  # Set the image_caption field to 'Reverse'
+                            else:
+                                img = Image(
+                                    postcard=mail
+                                )  # If neither 'Front' nor 'Reverse' is in the filename, don't set the image_caption field
+                            img.image.save(
+                                filename, File(f)
+                            )  # Save the image file to the Image instance
+                            img.save()  # Save the Image instance
                         self.stdout.write(
                             self.style.SUCCESS(
                                 f"Successfully updated image for Object "