Zergling_man/moment
Zergling_man/moment
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Limit concurrent thumbnail downloads to 8

Browse Source
This commit is contained in:
miruka 2019-07-18 20:49:47 -04:00
parent 9decd58992
commit 329c34cf94
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/python/image_provider.py
View File

@ -20,6 +20,8 @@ from nio.api import ResizingMethod
Size = Tuple[int, int] Size = Tuple[int, int]
ImageData = Tuple[bytearray, Size, int] # last int: pyotherside format enum ImageData = Tuple[bytearray, Size, int] # last int: pyotherside format enum
CONCURRENT_DOWNLOADS_LIMIT = asyncio.BoundedSemaphore(8)
@dataclass @dataclass
class Thumbnail: class Thumbnail:
@ -85,13 +87,14 @@ class Thumbnail:
) )
parsed = urlparse(self.mxc) parsed = urlparse(self.mxc)
response = await client.thumbnail( async with CONCURRENT_DOWNLOADS_LIMIT:
server_name = parsed.netloc, response = await client.thumbnail(
media_id = parsed.path.lstrip("/"), server_name = parsed.netloc,
width = self.server_size[0], media_id = parsed.path.lstrip("/"),
height = self.server_size[1], width = self.server_size[0],
method = self.resize_method, height = self.server_size[1],
) method = self.resize_method,
)
if isinstance(response, nio.ThumbnailError): if isinstance(response, nio.ThumbnailError):
# Return a transparent 1x1 PNG # Return a transparent 1x1 PNG