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

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

56
scripts/benchmark.py
View file

@ -2,38 +2,46 @@ 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:
# Copy all files from dirname to tempdir
for item in os.listdir(dirname):
s = os.path.join(dirname, item)
d = os.path.join(tempdir, item)
if os.path.isdir(s):
shutil.copytree(s, d, False, None)
else:
shutil.copy2(s, d)
# Copy .docs subdir to tempdir as 'docs'
docs_src = os.path.join(dirname, ".docs")
docs_dst = os.path.join(tempdir, "docs")
shutil.copytree(docs_src, docs_dst, False, None)
# Create a new git repo in tempdir def create_temp_repo(dirname, tempdir):
repo = Repo.init(tempdir) # Copy all files from dirname to tempdir
for item in os.listdir(dirname):
s = os.path.join(dirname, item)
d = os.path.join(tempdir, item)
if os.path.isdir(s):
shutil.copytree(s, d, False, None)
else:
shutil.copy2(s, d)
# Add all copied files to the repo # Copy .docs subdir to tempdir as 'docs'
repo.git.add(A=True) docs_src = os.path.join(dirname, ".docs")
docs_dst = os.path.join(tempdir, "docs")
shutil.copytree(docs_src, docs_dst, False, None)
# Commit with message "initial" # Create a new git repo in tempdir
repo.git.commit(m="initial") repo = Repo.init(tempdir)
return tempdir # Add all copied files to the repo
if __name__ == "__main__": repo.git.add(A=True)
# Commit with message "initial"
repo.git.commit(m="initial")
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)