GitHub/gamdl
1
0
Fork 0
mirror of https://github.com/glomatico/gamdl.git synced 2026-06-13 04:05:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Collect async generators for concurrency

Browse Source
This commit is contained in:
Rafael Moraes
2026-04-24 16:05:37 -03:00
parent 23220d1827
commit 137a739af2
1 changed files with 38 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
gamdl/interface/interface.py
+38 -9
View File
@@ -86,6 +86,14 @@ class AppleMusicInterface:
media.media_id,
)
async def _collect_generator(
self, generator_or_coroutine: AsyncGenerator[AppleMusicMedia, None]
) -> list[AppleMusicMedia]:
results = []
async for result in generator_or_coroutine:
results.append(result)
return results
async def _get_song_media(
self,
media_id: str,
@@ -218,9 +226,16 @@ class AppleMusicInterface:
for index, track in enumerate(tracks)
]
for task in tasks:
async for media in task:
yield media
if self.concurrency == 1:
for task in tasks:
async for media in task:
yield media
else:
collected_tasks = [self._collect_generator(task) for task in tasks]
batches = await safe_gather(*collected_tasks, limit=self.concurrency)
for batch in batches:
for media in batch:
yield media
async def _get_playlist_media(
self,
@@ -282,9 +297,16 @@ class AppleMusicInterface:
for index, track in enumerate(tracks)
]
for task in tasks:
async for media in task:
yield media
if self.concurrency == 1:
for task in tasks:
async for media in task:
yield media
else:
collected_tasks = [self._collect_generator(task) for task in tasks]
batches = await safe_gather(*collected_tasks, limit=self.concurrency)
for batch in batches:
for media in batch:
yield media
async def _get_artist_media(
self,
@@ -380,9 +402,16 @@ class AppleMusicInterface:
)
)
for task in tasks:
async for media in task:
yield media
if self.concurrency == 1:
for task in tasks:
async for media in task:
yield media
else:
collected_tasks = [self._collect_generator(task) for task in tasks]
batches = await safe_gather(*collected_tasks, limit=self.concurrency)
for batch in batches:
for media in batch:
yield media
async def get_media_from_url(
self,