SCP-8 added sentry and updated backoff on photo upload to bucket

Author: 2lives

skylab_studio/studio_client.py

@@ -14,7 +14,8 @@
 import base64
 import hashlib
 import requests
-from io import BytesIO
+# from io import BytesIO
+import sentry_sdk
 
 from .version import VERSION
 from .studio_exception import StudioException
@@ -62,7 +63,7 @@ def __init__(self, api_key=None, **kwargs):
 
         if 'debug' in kwargs:
             self.debug = kwargs['debug']
-        
+
         if 'max_concurrent_downloads' in kwargs:
             self.max_concurrent_downloads = kwargs['max_concurrent_downloads']
 
@@ -72,6 +73,18 @@ def __init__(self, api_key=None, **kwargs):
             LOGGER.debug('Debug enabled')
             LOGGER.propagate = True
 
+        # initialize sentry
+        sentry_sdk.init(
+          dsn="https://[email protected]/4507850876452864",
+          # Set traces_sample_rate to 1.0 to capture 100%
+          # of transactions for tracing.
+          traces_sample_rate=1.0,
+          # Set profiles_sample_rate to 1.0 to profile 100%
+          # of sampled transactions.
+          # We recommend adjusting this value in production.
+          profiles_sample_rate=1.0,
+        )
+
     def _build_http_auth(self):
         return (self.api_key, '')
 
@@ -331,6 +344,8 @@ def _upload_photo(self, photo_path, id, model='job'):
 
         try:
           upload_photo_resp = requests.put(upload_url, data, headers=headers)
+          # Will raise exception for any statuses 4xx-5xx
+          upload_photo_resp.raise_for_status()
 
           if not upload_photo_resp:
             print('First upload attempt failed, retrying...')
@@ -347,9 +362,15 @@ def _upload_photo(self, photo_path, id, model='job'):
                     time.sleep(1)  # Wait for a moment before retrying
                     retry += 1
 
+        except requests.exceptions.HTTPError as e:
+            sentry_sdk.capture_exception(e)
+            print("HTTP error occurred while trying to upload your photo:", e)
+
         except Exception as e:
+          sentry_sdk.capture_exception(e)
           print(f"An exception of type {type(e).__name__} occurred: {e}")
           print('Deleting created, but unuploaded photo...')
+
           self.delete_photo(photo_id)
 
         res['upload_response'] = upload_photo_resp.status_code