Source code for testplan.report.testing.parser

import argparse


class ReportTagsAction(argparse.Action):
    """
    Argparse action for parsing multiple report tag
    arguments, builds up a list of dictionary of sets.

    In:

    .. code-block:: bash

        --report-tags foo bar hello=world --report-tags one two color=red

    Out:

    .. code-block:: python

        [
            {
                'simple': {'foo', 'bar'},
                'hello': {'world'}
            },
            {
                'simple': {'one', 'two'},
                'color': {'red'},
            }
        ]
    """

    def __call__(self, parser, namespace, values, option_string=None):
        from testplan.testing import tagging

        items = getattr(namespace, self.dest) or []

        tag_arg = [tagging.parse_tag_arguments(v) for v in values]
        tag_arg = tagging.merge_tag_dicts(*tag_arg)

        items.append(tag_arg)

        setattr(namespace, self.dest, items)


class ReportFilterAction(argparse.Action):
    """
    Argparse action serving higher-precedence shortcuts for report filters.
    """

    def __init__(self, filter_rep: str, *args, **kwargs):
        self.filter_rep = filter_rep
        super().__init__(*args, **kwargs)

    def __call__(self, parser, namespace, values, option_string=None):
        setattr(namespace, "reporting_filter", self.filter_rep)

    @staticmethod
    def use_filter(filter_rep: str):
        return lambda *args, **kwargs: ReportFilterAction(
            filter_rep, *args, **kwargs
        )