Project: rich-argparse

Rich help formatters for argparse and optparse

Project Details

Latest version
1.4.0
Home Page
PyPI Page
https://pypi.org/project/rich-argparse/

Project Popularity

PageRank
0.0022081950779642224
Number of downloads
6507276

rich-argparse

python -m rich_argparse

tests pre-commit.ci status Downloads Python Version Release

Format argparse and optparse help using rich.

rich-argparse improves the look and readability of argparse's help while requiring minimal changes to the code.

Table of contents

Installation

Install from PyPI with pip or your favorite tool.

pip install rich-argparse

Usage

Simply pass formatter_class to the argument parser

import argparse
from rich_argparse import RichHelpFormatter

parser = argparse.ArgumentParser(..., formatter_class=RichHelpFormatter)
...

rich-argparse defines equivalents to argparse's built-in formatters:

rich_argparse formatter equivalent in argparse
RichHelpFormatter HelpFormatter
RawDescriptionRichHelpFormatter RawDescriptionHelpFormatter
RawTextRichHelpFormatter RawTextHelpFormatter
ArgumentDefaultsRichHelpFormatter ArgumentDefaultsHelpFormatter
MetavarTypeRichHelpFormatter MetavarTypeHelpFormatter

Output styles

The default styles used by rich-argparse are carefully chosen to work in different light and dark themes.

Customize the colors

You can customize the colors in the output by modifying the styles dictionary on the formatter class. rich-argparse defines the following styles:

{
    'argparse.args': 'cyan',  # for positional-arguments and --options (e.g "--help")
    'argparse.groups': 'dark_orange',  # for group names (e.g. "positional arguments")
    'argparse.help': 'default',  # for argument's help text (e.g. "show this help message and exit")
    'argparse.metavar': 'dark_cyan',  # for metavariables (e.g. "FILE" in "--file FILE")
    'argparse.prog': 'grey50',  # for %(prog)s in the usage (e.g. "foo" in "Usage: foo [options]")
    'argparse.syntax': 'bold',  # for highlights of back-tick quoted text (e.g. "`some text`")
    'argparse.text': 'default',  # for descriptions, epilog, and --version (e.g. "A program to foo")
    'argparse.default': 'italic',  # for %(default)s in the help (e.g. "Value" in "(default: Value)")
}

For example, to make the description and epilog italic, change the argparse.text style:

RichHelpFormatter.styles["argparse.text"] = "italic"

Customize the group name format

You can change how the names of the groups (like 'positional arguments' and 'options') are formatted by setting the RichHelpFormatter.group_name_formatter which is set to str.title by default. Any callable that takes the group name as an input and returns a str works:

RichHelpFormatter.group_name_formatter = str.upper  # Make group names UPPERCASE

Special text highlighting

You can highlight patterns in the arguments help and the description and epilog using regular expressions. By default, rich-argparse highlights patterns of --options-with-hyphens using the argparse.args style and patterns of `back tick quoted text` using the argparse.syntax style. You can control what patterns are highlighted by modifying the RichHelpFormatter.highlights list. To disable all highlights, you can clear this list using RichHelpFormatter.highlights.clear().

You can also add custom highlight patterns and styles. The following example highlights all occurrences of pyproject.toml in green:

# Add a style called `pyproject` which applies a green style (any rich style works)
RichHelpFormatter.styles["argparse.pyproject"] = "green"
# Add the highlight regex (the regex group name must match an existing style name)
RichHelpFormatter.highlights.append(r"\b(?P<pyproject>pyproject\.toml)\b")
# Pass the formatter class to argparse
parser = argparse.ArgumentParser(..., formatter_class=RichHelpFormatter)
...

Colors in the usage

The usage generated by the formatter is colored using the argparse.args and argparse.metavar styles. If you use a custom usage message in the parser, it will treated as "plain text" and will not be colored by default. You can enable colors in user defined usage message through console markup by setting RichHelpFormatter.usage_markup = True. If you enable this option, make sure to escape any square brackets in the usage text.

Colors in --version

If you use the "version" action from argparse, you can use console markup in the version string:

parser.add_argument(
    "--version", action="version", version="[argparse.prog]%(prog)s[/] version [i]1.0.0[/]"
)

Note that the argparse.text style is applied to the version string similar to the description and epilog.

Rich descriptions and epilog

You can use any rich renderable in the descriptions and epilog. This includes all built-in rich renderables like Table and Markdown and any custom renderables defined using the Console Protcol.

import argparse
from rich.markdown import Markdown
from rich_argparse import RichHelpFormatter

description = """
# My program

This is a markdown description of my program.

* It has a list
* And a table

| Column 1 | Column 2 |
| -------- | -------- |
| Value 1  | Value 2  |
"""
parser = argparse.ArgumentParser(
    description=Markdown(description, style="argparse.text"),
    formatter_class=RichHelpFormatter,
)
...

