config/printprereleasesuffix.py
author Sylvestre Ledru <sledru@mozilla.com>
Tue, 22 May 2018 00:01:01 +0200
changeset 419598 fba6f974041a3d3c22ef95c44f6e4262e8e5c52f
parent 123443 8f83edc05fa4d458d82668a95853c907544d0b8d
child 481687 9c5b3c7e1ca2240f765d62466cc917ad778407ef
permissions -rw-r--r--
bug 1463425 - autopep8 on config/ r=gps MozReview-Commit-ID: EaTAhH2CAee

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

# Prints the pre-release version suffix based on the version string
#
# Examples:
# 2.1a3    > " 2.1 Alpha 3"
# 2.1a3pre > ""
# 3.2b4    > " 3.2 Beta 4"
# 3.2b4pre > ""
from __future__ import print_function

import sys
import re


def get_prerelease_suffix(version):
    """ Returns the prerelease suffix from the version string argument """

    def mfunc(m):
        return " {0} {1} {2}".format(m.group('prefix'),
                                     {'a': 'Alpha', 'b': 'Beta'}[m.group('c')],
                                     m.group('suffix'))
    result, c = re.subn(r'^(?P<prefix>(\d+\.)*\d+)(?P<c>[ab])(?P<suffix>\d+)$',
                        mfunc, version)
    if c != 1:
        return ''
    return result


if len(sys.argv) == 2:
    print(get_prerelease_suffix(sys.argv[1]))