Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import git # pip3 install gitpython
import os
def get_repo(starting_directory: str = os.path.dirname(__file__)) -> git.Repo:
""" Try finding the repository by traversing up the tree.
By default, the repository containing this module is returned.
"""
directory = starting_directory
try:
return git.Repo(directory)
except git.InvalidGitRepositoryError:
pass
# We now have to traverse up the tree
while directory != "/" and os.path.exists(directory):
# Go to parent
directory = os.path.abspath(directory + os.path.sep + "..")
try:
return git.Repo(directory)
except git.InvalidGitRepositoryError:
pass
raise git.InvalidGitRepositoryError("Could not find git repository root in {}".format(starting_directory))
def get_version(repo: git.Repo = None) -> str:
""" Return a version string for the current commit. """
if repo is None:
repo = get_repo()
return "{} [{}{}]".format(repo.active_branch, repo.commit(), " (dirty)" if repo.is_dirty() else "")
if __name__ == "__main__":
print(get_version())