dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-25 14:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac

Browse source
This commit is contained in:
Paul Gauthier 2023-06-23 14:00:36 -07:00
parent c21ef8dd92
commit 6142488741
1 changed files with 32 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

20
scripts/benchmark.py
View file

@ -2,9 +2,11 @@ import os
import shutil import shutil
import sys import sys
from tempfile import TemporaryDirectory from tempfile import TemporaryDirectory
from git import Repo from git import Repo
def create_temp_repo(dirname):
with TemporaryDirectory() as tempdir:
def create_temp_repo(dirname, tempdir):
# Copy all files from dirname to tempdir # Copy all files from dirname to tempdir
for item in os.listdir(dirname): for item in os.listdir(dirname):
s = os.path.join(dirname, item) s = os.path.join(dirname, item)
@ -28,12 +30,18 @@ def create_temp_repo(dirname):
# Commit with message "initial" # Commit with message "initial"
repo.git.commit(m="initial") repo.git.commit(m="initial")
return tempdir
if __name__ == "__main__": def main(tempdir):
if len(sys.argv) != 2: if len(sys.argv) != 2:
print("Usage: python benchmark.py <dirname>") print("Usage: python benchmark.py <dirname>")
sys.exit(1) sys.exit(1)
dirname = sys.argv[1] dirname = sys.argv[1]
temp_repo_path = create_temp_repo(dirname)
print(f"Temporary repo created at: {temp_repo_path}") create_temp_repo(dirname, tempdir)
if __name__ == "__main__":
# with TemporaryDirectory() as tempdir:
tempdir = "tmp.benchmark"
main(tempdir)