Convert input version<->hex and print with --hex

Author: hugovk

Doc/tools/extensions/patchlevel.py

@@ -73,9 +73,89 @@ def get_version_info():
         return version, release
 
 
-if __name__ == "__main__":
-    short_ver, full_ver = format_version_info(get_header_version_info())
-    if sys.argv[1:2] == ["--short"]:
+def get_version_hex(info: version_info) -> int:
+    """Convert a version_info object to a hex version number."""
+    levels = {"alpha": 0xA, "beta": 0xB, "candidate": 0xC, "final": 0xF}
+    return (
+        (info.major << 24)
+        | (info.minor << 16)
+        | (info.micro << 8)
+        | (levels[info.releaselevel] << 4)
+        | (info.serial << 0)
+    )
+
+
+def parse_str_version(version: str) -> version_info:
+    """Convert a version string to a version_info object."""
+    tag_cre = re.compile(r"(\d+)(?:\.(\d+)(?:\.(\d+))?)?(?:([ab]|rc)(\d+))?$")
+    result = tag_cre.match(version)
+    if not result:
+        raise ValueError(f"Invalid version string: {version}")
+
+    parts = list(result.groups())
+    levels = {"a": "alpha", "b": "beta", "rc": "candidate"}
+    return version_info(
+        major=int(parts[0]),
+        minor=int(parts[1]),
+        micro=int(parts[2] or 0),
+        releaselevel=levels.get(parts[3], "final"),
+        serial=int(parts[4]) if parts[4] else 0,
+    )
+
+
+def parse_hex_version(version: int) -> version_info:
+    """Convert a hex version number to a version_info object."""
+    if not isinstance(version, int):
+        raise ValueError(f"Invalid hex version: {version}")
+
+    levels = {0xA: "alpha", 0xB: "beta", 0xC: "candidate", 0xF: "final"}
+    return version_info(
+        major=(version >> 24) & 0xFF,
+        minor=(version >> 16) & 0xFF,
+        micro=(version >> 8) & 0xFF,
+        releaselevel=levels.get((version >> 4) & 0xF, "final"),
+        serial=version & 0xF,
+    )
+
+
+def main() -> None:
+    import argparse
+
+    parser = argparse.ArgumentParser(color=True)
+    parser.add_argument(
+        "version",
+        nargs="?",
+        help="version to convert (default: repo version)",
+    )
+    group = parser.add_mutually_exclusive_group()
+    group.add_argument(
+        "--short",
+        action="store_true",
+        help="print version as x.y",
+    )
+    group.add_argument(
+        "--hex",
+        action="store_true",
+        help="print version as a 4-byte hex number",
+    )
+    args = parser.parse_args()
+
+    if args.version:
+        try:
+            info = parse_str_version(args.version)
+        except ValueError:
+            info = parse_hex_version(int(args.version, 16))
+    else:
+        info = get_header_version_info()
+
+    short_ver, full_ver = format_version_info(info)
+    if args.short:
         print(short_ver)
+    elif args.hex:
+        print(hex(get_version_hex(info)))
     else:
         print(full_ver)
+
+
+if __name__ == "__main__":
+    main()