Fix Source.DoesNotExist after migrating ImageField (#123)

ui · Jul 24, 2024 · 13fe83f · 13fe83f
1 parent 4970206
commit 13fe83f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/thumbnails/backends/metadata.py b/thumbnails/backends/metadata.py
@@ -76,7 +76,12 @@ def get_thumbnail(self, source_name, size):
             return None
 
     def add_thumbnail(self, source_name, size, name):
-        source = self.get_source(source_name)
+        try:
+            source = self.get_source(source_name)
+        except Source.DoesNotExist:
+            # If the source doesn't exist, create it
+            # For example when migrating from a regular ImageField to a thumbnailed ImageField
+            source = self.add_source(source_name)
         meta = ThumbnailMeta.objects.create(source=source, size=size, name=name)
         return ImageMeta(source_name, meta.name, meta.size)