Merge pull request #352 from desecho/sentry-block

Add SentryBlockException
ybw0619 · Mar 26, 2018 · cf0fe1d · cf0fe1d
2 parents b6f58a9 + f21becf
commit cf0fe1d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/InstagramAPI/InstagramAPI.py b/InstagramAPI/InstagramAPI.py
@@ -36,6 +36,8 @@
     # Issue 159, python3 import fix
     from .ImageUtils import getImageSize
 
+from .exceptions import SentryBlockException
+
 
 class InstagramAPI:
     API_URL = 'https://i.instagram.com/api/v1/'
@@ -978,6 +980,10 @@ def SendRequest(self, endpoint, post=None, login=False):
                 self.LastResponse = response
                 self.LastJson = json.loads(response.text)
                 print(self.LastJson)
+                if 'error_type' in self.LastJson and self.LastJson['error_type'] == 'sentry_block':
+                    raise SentryBlockException(self.LastJson['message'])
+            except SentryBlockException:
+                raise
             except:
                 pass
             return False

diff --git a/InstagramAPI/exceptions.py b/InstagramAPI/exceptions.py
@@ -0,0 +1,2 @@
+class SentryBlockException(Exception):
+    pass