pmb: Migrate aportgen to Command (MR 2428)

pmb/commands/__init__.py

@@ -10,6 +10,7 @@ from pmb.types import PmbArgs
 from pmb.helpers import frontend
 
 from .base import Command
+from .aportgen import Aportgen
 from .log import Log
 from .index import Index
 from .repo_bootstrap import RepoBootstrap
@@ -46,7 +47,6 @@ unmigrated_commands = [
     "chroot",
     "install",
     "checksum",
-    "aportgen",
     "build",
     "deviceinfo_parse",
     "apkbuild_parse",
@@ -64,6 +64,8 @@ def run_command(args: PmbArgs):
 
     command: Command
     match args.action:
+        case "aportgen":
+            command = Aportgen(args.packages, args.fork_alpine)
         case "log":
             command = Log(args.clear_log, args.lines)
         case "index":

pmb/commands/aportgen.py

+# Copyright 2024 Stefan Hansson
+# SPDX-License-Identifier: GPL-3.0-or-later
+from __future__ import annotations
+
+import pmb.aportgen
+from pmb import commands
+from pmb.helpers import logging
+
+
+class Aportgen(commands.Command):
+    def __init__(self, package_list: list[str], fork_alpine: bool) -> None:
+        self.package_list = package_list
+        self.fork_alpine = fork_alpine
+
+    def run(self) -> None:
+        for package in self.package_list:
+            logging.info(f"Generate aport: {package}")
+            pmb.aportgen.generate(package, self.fork_alpine)

pmb/helpers/frontend.py

@@ -102,12 +102,6 @@ def _install_ondev_verify_no_rootfs(device: str, ondev_cp: list[tuple[str, str]]
     )
 
 
-def aportgen(args: PmbArgs) -> None:
-    for package in args.packages:
-        logging.info("Generate aport: " + package)
-        pmb.aportgen.generate(package, args.fork_alpine)
-
-
 def build(args: PmbArgs) -> None:
     # Strict mode: zap everything
     if args.strict: