Source code for xcookie.rich_ext

import ubelt as ub
from rich import prompt




class FuzzyPrompt(prompt.Prompt):
    """
    The user just needs to enter a non-ambiguous prefix
    """



    def process_response(self, value: str) -> str:
        """Normalize the response"""
        assert self.choices is not None
        got = value.strip().lower()
        norm_choices = [c.lower() for c in self.choices]
        flags = [c.startswith(got) for c in norm_choices]
        if sum(flags) != 1:
            raise prompt.InvalidResponse(self.validate_error_message)
        norm: str = ub.peek(ub.compress(self.choices, flags))  # type: ignore
        return norm