Improved Login

bbsan2k · May 13, 2021 · 3dbb7d0 · 3dbb7d0
1 parent ec940e2
commit 3dbb7d0
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/resources/lib/pyf1tv.py b/resources/lib/pyf1tv.py
@@ -4,6 +4,11 @@
 import time
 import requests
 
+class F1TVError(Exception):
+    pass
+
+class F1TVLoginError(F1TVError):
+    pass
 
 class PYF1TV:
     """API Adapter for F1TV"""
@@ -71,6 +76,8 @@ def _login(self, username, password):
                 json=data,
             )
             # return the ascendontoken
+            if auth_data.status_code != 200:
+                raise F1TVLoginError()
             self.ascendontoken = auth_data.json()["data"]["subscriptionToken"]
             self.auth_data = {
                 "time": int(time.time()),