diff --git a/tools/readme_generator/regen_readme_in_batch.py b/tools/readme_generator/regen_readme_in_batch.py index 477f1cc8db..fc6ae2c9d1 100644 --- a/tools/readme_generator/regen_readme_in_batch.py +++ b/tools/readme_generator/regen_readme_in_batch.py @@ -4,6 +4,7 @@ import shlex import asyncio import tempfile +import requests from make_readme import generate_READMEs from pathlib import Path @@ -44,6 +45,19 @@ async def regen_readme(repository, branch): print(f"{repository} -> branch '{branch}'") print("=" * len(f"{repository} -> branch '{branch}'")) + branches = requests.get( + f"https://api.github.com/repos/{repository}/branches", + headers={ + "Authorization": f"Bearer {token}", + "X-GitHub-Api-Version": "2022-11-28", + "Accept": "application/vnd.github+json", + } + ).json() + + branches = {x["name"] for x in branches} + if "testing" in branches: + branch = "testing" + with tempfile.TemporaryDirectory() as folder: await git(["clone", f"https://{login}:{token}@github.com/{repository}", "--single-branch", "--branch", branch, folder])