source: this_version.py @ 8774047

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5
Last change on this file since 8774047 was ddc9733, checked in by Paul Brossier <piem@piem.org>, 7 years ago

this_version.py: add options to get only libaubio or python-aubio version

  • Property mode set to 100644
File size: 4.2 KB
Line 
1#! python
2import os
3import sys
4
5__version_info = {} # keep a reference to parse VERSION once
6
7def get_version_info():
8    # read from VERSION
9    # return dictionary filled with content of version
10    if not __version_info:
11        this_file_dir = os.path.dirname(os.path.abspath(__file__))
12        version_file = os.path.join(this_file_dir, 'VERSION')
13
14        if not os.path.isfile(version_file):
15            raise SystemError("VERSION file not found.")
16
17        for l in open(version_file).readlines():
18            if l.startswith('AUBIO_MAJOR_VERSION'):
19                __version_info['AUBIO_MAJOR_VERSION'] = int(l.split('=')[1])
20            if l.startswith('AUBIO_MINOR_VERSION'):
21                __version_info['AUBIO_MINOR_VERSION'] = int(l.split('=')[1])
22            if l.startswith('AUBIO_PATCH_VERSION'):
23                __version_info['AUBIO_PATCH_VERSION'] = int(l.split('=')[1])
24            if l.startswith('AUBIO_VERSION_STATUS'):
25                __version_info['AUBIO_VERSION_STATUS'] = \
26                    l.split('=')[1].strip()[1:-1]
27
28            if l.startswith('LIBAUBIO_LT_CUR'):
29                __version_info['LIBAUBIO_LT_CUR'] = int(l.split('=')[1])
30            if l.startswith('LIBAUBIO_LT_REV'):
31                __version_info['LIBAUBIO_LT_REV'] = int(l.split('=')[1])
32            if l.startswith('LIBAUBIO_LT_AGE'):
33                __version_info['LIBAUBIO_LT_AGE'] = int(l.split('=')[1])
34
35        if len(__version_info) < 6:
36            raise SystemError("Failed parsing VERSION file.")
37
38        # switch version status with commit sha in alpha releases
39        if __version_info['AUBIO_VERSION_STATUS'] and \
40                '~alpha' in __version_info['AUBIO_VERSION_STATUS']:
41            AUBIO_GIT_SHA = get_git_revision_hash()
42            if AUBIO_GIT_SHA:
43                __version_info['AUBIO_VERSION_STATUS'] = '~git+' + AUBIO_GIT_SHA
44
45    return __version_info
46
47def get_libaubio_version():
48    verfmt = '%(LIBAUBIO_LT_CUR)s.%(LIBAUBIO_LT_REV)s.%(LIBAUBIO_LT_AGE)s'
49    return str(verfmt % get_version_info())
50
51def get_aubio_version():
52    verfmt = '%(AUBIO_MAJOR_VERSION)s.%(AUBIO_MINOR_VERSION)s.%(AUBIO_PATCH_VERSION)s%(AUBIO_VERSION_STATUS)s'
53    return str(verfmt % get_version_info())
54
55def get_aubio_pyversion():
56    # convert to version for python according to pep 440
57    # see https://www.python.org/dev/peps/pep-0440/
58    # outputs MAJ.MIN.PATCH[a0[+git.<sha>[.mods]]]
59    aubio_version = get_aubio_version()
60    if '~git+' in aubio_version:
61        pep440str = aubio_version.replace('+', '.')
62        verstr = pep440str.replace('~git.', 'a0+')
63    elif '~alpha' in aubio_version:
64        verstr = aubio_version.replace('~alpha', 'a0')
65    else:
66        verstr = aubio_version
67    return verstr
68
69def get_git_revision_hash(short=True):
70    # get commit id, with +mods if local tree is not clean
71    if not os.path.isdir('.git'):
72        # print('Version : not in git repository : can\'t get sha')
73        return None
74    import subprocess
75    aubio_dir = os.path.dirname(os.path.abspath(__file__))
76    if not os.path.exists(aubio_dir):
77        raise SystemError("git / root folder not found")
78    gitcmd = ['git', '-C', aubio_dir, 'rev-parse']
79    if short:
80        gitcmd.append('--short')
81    gitcmd.append('HEAD')
82    try:
83        gitsha = subprocess.check_output(gitcmd).strip().decode('utf8')
84    except Exception as e:
85        sys.stderr.write('git command error :%s\n' % e)
86        return None
87    # check if we have a clean tree
88    gitcmd = ['git', '-C', aubio_dir, 'status', '--porcelain']
89    try:
90        output = subprocess.check_output(gitcmd).decode('utf8')
91        if len(output):
92            sys.stderr.write('Info: current tree is not clean\n\n')
93            sys.stderr.write(output + '\n')
94            gitsha += '+mods'
95    except subprocess.CalledProcessError as e:
96        sys.stderr.write('git command error :%s\n' % e)
97        pass
98    return gitsha
99
100if __name__ == '__main__':
101    if len(sys.argv) > 1 and sys.argv[1] == '-v':
102        print (get_aubio_version())
103    elif len(sys.argv) > 1 and sys.argv[1] == '-p':
104        print (get_aubio_version())
105    else:
106        print ('%30s'% 'aubio version:', get_aubio_version())
107        print ('%30s'% 'python-aubio version:', get_aubio_pyversion())
Note: See TracBrowser for help on using the repository browser.