# 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]))