diff --git a/utils.py b/utils.py index 2e51f84..922bac8 100644 --- a/utils.py +++ b/utils.py @@ -37,14 +37,16 @@ def has_only_1_round_of_comments(commits: PaginatedList[Commit], comments: Pagin return n_before >= 1 and n_after >= 1 -def has_only_1_comment(commits: PaginatedList[Commit], comments: PaginatedList[PullRequestComment]): +def has_only_1_comment(commits: PaginatedList[Commit], comments: PaginatedList[PullRequestComment], verbose: bool = False): if ( comments is None or commits is None or comments.totalCount == 0 or commits.totalCount == 0 ): + if verbose: print(f"No comments or commits: {comments.totalCount} comments, {commits.totalCount} commits") return False if comments.totalCount != 1: + if verbose: print(f"More than 1 comment: {comments.totalCount} comments") return False commit_dates = [commit.commit.author.date for commit in tqdm(commits, total=commits.totalCount, desc="Extracting date from commits", leave=False)] @@ -59,4 +61,5 @@ def has_only_1_comment(commits: PaginatedList[Commit], comments: PaginatedList[P if commit_date > comment_date: n_after += 1 continue + if verbose: print(f"n_before: {n_before}, n_after: {n_after}") return n_before >= 1 and n_after >= 1