瀏覽代碼

Fix catching subscribe/unsubscribe errors

Tulir Asokan 3 年之前
父節點
當前提交
a26576bdc2
共有 1 個文件被更改,包括 3 次插入3 次删除
  1. 3 3
      mausignald/signald.py

+ 3 - 3
mausignald/signald.py

@@ -8,7 +8,7 @@ import asyncio
 
 from mautrix.util.logging import TraceLogger
 
-from .errors import UnexpectedError, UnexpectedResponse
+from .errors import RPCError, UnexpectedResponse
 from .rpc import CONNECT_EVENT, DISCONNECT_EVENT, SignaldRPCClient
 from .types import (
     Account,
@@ -98,7 +98,7 @@ class SignaldClient(SignaldRPCClient):
             await self.request_v1("subscribe", account=username)
             self._subscriptions.add(username)
             return True
-        except UnexpectedError as e:
+        except RPCError as e:
             self.log.debug("Failed to subscribe to %s: %s", username, e)
             evt = WebsocketConnectionStateChangeEvent(
                 state=(
@@ -116,7 +116,7 @@ class SignaldClient(SignaldRPCClient):
             await self.request_v1("unsubscribe", account=username)
             self._subscriptions.remove(username)
             return True
-        except UnexpectedError as e:
+        except RPCError as e:
             self.log.debug("Failed to unsubscribe from %s: %s", username, e)
             return False