build: update get_version for python 3

This commit is contained in:
Frédéric Péters 2020-01-29 12:10:47 +01:00
parent c0cd299660
commit d853bca95d
1 changed files with 16 additions and 8 deletions

View File

@ -23,18 +23,26 @@ class eo_sdist(sdist):
def get_version(): def get_version():
if os.path.exists('VERSION'): if os.path.exists('VERSION'):
version_file = open('VERSION', 'r') with open('VERSION', 'r') as v:
version = version_file.read() return v.read()
version_file.close()
return version
if os.path.exists('.git'): if os.path.exists('.git'):
p = subprocess.Popen(['git', 'describe', '--dirty', '--match=v*'], stdout=subprocess.PIPE) p = subprocess.Popen(
['git', 'describe', '--dirty=.dirty', '--match=v*'],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
)
result = p.communicate()[0] result = p.communicate()[0]
if p.returncode == 0: if p.returncode == 0:
version = result.split()[0][1:] result = result.decode('ascii').strip()[1:] # strip spaces/newlines and initial v
version = version.replace('-', '.').replace('.g', '+g') if '-' in result: # not a tagged version
real_number, commit_count, commit_hash = result.split('-', 2)
version = '%s.post%s+%s' % (real_number, commit_count, commit_hash)
else:
version = result
return version return version
return '0' else:
return '0.0.post%s' % len(subprocess.check_output(['git', 'rev-list', 'HEAD']).splitlines())
return '0.0'
setup( setup(