config/printprereleasesuffix.py

   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 from __future__ import absolute_import
  13 from __future__ import print_function
  14
  15 import sys
  16 import re
  17
  18
  19 def 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
  37 if len(sys.argv) == 2:
  38     print(get_prerelease_suffix(sys.argv[1]))