Handle http status codes 401 or 504

BaQs · Mar 8, 2021 · f6b36ed · f6b36ed
1 parent 75aea92
commit f6b36ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyezviz/client.py b/pyezviz/client.py
@@ -251,7 +251,7 @@ def get_alarminfo(self, serial, max_retries=0):
             req.raise_for_status()
 
         except requests.HTTPError as err:
-            if err.response.status_code == 401 or 501:
+            if err.response.status_code == 401 or 504:
                 # session is wrong, need to relogin
                 self.login()
                 logging.info(