diff options
| -rw-r--r-- | git_deps/detector.py | 13 | ||||
| -rw-r--r-- | git_deps/gitutils.py | 10 |
2 files changed, 15 insertions, 8 deletions
diff --git a/git_deps/detector.py b/git_deps/detector.py index 3f51f0e..3cb8828 100644 --- a/git_deps/detector.py +++ b/git_deps/detector.py @@ -20,20 +20,17 @@ class DependencyDetector(object): tree represented (conceptually) by a list of edges. """ - def __init__(self, options, repo_path=None, logger=None): + def __init__(self, options, repo=None, logger=None): self.options = options if logger is None: self.logger = standard_logger(self.__class__.__name__, options.debug) - if repo_path is None: - try: - repo_path = pygit2.discover_repository('.') - except KeyError: - abort("Couldn't find a repository in the current directory.") - - self.repo = pygit2.Repository(repo_path) + if repo is None: + self.repo = GitUtils.get_repo() + else: + self.repo = repo # Nested dict mapping dependents -> dependencies -> files # causing that dependency -> numbers of lines within that file diff --git a/git_deps/gitutils.py b/git_deps/gitutils.py index f349f44..f7b5470 100644 --- a/git_deps/gitutils.py +++ b/git_deps/gitutils.py @@ -3,6 +3,8 @@ import re import subprocess from git_deps.errors import InvalidCommitish +from git_deps.utils import abort + class GitUtils(object): @classmethod @@ -89,3 +91,11 @@ class GitUtils(object): return commit + @classmethod + def get_repo(cls, path='.'): + try: + repo_path = pygit2.discover_repository(path) + except KeyError: + abort("Couldn't find a repository in the current directory.") + + return pygit2.Repository(repo_path) |