Просмотр исходного кода

Merge branch '1814-Ability_to_use_object_store_for_images' of github.com:steffann/netbox into 1814-Ability_to_use_object_store_for_images

Sander Steffann 6 лет назад
Родитель
Сommit
f7a1595ce5
1 измененных файлов с 7 добавлено и 4 удалено
  1. 7 4
      netbox/extras/models.py

+ 7 - 4
netbox/extras/models.py

@@ -616,13 +616,16 @@ class ImageAttachment(models.Model):
         """
         from django.conf import settings
         if settings.MEDIA_STORAGE and settings.MEDIA_STORAGE['BACKEND'] == 'S3':
-            from botocore.exceptions import ClientError as AccessError
-        else:
-            AccessError = OSError
+            # For S3 we need to handle a different exception
+            from botocore.exceptions import ClientError
+            try:
+                return self.image.size
+            except ClientError:
+                return None
 
         try:
             return self.image.size
-        except AccessError:
+        except OSError:
             return None