Fix KeyError when login fails on some occasions

iamkroot · Apr 8, 2020 · e306dd5 · e306dd5
1 parent 04b4191
commit e306dd5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/ilc_scrape.py b/ilc_scrape.py
@@ -198,10 +198,14 @@ def login(username, password):
     except (requests.ConnectionError, requests.Timeout) as e:
         print_quit(f"Connection Error {e}")
     if response.status_code >= 500:
-        print_quit("Impartus not responding properly", 1)
+        print_quit("Impartus not responding properly")
+    elif response.status_code == 400:
+        print_quit("Invalid login request. Impartus changed something.")
+    elif response.status_code == 401:
+        print_quit("Invalid login credentials!")
     resp = response.json()
     if not resp["success"]:
-        print_quit("Impartus:" + resp["message"], 1)
+        print_quit(f"Impartus: {resp.get('message', resp)}", 1)
     return resp["token"]
 
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ilc-scraper"
-version = "0.6.2"
+version = "0.6.3"
 description = "A scraper for Impartus Lecture Capture videos for BITS Hyderabad"
 authors = ["iamkroot <kroot.patel@gmail.com>"]
 readme = "README.md"