[gentoo-commits] proj/pkgcore/pkgcore:master commit in: src/pkgcore/scripts/

["Arthur Zamarin" <arthurzam@gentoo.org>]

commit:     590658425abffe0253602adc30f7c22172f92b13
Author:     Arthur Zamarin <arthurzam <AT> gentoo <DOT> org>
AuthorDate: Mon Oct 23 17:03:30 2023 +0000
Commit:     Arthur Zamarin <arthurzam <AT> gentoo <DOT> org>
CommitDate: Mon Oct 23 17:03:30 2023 +0000
URL:        https://gitweb.gentoo.org/proj/pkgcore/pkgcore.git/commit/?id=59065842

pmaint eclass: add better invocation with output filename format

Signed-off-by: Arthur Zamarin <arthurzam <AT> gentoo.org>

 src/pkgcore/scripts/pmaint.py | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/src/pkgcore/scripts/pmaint.py b/src/pkgcore/scripts/pmaint.py
index 008de533f..c148454a7 100644
--- a/src/pkgcore/scripts/pmaint.py
+++ b/src/pkgcore/scripts/pmaint.py
@@ -496,6 +496,19 @@ eclass_opts = eclass.add_argument_group("subcommand options")
 eclass_opts.add_argument(
     "--dir", dest="output_dir", type=arghparse.create_dir, help="output directory"
 )
+eclass_opts.add_argument(
+    "-o",
+    "--output",
+    dest="output_format",
+    default="{eclass}.eclass.{format}",
+    help="output file name format",
+    docs="""
+        Output file name format. Defaults to ``{eclass}.eclass.{format}``. You
+        can use ``{eclass}`` and ``{format}`` placeholders to customize the
+        output file name. The filename can have path separator, for example:
+        ``{eclass}/{eclass}.eclass.{format}``.
+    """,
+)
 eclass_opts.add_argument(
     "-f",
     "--format",
@@ -541,9 +554,16 @@ def _eclass_main(options, out, err):
 
     for path in options.eclasses:
         try:
-            with open(
-                pjoin(options.output_dir, f"{os.path.basename(path)}.{ext}"), "wt"
-            ) as f:
+            filename = pjoin(
+                options.output_dir,
+                options.output_format.format(
+                    eclass=os.path.basename(path).removesuffix(".eclass"),
+                    format=ext,
+                ),
+            )
+            out.write("Compiling: ", path)
+            os.makedirs(os.path.dirname(filename), exist_ok=True)
+            with open(filename, "wt") as f:
                 obj = EclassDoc(path, sourced=True)
                 convert_func = getattr(obj, f"to_{options.format}")
                 f.write(convert_func())