Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions roots/test-prog-subcommands/conf.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
from __future__ import annotations

import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent))
extensions = ["sphinx_argparse_cli"]
nitpicky = True
4 changes: 4 additions & 0 deletions roots/test-prog-subcommands/index.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
.. sphinx_argparse_cli::
:module: parser
:func: make
:prog: my-tool
10 changes: 10 additions & 0 deletions roots/test-prog-subcommands/parser.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
from __future__ import annotations

from argparse import ArgumentParser


def make() -> ArgumentParser:
parser = ArgumentParser(prog="original-name")
sub = parser.add_subparsers()
sub.add_parser("foo", help="foo help")
return parser
13 changes: 12 additions & 1 deletion src/sphinx_argparse_cli/_logic.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,9 @@ def parser(self) -> ArgumentParser:
raise self.error(msg)

if "prog" in self.options:
self._parser.prog = self.options["prog"]
old_prog, new_prog = self._parser.prog, self.options["prog"]
self._parser.prog = new_prog
_update_sub_parser_prog(self._parser, old_prog, new_prog)

self._raw_format = self._parser.formatter_class == RawDescriptionHelpFormatter
return self._parser
Expand Down Expand Up @@ -453,6 +455,15 @@ def _strip_ansi_colors(text: str) -> str: # pragma: >=3.14 cover
return _ANSI_COLOR_RE.sub("", text)


def _update_sub_parser_prog(parser: ArgumentParser, old_prog: str, new_prog: str) -> None:
if not (sub_parsers := parser._subparsers): # noqa: SLF001
return
sub_action: _SubParsersAction[ArgumentParser] = sub_parsers._group_actions[0] # type: ignore[assignment] # noqa: SLF001
for sub_parser in sub_action.choices.values():
sub_parser.prog = sub_parser.prog.replace(old_prog, new_prog, 1)
_update_sub_parser_prog(sub_parser, old_prog, new_prog)


__all__ = [
"SphinxArgparseCli",
]
7 changes: 7 additions & 0 deletions tests/test_logic.py
Original file line number Diff line number Diff line change
Expand Up @@ -390,3 +390,10 @@ def test_tuple_metavar(build_outcome: str) -> None:
assert "select a pair" in build_outcome
assert "default: None" not in build_outcome
assert '"VAL"' in build_outcome


@pytest.mark.sphinx(buildername="text", testroot="prog-subcommands")
def test_prog_subcommands(build_outcome: str) -> None:
assert "my-tool" in build_outcome
assert "original-name" not in build_outcome
assert "my-tool foo" in build_outcome