Merge pull request #2040 from MachciaD/vassert-except-assert

Vassert except assert

instagrapi/exceptions.py

@@ -325,3 +325,7 @@ class InvalidMediaId(PrivateError):
 
 class MediaUnavailable(PrivateError):
     """Media is unavailable"""
+
+
+class ValidationError(AssertionError):
+    pass

instagrapi/mixins/highlight.py

@@ -9,7 +9,7 @@
 from instagrapi.exceptions import HighlightNotFound
 from instagrapi.extractors import extract_highlight_v1
 from instagrapi.types import Highlight
-from instagrapi.utils import dumps
+from instagrapi.utils import dumps, vassert
 
 
 class HighlightMixin:
@@ -31,7 +31,7 @@ def highlight_pk_from_url(self, url: str) -> str:
         --------
         https://www.instagram.com/stories/highlights/17895485201104054/ -> 17895485201104054
         """
-        assert "/highlights/" in url, 'URL must contain "/highlights/"'
+        vassert("/highlights/" in url, "URL must contain '/highlights/'")
         path = urlparse(url).path
         parts = [p for p in path.split("/") if p and p.isdigit()]
         return str(parts[0])

instagrapi/utils.py

@@ -5,6 +5,7 @@
 import string
 import time
 import urllib
+from exceptions import ValidationError
 
 
 class InstagramIdCodec:
@@ -107,3 +108,8 @@ def date_time_original(localtime):
 def random_delay(delay_range: list):
     """Trigger sleep of a random floating number in range min_sleep to max_sleep"""
     return time.sleep(random.uniform(delay_range[0], delay_range[1]))
+
+
+def vassert(pred, message):
+    if not pred:
+        raise ValidationError(message)