custom error message if not wharton student

pennlabs · Mar 31, 2019 · dd701ae · dd701ae
1 parent 979c406
commit dd701ae
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/penn/wharton.py b/penn/wharton.py
@@ -54,6 +54,10 @@ def book_reservation(self, sessionid, roomid, start, end):
         format = "%Y-%m-%dT%H:%M:%S-{}".format(self.get_dst_gmt_timezone())
         booking_url = "{}/reserve/{}/{}/?d={}".format(BASE_URL, roomid, start.strftime(format), duration)
         resp = requests.get(booking_url, cookies={"sessionid": sessionid})
+
+        if resp.status_code == 403:
+            return {"success": False, "error": "Your account does not have permission to book Wharton GSRs!"}
+
         resp.raise_for_status()
 
         csrfheader = re.search(r"csrftoken=(.*?);", resp.headers["Set-Cookie"]).group(1)