printprereleasesuffix.py (915B)
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 > "" 12 import re 13 import sys 14 15 16 def get_prerelease_suffix(version): 17 """Returns the prerelease suffix from the version string argument""" 18 19 def mfunc(m): 20 return " {} {} {}".format( 21 m.group("prefix"), 22 {"a": "Alpha", "b": "Beta"}[m.group("c")], 23 m.group("suffix"), 24 ) 25 26 result, c = re.subn( 27 r"^(?P<prefix>(\d+\.)*\d+)(?P<c>[ab])(?P<suffix>\d+)$", mfunc, version 28 ) 29 if c != 1: 30 return "" 31 return result 32 33 34 if len(sys.argv) == 2: 35 print(get_prerelease_suffix(sys.argv[1]))