diff --git a/dvc/version.py b/dvc/version.py index 255d3c71b..dbd6705cc 100644 --- a/dvc/version.py +++ b/dvc/version.py @@ -3,6 +3,9 @@ import os import subprocess +BASE_VERSION = "0.35.0" + + def generate_version(base_version): """Generate a version with information about the git repository""" pkg_dir = os.path.abspath(os.path.dirname(os.path.dirname(__file__))) @@ -10,6 +13,9 @@ def generate_version(base_version): if not is_git_repo(pkg_dir) or not have_git(): return base_version + if is_release(pkg_dir, base_version): + return base_version + return "{base_version}+{short_sha}{dirty}".format( base_version=base_version, short_sha=git_revision(pkg_dir).decode("utf-8")[0:6], @@ -33,6 +39,17 @@ def have_git(): return False +def is_release(dir_path, base_version): + try: + output = subprocess.check_output( + ["git", "describe", "--tags", "--exact-match"], cwd=dir_path + ).decode("utf-8") + tag = output.strip() + return tag == base_version + except subprocess.CalledProcessError: + return False + + def git_revision(dir_path): """Get the SHA-1 of the HEAD of a git repository.""" return subprocess.check_output( @@ -49,4 +66,4 @@ def is_dirty(dir_path): return True -__version__ = generate_version(base_version="0.35.0") +__version__ = generate_version(BASE_VERSION) diff --git a/tests/unit/test_version.py b/tests/unit/test_version.py new file mode 100644 index 000000000..249f58e7e --- /dev/null +++ b/tests/unit/test_version.py @@ -0,0 +1,22 @@ +import mock +import subprocess + + +import dvc.version +from dvc.utils.compat import cast_bytes + + +def test_is_release(): + with mock.patch.object(dvc.version.subprocess, "check_output") as m: + m.side_effect = subprocess.CalledProcessError(1, "cmd") + ret = dvc.version.is_release(None, dvc.version.BASE_VERSION) + assert ret is False + + m.side_effect = None + m.return_value = cast_bytes(dvc.version.BASE_VERSION) + ret = dvc.version.is_release(None, dvc.version.BASE_VERSION) + assert ret + + m.return_value = cast_bytes("630d1741c2d5dd89a3176bd15b63121b905d35c9") + ret = dvc.version.is_release(None, dvc.version.BASE_VERSION) + assert ret is False