karma/crab
1
0
Fork 0
mirror of https://github.com/karma-riuk/crab.git synced 2025-07-05 13:38:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

improved caching

Browse Source
This commit is contained in:
Karma Riuk
2025-03-23 10:54:43 +01:00
parent 2ba71d8523
commit 1aa42c32cd
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pull_requests.py
View File

@ -191,6 +191,13 @@ def process_repo(repo_name: str, stats_df: Optional[pd.DataFrame], dataset: Data
repo = g.get_repo(repo_name)
good_prs = get_good_prs(repo, stats_df)
if repo_name in cache:
for pr_number in tqdm(cache[repo_name], desc="Copying cached entries", leave=False):
dataset.entries.append(cache[repo_name][pr_number])
dataset.to_json(args.output)
good_prs = [pr for pr in good_prs if pr.number not in cache[repo_name]]
with tqdm(good_prs, desc="Processing good prs", leave=False) as pbar:
for pr in pbar:
pbar.set_postfix({"pr": pr.number})
@ -260,6 +267,6 @@ if __name__ == "__main__":
dataset = Dataset()
try:
# try and finally to save, regardless of an error occuring or the program finished correctly
process_repos(df, stats_df, dataset, args.repos)
process_repos(df, stats_df, dataset, args.repos, cache)
finally:
dataset.to_json(args.output)