1# This Source Code Form is subject to the terms of the Mozilla Public
2# License, v. 2.0. If a copy of the MPL was not distributed with this
3# file, You can obtain one at http://mozilla.org/MPL/2.0/.
4
5# Prints the pre-release version suffix based on the version string
6#
7# Examples:
8# 2.1a3    > " 2.1 Alpha 3"
9# 2.1a3pre > ""
10# 3.2b4    > " 3.2 Beta 4"
11# 3.2b4pre > ""
12from __future__ import absolute_import
13from __future__ import print_function
14
15import sys
16import re
17
18
19def get_prerelease_suffix(version):
20    """ Returns the prerelease suffix from the version string argument """
21
22    def mfunc(m):
23        return " {0} {1} {2}".format(
24            m.group("prefix"),
25            {"a": "Alpha", "b": "Beta"}[m.group("c")],
26            m.group("suffix"),
27        )
28
29    result, c = re.subn(
30        r"^(?P<prefix>(\d+\.)*\d+)(?P<c>[ab])(?P<suffix>\d+)$", mfunc, version
31    )
32    if c != 1:
33        return ""
34    return result
35
36
37if len(sys.argv) == 2:
38    print(get_prerelease_suffix(sys.argv[1]))
39