image library migration step 1

hub/services/models/providers.py

@@ -4,9 +4,10 @@ from django.utils.text import slugify
 from django_prose_editor.fields import ProseEditorField
 
 from .base import validate_image_size
+from .images import ImageReference
 
 
-class CloudProvider(models.Model):
+class CloudProvider(ImageReference):
     name = models.CharField(max_length=100)
     slug = models.SlugField(unique=True)
     description = ProseEditorField()
@@ -15,6 +16,7 @@ class CloudProvider(models.Model):
     phone = models.CharField(max_length=25, blank=True, null=True)
     email = models.EmailField(max_length=254, blank=True, null=True)
     address = models.TextField(max_length=250, blank=True, null=True)
+    # Original logo field - keep temporarily for migration
     logo = models.ImageField(
         upload_to="cloud_provider_logos/",
         validators=[validate_image_size],
@@ -39,11 +41,19 @@ class CloudProvider(models.Model):
     def get_absolute_url(self):
         return reverse("services:provider_detail", kwargs={"slug": self.slug})
 
+    @property
+    def get_logo(self):
+        """Returns the logo from library or falls back to legacy logo"""
+        if self.image_library and self.image_library.image:
+            return self.image_library.image
+        return self.logo
 
-class ConsultingPartner(models.Model):
+
+class ConsultingPartner(ImageReference):
     name = models.CharField(max_length=200)
     slug = models.SlugField(unique=True)
     description = ProseEditorField()
+    # Original logo field - keep temporarily for migration
     logo = models.ImageField(
         upload_to="partner_logos/",
         validators=[validate_image_size],
@@ -83,3 +93,10 @@ class ConsultingPartner(models.Model):
 
     def get_absolute_url(self):
         return reverse("services:partner_detail", kwargs={"slug": self.slug})
+
+    @property
+    def get_logo(self):
+        """Returns the logo from library or falls back to legacy logo"""
+        if self.image_library and self.image_library.image:
+            return self.image_library.image
+        return self.logo