Replace generators with functions returning lists

m-aciek · m-aciek · commit 1e79cf74f650 · 2023-06-07T00:39:12.000+02:00
diff --git a/README.md b/README.md
@@ -3,7 +3,7 @@ Polskie tłumaczenie dokumentacji Pythona
 <!-- [[[cog
 from manage_translation import get_resource_language_stats, progress_from_resources, language_switcher, get_number_of_translators
 
-stats = list(get_resource_language_stats())
+stats = get_resource_language_stats()
 switcher = progress_from_resources(filter(language_switcher, stats))
 total = progress_from_resources(stats)
 translators = get_number_of_translators()
diff --git a/manage_translation.py b/manage_translation.py
@@ -119,17 +119,17 @@ def _get_tx_token() -> str:
     return transifex_api_key
 
 
-def _get_resources() -> Generator[transifex_api.Resource, None, None]:
+def _get_resources() -> list[transifex_api.Resource]:
     transifex_api.setup(auth=_get_tx_token())
-    yield from transifex_api.Resource.filter(project=f'o:python-doc:p:{PROJECT_SLUG}').all()
+    return transifex_api.Resource.filter(project=f'o:python-doc:p:{PROJECT_SLUG}').all()
 
 
-def get_resource_language_stats() -> Generator[ResourceLanguageStatistics, None, None]:
+def get_resource_language_stats() -> list[ResourceLanguageStatistics]:
     transifex_api.setup(auth=_get_tx_token())
     resources = transifex_api.ResourceLanguageStats.filter(
         project=f'o:python-doc:p:{PROJECT_SLUG}', language=f'l:{LANGUAGE}'
     ).all()
-    yield from (ResourceLanguageStatistics.from_api_entry(entry) for entry in resources)
+    return [ResourceLanguageStatistics.from_api_entry(entry) for entry in resources]
 
 
 def progress_from_resources(resources: Iterable[ResourceLanguageStatistics]) -> float:
