tobast/matrix-alertbot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix context managers

Browse source
This commit is contained in:
HgO 2022-07-08 22:38:51 +02:00
parent eda0dc8193
commit d8ab7c1d9b
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
matrix_alertbot/alertmanager.py
View file

@ -22,8 +22,8 @@ class AlertmanagerClient(AsyncContextManager):
self.cache = cache
self.session = aiohttp.ClientSession()
def __aenter__(self) -> AlertmanagerClient:
return self
async def __aenter__(self) -> AlertmanagerClient:
return await self
async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
await super().__aexit__(*args, **kwargs)

4
matrix_alertbot/webhook.py
View file

@ -65,8 +65,8 @@ class Webhook(AsyncContextManager):
self.port = config.port
self.socket = config.socket
def __aenter__(self) -> Webhook:
return self
async def __aenter__(self) -> Webhook:
return await self
async def __aexit__(self, *args: Any, **kwargs: Any) -> None:
await super().__aexit__(*args, **kwargs)