feat: add progress bar while collecting issues

This commit is contained in:
Paul Gauthier (aider) 2024-09-21 16:05:16 -07:00
parent 10fee78ddd
commit de87418647

View file

@ -24,18 +24,20 @@ headers = {"Authorization": f"token {TOKEN}", "Accept": "application/vnd.github.
def get_issues(state="open"): def get_issues(state="open"):
issues = [] issues = []
page = 1 page = 1
while True: with tqdm(desc="Collecting issues", unit="page") as pbar:
response = requests.get( while True:
f"{GITHUB_API_URL}/repos/{REPO_OWNER}/{REPO_NAME}/issues", response = requests.get(
headers=headers, f"{GITHUB_API_URL}/repos/{REPO_OWNER}/{REPO_NAME}/issues",
params={"state": state, "page": page, "per_page": 100}, headers=headers,
) params={"state": state, "page": page, "per_page": 100},
response.raise_for_status() )
page_issues = response.json() response.raise_for_status()
if not page_issues: page_issues = response.json()
break if not page_issues:
issues.extend(page_issues) break
page += 1 issues.extend(page_issues)
page += 1
pbar.update(1)
return issues return issues