From 9c8354041536e7261868dba77133584c601430b3 Mon Sep 17 00:00:00 2001 From: Benjamin Dauvergne Date: Mon, 9 Mar 2015 13:01:43 +0100 Subject: [PATCH] Set version only from git tags --- mellon/__init__.py | 1 - setup.py | 42 ++++++++++++++++-------------------------- 2 files changed, 16 insertions(+), 27 deletions(-) diff --git a/mellon/__init__.py b/mellon/__init__.py index 03717d2..668d67a 100644 --- a/mellon/__init__.py +++ b/mellon/__init__.py @@ -1,4 +1,3 @@ -__version__ = '1.2.11' try: import lasso except ImportError: diff --git a/setup.py b/setup.py index 9d83d16..22a75f6 100755 --- a/setup.py +++ b/setup.py @@ -3,6 +3,8 @@ ''' Setup script for mellon ''' +import os +import subprocess from setuptools import setup, find_packages from setuptools.command.install_lib import install_lib as _install_lib from distutils.command.build import build as _build @@ -42,35 +44,23 @@ class install_lib(_install_lib): _install_lib.run(self) def get_version(): - import glob - import re - import os - - version = None - for d in glob.glob('*'): - if not os.path.isdir(d): - continue - module_file = os.path.join(d, '__init__.py') - if not os.path.exists(module_file): - continue - for v in re.findall("""__version__ *= *['"](.*)['"]""", - open(module_file).read()): - assert version is None - version = v - if version: - break - assert version is not None + '''Use the VERSION, if absent generates a version with git describe, if not + tag exists, take 0.0.0- and add the length of the commit log. + ''' + if os.path.exists('VERSION'): + with open('VERSION', 'r') as v: + return v.read() if os.path.exists('.git'): - import subprocess p = subprocess.Popen(['git','describe','--dirty','--match=v*'], - stdout=subprocess.PIPE) + stdout=subprocess.PIPE, stderr=subprocess.PIPE) result = p.communicate()[0] - assert p.returncode == 0, 'git returned non-zero' - new_version = result.split()[0][1:] - assert new_version.split('-')[0] == version, '__version__ must match the last git annotated tag' - version = new_version.replace('-', '.') - return version - + if p.returncode == 0: + return result.split()[0][1:].replace('-', '.') + else: + return '0.0.0-%s' % len( + subprocess.check_output( + ['git', 'rev-list', 'HEAD']).splitlines()) + return '0.0.0' setup(name="django-mellon", version=get_version(),