Add image uploading feature for organization (#122)

judge/scripts/migrate_organization_image.py

@@ -0,0 +1,64 @@
+# Download organization images from "logo_override_image" and upload to organization_images folder to use "organization_image"
+# In folder online_judge, run python3 manage.py shell < judge/scripts/migrate_organization_image.py
+
+import os
+import requests
+from urllib.parse import urlparse
+from django.core.files.base import ContentFile
+from django.core.files.storage import default_storage
+from django.conf import settings
+from django.db import transaction
+from judge.models import Organization
+
+
+def is_valid_image_url(url):
+    try:
+        parsed_url = urlparse(url)
+        _, ext = os.path.splitext(parsed_url.path)
+        return ext.lower() in [".jpg", ".jpeg", ".png", ".gif", ".svg"]
+    except Exception as e:
+        return False
+
+
+def download_image(url):
+    response = requests.get(url)
+    response.raise_for_status()
+    return ContentFile(response.content)
+
+
+def organization_image_path(organization, filename):
+    tail = filename.split(".")[-1]
+    new_filename = f"organization_{organization.id}.{tail}"
+    return os.path.join(settings.DMOJ_ORGANIZATION_IMAGE_ROOT, new_filename)
+
+
+@transaction.atomic
+def migrate_images():
+    print("Start")
+    organizations = Organization.objects.all()
+    for org in organizations:
+        if org.logo_override_image:
+            if is_valid_image_url(org.logo_override_image):
+                try:
+                    # Download the image
+                    image_content = download_image(org.logo_override_image)
+                    # Determine the file extension
+                    file_ext = org.logo_override_image.split(".")[-1]
+                    filename = f"organization_{org.id}.{file_ext}"
+                    # Save the image to the new location
+                    new_path = organization_image_path(org, filename)
+                    saved_path = default_storage.save(new_path, image_content)
+                    # Update the organization_image field
+                    org.organization_image = saved_path
+                    org.save()
+                    print(f"Image for organization {org.id} migrated successfully.")
+                except Exception as e:
+                    print(f"Failed to migrate image for organization {org.id}: {e}")
+            else:
+                print(
+                    f"Invalid image URL for organization {org.id}: {org.logo_override_image}"
+                )
+    print("Finish")
+
+
+migrate_images()