Matrix-Mirrors/pantalaimon
1
0
Fork 0
mirror of https://github.com/matrix-org/pantalaimon.git synced 2025-03-10 16:30:06 -04:00
Code Issues Packages Projects Releases Wiki Activity

pantalaimon: Catch all aiohttp client connection errors.

Browse Source
This commit is contained in:
Damir Jelić 2019-05-02 13:49:48 +02:00
parent f1fefb952c
commit e84bcea389
2 changed files with 8 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pantalaimon/client.py
View File

@ -2,8 +2,7 @@ import asyncio
from pprint import pformat
from typing import Any, Dict, Optional
from aiohttp.client_exceptions import (ClientProxyConnectionError,
ServerDisconnectedError)
from aiohttp.client_exceptions import ClientConnectionError
from nio import (AsyncClient, ClientConfig, EncryptionError,
GroupEncryptionError, KeysQueryResponse, MegolmEvent,
RoomEncryptedEvent, SyncResponse,
@ -212,8 +211,7 @@ class PanClient(AsyncClient):
break
except (
ClientProxyConnectionError,
ServerDisconnectedError,
ClientConnectionError,
ConnectionRefusedError
):
try:

23
pantalaimon/daemon.py
View File

@ -18,8 +18,7 @@ import keyring
import logbook
from aiohttp import ClientSession, web
from aiohttp.client_exceptions import (ContentTypeError,
ClientProxyConnectionError,
ServerDisconnectedError)
ClientConnectionError)
from appdirs import user_data_dir
from logbook import StderrHandler
from multidict import CIMultiDict
@ -322,9 +321,7 @@ class ProxyDaemon:
status=response.status,
text=await response.text()
)
except (ClientProxyConnectionError,
ServerDisconnectedError,
ConnectionRefusedError) as e:
except ClientConnectionError as e:
return web.Response(status=500, text=str(e))
async def router(self, request):
@ -409,9 +406,7 @@ class ProxyDaemon:
try:
response = await self.forward_request(request)
except (ClientProxyConnectionError,
ServerDisconnectedError,
ConnectionRefusedError) as e:
except ClientConnectionError as e:
return web.Response(status=500, text=str(e))
try:
@ -522,9 +517,7 @@ class ProxyDaemon:
params=query,
token=client.access_token
)
except (ClientProxyConnectionError,
ServerDisconnectedError,
ConnectionRefusedError) as e:
except ClientConnectionError as e:
return web.Response(status=500, text=str(e))
if response.status == 200:
@ -556,9 +549,7 @@ class ProxyDaemon:
try:
response = await self.forward_request(request)
except (ClientProxyConnectionError,
ServerDisconnectedError,
ConnectionRefusedError) as e:
except ClientConnectionError as e:
return web.Response(status=500, text=str(e))
if response.status == 200:
@ -613,9 +604,7 @@ class ProxyDaemon:
except GroupEncryptionError:
await client.share_group_session(room_id)
response = await client.room_send(room_id, msgtype, content, txnid)
except (ClientProxyConnectionError,
ServerDisconnectedError,
ConnectionRefusedError) as e:
except ClientConnectionError as e:
return web.Response(status=500, text=str(e))
return web.Response(