Convert main loop into iterator

Author: m-aciek

generate.py

@@ -9,6 +9,7 @@
 # ]
 # ///
 import subprocess
+from collections.abc import Iterator
 from datetime import datetime, timezone
 from pathlib import Path
 from tempfile import TemporaryDirectory
@@ -24,51 +25,55 @@
 completion_progress = []
 generation_time = datetime.now(timezone.utc)
 
-with TemporaryDirectory() as clones_dir:
-    Repo.clone_from(
-        'https://github.com/python/devguide.git',
-        devguide_dir := Path(clones_dir, 'devguide'),
-        depth=1,
-    )
-    latest_branch = branches_from_devguide(devguide_dir)[0]
-    Repo.clone_from(
-        'https://github.com/python/cpython.git',
-        Path(clones_dir, 'cpython'),
-        depth=1,
-        branch=latest_branch,
-    )
-    subprocess.run(['make', '-C', Path(clones_dir, 'cpython/Doc'), 'venv'], check=True)
-    subprocess.run(
-        ['make', '-C', Path(clones_dir, 'cpython/Doc'), 'gettext'], check=True
-    )
-    languages_built = dict(build_status.get_languages())
-    for language, repo in repositories.get_languages_and_repos(devguide_dir):
-        built_on_docs_python_org = language in languages_built
-        if repo:
+
+def get_completion_progress() -> (
+    Iterator[tuple[str, str, float, int, int, bool, bool | None]]
+):
+    with TemporaryDirectory() as clones_dir:
+        Repo.clone_from(
+            'https://github.com/python/devguide.git',
+            devguide_dir := Path(clones_dir, 'devguide'),
+            depth=1,
+        )
+        latest_branch = branches_from_devguide(devguide_dir)[0]
+        Repo.clone_from(
+            'https://github.com/python/cpython.git',
+            Path(clones_dir, 'cpython'),
+            depth=1,
+            branch=latest_branch,
+        )
+        subprocess.run(
+            ['make', '-C', Path(clones_dir, 'cpython/Doc'), 'venv'], check=True
+        )
+        subprocess.run(
+            ['make', '-C', Path(clones_dir, 'cpython/Doc'), 'gettext'], check=True
+        )
+        languages_built = dict(build_status.get_languages())
+        for language, repo in repositories.get_languages_and_repos(devguide_dir):
+            built_on_docs_python_org = language in languages_built
+            in_switcher = languages_built.get(language)
+            if not repo:
+                yield language, repo, 0.0, 0, 0, built_on_docs_python_org, in_switcher
             completion_number, translators_number = get_completion(clones_dir, repo)
             if built_on_docs_python_org:
                 visitors_number = visitors.get_number_of_visitors(language)
             else:
                 visitors_number = 0
-        else:
-            completion_number, translators_number, visitors_number = 0.0, 0, 0
-        completion_progress.append(
-            (
+            yield (
                 language,
                 repo,
                 completion_number,
                 translators_number,
                 visitors_number,
                 built_on_docs_python_org,
-                languages_built.get(language),  # in switcher
+                in_switcher,
             )
-        )
-        print(completion_progress[-1])
+
 
 template = Template(Path('template.html').read_text())
 
 output = template.render(
-    completion_progress=completion_progress, generation_time=generation_time
+    completion_progress=get_completion_progress(), generation_time=generation_time
 )
 
 Path('index.html').write_text(output)