From f59c6d379d62ded119f342dfbba7c79f48e39ea2 Mon Sep 17 00:00:00 2001
From: Willem Melching <willem.melching@gmail.com>
Date: Wed, 12 May 2021 16:03:56 +0200
Subject: [PATCH] athenad: catch WebSocketException

---
 selfdrive/athena/athenad.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/selfdrive/athena/athenad.py b/selfdrive/athena/athenad.py
index bd80095675..e71b5b80e6 100755
--- a/selfdrive/athena/athenad.py
+++ b/selfdrive/athena/athenad.py
@@ -17,7 +17,7 @@ from typing import Any
 
 import requests
 from jsonrpc import JSONRPCResponseManager, dispatcher
-from websocket import ABNF, WebSocketTimeoutException, create_connection
+from websocket import ABNF, WebSocketTimeoutException, WebSocketException, create_connection
 
 import cereal.messaging as messaging
 from cereal.services import service_list
@@ -433,7 +433,7 @@ def main():
       handle_long_poll(ws)
     except (KeyboardInterrupt, SystemExit):
       break
-    except (ConnectionError, TimeoutError):
+    except (ConnectionError, TimeoutError, WebSocketException):
       conn_retries += 1
       params.delete("LastAthenaPingTime")
     except Exception: