mirrors/pferd
1
0
Fork 0
mirror of https://github.com/Garmelon/PFERD.git synced 2023-12-21 10:23:01 +01:00
Code Issues Packages Projects Releases Wiki Activity

Clean up and fix conductor and limiter

Browse Source 
Turns out you have to await an async lock, who knew...
This commit is contained in:
Joscha
2021-04-29 13:43:50 +02:00
parent c4fb92c658
commit 7e127cd5cc
2 changed files with 14 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
PFERD/limiter.py
View File

@ -1,7 +1,6 @@
import asyncio
from contextlib import asynccontextmanager
# TODO If we upgrade to python 3.9, this context manager hint is deprecated
from typing import AsyncContextManager, AsyncIterator
from typing import AsyncIterator
class Limiter:
@ -9,12 +8,9 @@ class Limiter:
self._semaphore = asyncio.Semaphore(limit)
@asynccontextmanager
async def _context_manager(self) -> AsyncIterator[None]:
async def limit(self) -> AsyncIterator[None]:
await self._semaphore.acquire()
try:
yield
finally:
self._semaphore.release()
def limit(self) -> AsyncContextManager[None]:
return self._context_manager()