SCP-8 Added PhotoNotFoundException to the list of exceptions and to the sentry ignore list.

Author: 2lives

exceptions/__init__.py

@@ -1,3 +1,3 @@
-from .exceptions import JobNotFoundException, StudioException
+from .exceptions import JobNotFoundException, StudioException, PhotoNotFoundException
 
-__all__ = ['JobNotFoundException', 'StudioException']
+__all__ = ['JobNotFoundException', 'StudioException', 'PhotoNotFoundException']

exceptions/exceptions.py

@@ -1,6 +1,9 @@
 class JobNotFoundException(Exception):
     def __init__(self, message):
         super().__init__(message)
+class PhotoNotFoundException(Exception):
+    def __init__(self, message):
+        super().__init__(message)
 
 class StudioException(Exception):
     def __init__(self, status_code, message="Studio exception occurred"):

skylab_studio/studio_client.py

@@ -82,7 +82,7 @@ def __init__(self, api_key=None, **kwargs):
           # of sampled transactions.
           # We recommend adjusting this value in production.
           profiles_sample_rate=1.0,
-          ignore_errors=[JobNotFoundException]
+          ignore_errors=[JobNotFoundException, PhotoNotFoundException]
         )
 
     def _build_http_auth(self):
@@ -508,6 +508,9 @@ async def download_photo(self, photo_id, output_path, profile = None, options =
             await semaphore.acquire()
 
         photo = self.get_photo(photo_id)
+
+        if not 'job' in photo:
+            raise PhotoNotFoundException(f"Unable to find photo with id: {photo_id}")
         profile_id = photo['job']['profileId']
         file_name = photo['name']