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

now archiving both before (context for AI) and after (tests for benchmark)

Browse Source
This commit is contained in:
Karma Riuk
2025-03-31 21:50:11 +02:00
parent d8ab48dc82
commit e697890395
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
pull_requests.py
View File

@ -197,7 +197,7 @@ def get_comments(pr: PullRequest) -> list[Comment]:
return ret
def archive_repo(repo_path: str, pr_number: int, destination: str) -> None:
def archive_repo(repo_path: str, pr_number: int, destination: str, post_fix: str) -> None:
"""
Archives the repo at the specified path, including only the files tracked by git.
The archive is stored in the destination directory with a filename based on the PR number.
@ -205,7 +205,7 @@ def archive_repo(repo_path: str, pr_number: int, destination: str) -> None:
if not os.path.exists(destination):
os.makedirs(destination)
archive_name = f"repo_pr_{pr_number}.tar.gz"
archive_name = f"{repo_path.replace('/', '_')}_{pr_number}_{post_fix}.tar.gz"
archive_path = os.path.join(destination, archive_name)
result = run_git_cmd(["ls-files"], repo_path)
@ -274,11 +274,18 @@ def process_pull(
lambda: entry.comments.extend(get_comments(pr)),
),
("Checkout out base commit...", lambda: checkout(repo_path, pr.base.sha, pr.number)),
("Archiving the repo...", lambda: archive_repo(repo_path, pr.number, archive_destination)),
(
"Archiving the repo...",
lambda: archive_repo(repo_path, pr.number, archive_destination, "base"),
),
(
"Checkout out merge commit...",
lambda: checkout(repo_path, pr.merge_commit_sha, pr.number),
),
(
"Archiving the repo...",
lambda: archive_repo(repo_path, pr.number, archive_destination, "merged"),
),
]
pbar = tqdm(total=len(setup_steps) + 6, desc="Processing PR", leave=False)