Certain features are disabled for arbitrary renderables other than strings, including:

  • Syntax highlighting with RichHelpFormatter.highlights
  • Styling with the "argparse.text" style defined in RichHelpFormatter.styles
  • Replacement of %(prog)s with the program name

Arbitrary renderables are displayed "as is" except for long runs of empty lines that get truncated to two empty lines following the behavior of argparse.

Working with subparsers

Subparsers do not inherit the formatter class from the parent parser by default. You have to pass the formatter class explicitly:

subparsers = parser.add_subparsers(...)
p1 = subparsers.add_parser(..., formatter_class=parser.formatter_class)
p2 = subparsers.add_parser(..., formatter_class=parser.formatter_class)

Generate help preview

You can generate a preview of the help message for your CLI in SVG, HTML, or TXT formats using the HelpPreviewAction action. This is useful for including the help message in the documentation of your app. The action uses the rich exporting API internally.

import argparse
from rich.terminal_theme import DIMMED_MONOKAI
from rich_argparse import HelpPreviewAction, RichHelpFormatter

parser = argparse.ArgumentParser(..., formatter_class=RichHelpFormatter)
...
parser.add_argument(
    "--generate-help-preview",
    action=HelpPreviewAction,
    path="help-preview.svg",  # (optional) or "help-preview.html" or "help-preview.txt"
    export_kwds={"theme": DIMMED_MONOKAI},  # (optional) keywords passed to console.save_... methods
)

This action is hidden, it won't show up in the help message or in the parsed arguments namespace.

Use it like this:

python my_cli.py --generate-help-preview  # generates help-preview.svg (default path specified above)
# or
python my_cli.py --generate-help-preview my-help.svg  # generates my-help.svg
# or
COLUMNS=120 python my_cli.py --generate-help-preview  # force the width of the output to 120 columns

Working with third party formatters

rich-argparse can be used with other formatters that do not rely on the private internals of argparse.HelpFormatter. A popular example is django that defines a custom help formatter that is used with its built in commands as well as with extension libraries and user defined commands. To use rich-argparse in your django project, change your manage.py file as follows:

diff --git a/my_project/manage.py b/my_project/manage.py
index 7fb6855..5e5d48a 100755
--- a/my_project/manage.py
+++ b/my_project/manage.py
@@ -1,22 +1,38 @@
 #!/usr/bin/env python
 """Django's command-line utility for administrative tasks."""
 import os
 import sys


 def main():
     """Run administrative tasks."""
     os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'my_project.settings')
     try:
         from django.core.management import execute_from_command_line
     except ImportError as exc:
         raise ImportError(
             "Couldn't import Django. Are you sure it's installed and "
             "available on your PYTHONPATH environment variable? Did you "
             "forget to activate a virtual environment?"
         ) from exc
+
+    from django.core.management.base import BaseCommand, DjangoHelpFormatter
+    from rich_argparse import RichHelpFormatter
+
+    class DjangoRichHelpFormatter(DjangoHelpFormatter, RichHelpFormatter):  # django first
+        """A rich-based help formatter for django commands."""
+
+    original_create_parser = BaseCommand.create_parser
+
+    def create_parser(*args, **kwargs):
+        parser = original_create_parser(*args, **kwargs)
+        parser.formatter_class = DjangoRichHelpFormatter  # set the formatter_class
+        return parser
+
+    BaseCommand.create_parser = create_parser
+
     execute_from_command_line(sys.argv)


 if __name__ == '__main__':
     main()

Now try out some command like python manage.py runserver --help and notice how the special ordering of the arguments applied by django is respected by the new help formatter.

Optparse support

rich-argparse now ships with experimental support for optparse.

Import optparse help formatters from rich_argparse.optparse:

import optparse
from rich_argparse.optparse import IndentedRichHelpFormatter  # or TitledRichHelpFormatter

parser = optparse.OptionParser(formatter=IndentedRichHelpFormatter())
...

You can also generated a more helpful usage message by passing usage=GENERATE_USAGE to the parser. This is similar to the default behavior of argparse.

from rich_argparse.optparse import GENERATE_USAGE, IndentedRichHelpFormatter

parser = optparse.OptionParser(usage=GENERATE_USAGE, formatter=IndentedRichHelpFormatter())

Similar to argparse, you can customize the styles used by the formatter by modifying the RichHelpFormatter.styles dictionary. These are the same styles used by argparse but with the optparse. prefix instead:

RichHelpFormatter.styles["optparse.metavar"] = "bold magenta"

Syntax highlighting works the same as with argparse.

Colors in the usage are only supported when using GENERATE_USAGE.

Customizing the group name format is not supported. optparse uses Title Case format by default.

Legacy Windows support

If your application still runs on legacy Windows versions (older than Windows 10), you'll need to enable ANSI escape sequences by calling colorama.init() otherwise colors will be disabled:

import argparse
import colorama
from rich_argparse import RichHelpFormatter

colorama.init()
parser = argparse.ArgumentParser(..., formatter_class=RichHelpFormatter)
...

This is not required on Windows 10 and newer or on other operating systems.