"""UniqueLookup class."""
from typing import TYPE_CHECKING
from text_lint.config import LOOKUP_TRANSFORMATION_PREFIX
from text_lint.operations.lookups.bases.lookup_encoder_base import (
LookupEncoderBase,
)
from text_lint.operations.lookups.encoders.unique import UniqueEncoder
from text_lint.utilities.translations import _
if TYPE_CHECKING: # pragma: no cover
from text_lint.linter.states import LookupState
YAML_EXAMPLE_COMPONENTS = (
_("unique save id transformation lookup example"),
)
YAML_EXAMPLE = """
- name: {0}
operation: validate_debug
saved:
- example.capture(1).to_unique()
""".format(*YAML_EXAMPLE_COMPONENTS)
[docs]
class UniqueLookup(LookupEncoderBase):
"""UniqueLookup operation for ResultForest instances."""
encoder_class = UniqueEncoder
hint = _("select only unique values from a save id")
operation = LOOKUP_TRANSFORMATION_PREFIX + "unique"
yaml_example = YAML_EXAMPLE
[docs]
def apply(
self,
state: "LookupState",
) -> None:
"""Select only unique values from a save id."""
state.results = self.encode(state.results)