1# -*- coding: utf-8 -*-
2"""
3    patchlevel.py
4    ~~~~~~~~~~~~~
5
6    Extract version info from Include/patchlevel.h.
7    Adapted from Doc/tools/getversioninfo.
8
9    :copyright: 2007-2008 by Georg Brandl.
10    :license: Python license.
11"""
12
13from __future__ import print_function
14
15import os
16import re
17import sys
18
19def get_header_version_info(srcdir):
20    patchlevel_h = os.path.join(srcdir, '..', 'Include', 'patchlevel.h')
21
22    # This won't pick out all #defines, but it will pick up the ones we
23    # care about.
24    rx = re.compile(r'\s*#define\s+([a-zA-Z][a-zA-Z_0-9]*)\s+([a-zA-Z_0-9]+)')
25
26    d = {}
27    with open(patchlevel_h) as f:
28        for line in f:
29            m = rx.match(line)
30            if m is not None:
31                name, value = m.group(1, 2)
32                d[name] = value
33
34    release = version = '%s.%s' % (d['PY_MAJOR_VERSION'], d['PY_MINOR_VERSION'])
35    micro = int(d['PY_MICRO_VERSION'])
36    release += '.' + str(micro)
37
38    level = d['PY_RELEASE_LEVEL']
39    suffixes = {
40        'PY_RELEASE_LEVEL_ALPHA': 'a',
41        'PY_RELEASE_LEVEL_BETA':  'b',
42        'PY_RELEASE_LEVEL_GAMMA': 'rc',
43        }
44    if level != 'PY_RELEASE_LEVEL_FINAL':
45        release += suffixes[level] + str(int(d['PY_RELEASE_SERIAL']))
46    return version, release
47
48
49def get_sys_version_info():
50    major, minor, micro, level, serial = sys.version_info
51    release = version = '%s.%s' % (major, minor)
52    release += '.%s' % micro
53    if level != 'final':
54        release += '%s%s' % (level[0], serial)
55    return version, release
56
57
58def get_version_info():
59    try:
60        return get_header_version_info('.')
61    except (IOError, OSError):
62        version, release = get_sys_version_info()
63        print('Can\'t get version info from Include/patchlevel.h, ' \
64              'using version of this interpreter (%s).' % release, file=sys.stderr)
65        return version, release
66
67if __name__ == '__main__':
68    print(get_header_version_info('.')[1])
69