소스 검색

signald.subscribe: send AUTHENTICATION_FAILED if 401

Sumner Evans 3 년 전
부모
커밋
a43ee444f3
1개의 변경된 파일5개의 추가작업 그리고 1개의 파일을 삭제
  1. 5 1
      mausignald/signald.py

+ 5 - 1
mausignald/signald.py

@@ -79,7 +79,11 @@ class SignaldClient(SignaldRPCClient):
         except UnexpectedError as e:
             self.log.debug("Failed to subscribe to %s: %s", username, e)
             evt = WebsocketConnectionStateChangeEvent(
-                state=WebsocketConnectionState.DISCONNECTED,
+                state=(
+                    WebsocketConnectionState.AUTHENTICATION_FAILED
+                    if str(e) == "[401] Authorization failed!"
+                    else WebsocketConnectionState.DISCONNECTED
+                ),
                 account=username,
             )
             await self._run_event_handler(evt)