From c23c258578c5b6fdf40709d762707f130d3af318 Mon Sep 17 00:00:00 2001
From: James Kachel <jkachel@mit.edu>
Date: Fri, 20 Dec 2024 06:27:49 -0800
Subject: [PATCH] Merge main into release (#188)

---
 .../commands/update_product_image_data.py     | 58 +++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 system_meta/management/commands/update_product_image_data.py

diff --git a/system_meta/management/commands/update_product_image_data.py b/system_meta/management/commands/update_product_image_data.py
new file mode 100644
index 00000000..45f7bdc2
--- /dev/null
+++ b/system_meta/management/commands/update_product_image_data.py
@@ -0,0 +1,58 @@
+from django.core.management.base import BaseCommand
+
+from system_meta.models import Product
+from system_meta.tasks import update_products
+
+
+class Command(BaseCommand):
+    """
+    A management command to update image_metadata for all Product objects
+    Example usage: python manage.py update_product_image_data --product_id 1
+    """
+
+    help = "Update image_metadata for all Product objects"
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "--product-id",
+            type=int,
+            help="The ID of the product to update",
+        )
+        parser.add_argument(
+            "--sku",
+            type=str,
+            help="The SKU of the product to update",
+        )
+        parser.add_argument(
+            "--name",
+            type=str,
+            help="The name of the product to update",
+        )
+        parser.add_argument(
+            "--system-name",
+            type=str,
+            help="The system name of the product to update",
+        )
+
+    def handle(self, *args, **kwargs):  # noqa: ARG002
+        product_id = kwargs.get("product_id")
+        sku = kwargs.get("sku")
+        name = kwargs.get("name")
+        system_name = kwargs.get("system_name")
+
+        if product_id:
+            products = Product.objects.filter(id=product_id)
+        elif sku:
+            products = Product.objects.filter(sku=sku, system__name=system_name)
+        elif name:
+            products = Product.objects.filter(name=name)
+        else:
+            products = Product.objects.all()
+
+        for product in products:
+            update_products.delay(product.id)
+            self.stdout.write(
+                self.style.SUCCESS(
+                    f"Successfully updated image metadata for product {product.id}"
+                )
+            )