Add custom exception for Presentation Links

Author: jhodges10

frameioclient/client.py

@@ -1,9 +1,11 @@
+import re
 import sys
 import requests
 import warnings
 from requests.adapters import HTTPAdapter
 from requests.packages.urllib3.util.retry import Retry
 from .download import FrameioDownloader
+from .exceptions import PresentationException
 
 if sys.version_info.major >= 3:
   from .py3_uploader import FrameioUploader
@@ -75,6 +77,9 @@ def _api_call(self, method, endpoint, payload={}):
           )
 
       return r.json()
+    
+    if r.status_code == 422 and "presentation" in endpoint:
+      raise PresentationException
 
     return r.raise_for_status()
 

frameioclient/exceptions.py

@@ -0,0 +1,15 @@
+
+class PresentationException(Exception):
+    """Exception raised when you try to create a Presentation Link for an asset
+    that already has one.
+
+    Attributes:
+        asset_id -- input asset ID
+    """
+
+    def __init__(
+        self, 
+        message="Your asset already has a presentation link associated with it."
+    ):
+        self.message = message
+        super().__init__(self.message)