name : grub-sort-version #!/usr/bin/python3 # """Script to sort kernel versions.""" import apt_pkg import argparse import re import sys import os import typing def order_index(order: list[re.Pattern[str]], abi: str) -> int: """Return the index of abi in order""" for i, entry in zip(range(len(order)), order): if entry.match(abi): return i return len(order) class KernelABI: """Orderable kernel ABI string.""" def __init__(self, abi: str, order: list[re.Pattern[str]]): self.abi = abi self._index = order_index(order, self.abi) def __lt__(self, other: "KernelABI") -> bool: if self._index != other._index: # Ordering is reversed, what should be considered highest comes first. return self._index > other._index return apt_pkg.version_compare(f"{self.abi}-0", f"{other.abi}-0") < 0 def main() -> None: """Entry point.""" apt_pkg.init() parser = argparse.ArgumentParser( prog="grub-sort-version", description="sort kernel ABIs" ) parser.add_argument("-r", "--reverse", action="store_true") args = parser.parse_args() order = [] for flavour in os.environ.get("GRUB_FLAVOUR_ORDER", "").split(): order.append(re.compile(f"[\\s\\S]*-{flavour}(\\s*\\d*)$")) versions = [KernelABI(line.rstrip(), order) for line in sys.stdin] versions.sort(reverse=args.reverse) for v in versions: print(v.abi) if __name__ == "__main__": main()