From 561c81929cc4191ebcbcb7baed2f01d9c9ec4619 Mon Sep 17 00:00:00 2001
From: "r.jaepel" <r.jaepel@fz-juelich.de>
Date: Mon, 7 Aug 2023 15:56:15 +0200
Subject: [PATCH] Update class init
---
git_lfs_utils/utils.py | 36 ++++++++++++++++++++----------------
1 file changed, 20 insertions(+), 16 deletions(-)
diff --git a/git_lfs_utils/utils.py b/git_lfs_utils/utils.py
index 188f9e6..14e6163 100644
--- a/git_lfs_utils/utils.py
+++ b/git_lfs_utils/utils.py
@@ -10,6 +10,26 @@ def update_package_list():
class GitRepo(git.Repo):
+ def __init__(self, repository_path=None, *args, **kwargs):
+ if repository_path is None:
+ starting_path = os.getcwd()
+ path = starting_path
+ else:
+ starting_path = repository_path
+ path = repository_path
+ repository_found = False
+ while repository_found is False:
+ # step upwards in path until a git repo is found or root is reached
+ try:
+ super().__init__(path, *args, **kwargs)
+ repository_found = True
+ except git.exc.InvalidGitRepositoryError:
+ # Step upwards in the path once
+ path, directory = os.path.split(path)
+ if len(directory) == 0:
+ # This means that root was reached without finding a git repo.
+ raise ValueError(f"Path {starting_path} does not contain a git repository.")
+
def add_all_files(self):
if len(self.untracked_files) > 0:
untracked_files = "\n".join(["- " + file for file in self.untracked_files])
@@ -43,19 +63,3 @@ class GitRepo(git.Repo):
def print_status(self):
print(self.git.status())
-
-
-def get_git_repo():
- cwd = os.getcwd()
- path = cwd
- repository = None
- while repository is None:
- # step upwards in path until a git repo is found or root is reached
- try:
- repository = GitRepo(path)
- except git.exc.InvalidGitRepositoryError:
- base, directory = os.path.split(path)
- if len(directory) == 0:
- raise ValueError(f"Path {cwd} does not contain a git repository.")
- path = base
- return repository
--
GitLab