diff --git a/pull_requests.py b/pull_requests.py index fa55f7e..54e2d23 100644 --- a/pull_requests.py +++ b/pull_requests.py @@ -34,18 +34,22 @@ def get_good_prs(repo: Repository, stats_df: Optional[pd.DataFrame]) -> list[Pul good_prs = [] prs = repo.get_pulls(state="closed") - if stats_df is None: + if stats_df is None or repo.full_name not in stats_df["repo"].unique(): potenially_good_prs = prs number_of_prs = prs.totalCount + from_cached = False else: potenially_good_prs_numbers = stats_df.loc[(stats_df["repo"] == repo.full_name) & (stats_df["has_only_1_comment"] == True)]["pr_number"] potenially_good_prs = [repo.get_pull(n) for n in potenially_good_prs_numbers] number_of_prs = len(potenially_good_prs) + from_cached = True + if number_of_prs == 0: + return [] with tqdm(total=number_of_prs, desc=f"Extracting good PRs from {repo.full_name}", leave=False) as pbar: for pr in potenially_good_prs: - pbar.set_postfix({"found": len(good_prs), "pr_number": pr.number}) + pbar.set_postfix({"found": len(good_prs), "pr_number": pr.number, "from_cached": from_cached}) if pr.merged_at is None: pbar.update(1) continue