This commit is contained in:
Paul Gauthier 2023-06-23 14:00:36 -07:00
parent c21ef8dd92
commit 6142488741

View file

@ -2,9 +2,11 @@ import os
import shutil
import sys
from tempfile import TemporaryDirectory
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
for item in os.listdir(dirname):
s = os.path.join(dirname, item)
@ -28,12 +30,18 @@ def create_temp_repo(dirname):
# Commit with message "initial"
repo.git.commit(m="initial")
return tempdir
if __name__ == "__main__":
def main(tempdir):
if len(sys.argv) != 2:
print("Usage: python benchmark.py <dirname>")
sys.exit(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)