Browse Source

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 years ago
parent
commit
f7a1595ce5
1 changed files with 7 additions and 4 deletions
  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
         from django.conf import settings
         if settings.MEDIA_STORAGE and settings.MEDIA_STORAGE['BACKEND'] == 'S3':
         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:
         try:
             return self.image.size
             return self.image.size
-        except AccessError:
+        except OSError:
             return None
             return None