o i @sdZddlmZddlmZmZddlmZmZm Z m Z m Z m Z m Z ddlmZddlmZmZddlmZddlmZmZd d gZeGd d d ZGd d d eZd S)zEContains the content control class :class:`.InquirerPyUIListControl`.)abstractmethod)asdict dataclass)AnyCallableDictListOptionalTuplecast)FormattedTextControl)InvalidArgumentRequiredKeyNotFound) Separator)InquirerPyListChoicesInquirerPySessionResultChoiceInquirerPyUIListControlc@s>eZdZUdZeed<dZeeed<dZ e ed<ddZ dS) raRClass to create choices for list type prompts. A simple dataclass that can be used as an alternate to using :class:`dict` when working with choices. Args: value: The value of the choice when user selects this choice. name: The value that should be presented to the user prior/after selection of the choice. This value is optional, if not provided, it will fallback to the string representation of `value`. enabled: Indicates if the choice should be pre-selected. This only has effects when the prompt has `multiselect` enabled. valueNnameFenabledcCs|jdur t|j|_dSdS)z-Assign strinify value to name if not present.N)rstrrselfrT/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/InquirerPy/base/control.py __post_init__"s zChoice.__post_init__) __name__ __module__ __qualname____doc__r__annotations__rr rrboolrrrrrrs    c seZdZdZ   d%dedededeeddf fd d Z de edede e e effd d Z edefd dZejdeddfddZede e e effddZejde e e efddfddZd&ddZde ee e ffddZd&ddZede ee e ffddZede ee e ffddZedefddZede e effd d!Zedefd"d#Zejdeddfd$d#ZZS)'raA base class to create :class:`~prompt_toolkit.layout.UIControl` to display list type contents. Args: choices(InquirerPyListChoices): List of choices to display as the content. Can also be a callable or async callable that returns a list of choices. default: Default value, this will affect the cursor position. multiselect: Indicate if the current prompt has `multiselect` enabled. session_result: Current session result. NFchoicesdefault multiselectsession_resultreturncs|pi|_d|_d|_||_t|ts|ntt||j|_t|ts%|ntt||j|_| |j|j|_ | | t |jdS)Nr)_session_result_selected_choice_index _choice_func _multiselect isinstancerr _default _raw_choices _get_choices_choices _safety_check_format_choicessuper__init___get_formatted_choices)rr#r$r%r& __class__rrr43s z InquirerPyUIListControl.__init__c Cs(g}zt|ddD]|\}}t|tr4|d|kr||_|t|d|d|jr.|ddnddq t|trT|j|krH|jdt ||_|t||ddq t|t rst |}|d|krf||_|jsmd|d<||q ||krz||_|t||ddq W|St yt d w) aProcess the raw user input choices and format it into dictionary. Args: choices: List of chices to display. default: Default value, this will affect the :attr:`.InquirerPyUIListControl.selected_choice_index` Returns: List of choices. Raises: RequiredKeyNotFound: When the provided choice is missing the `name` or `value` key. r)startrrrF)rrrz@dictionary type of choice require a 'name' key and a 'value' key) enumerater,dictselected_choice_indexappendrr+getrlenrrKeyErrorr)rr#r$processed_choicesindexchoice dict_choicerrrr/MsR        & z$InquirerPyUIListControl._get_choicescC|jS)zint: Current highlighted index.r)rrrrr<z-InquirerPyUIListControl.selected_choice_indexrcC ||_dSNrFrrrrrr< cCrE)z0List[Dict[str, Any]]: Get all processed choices.r0rrrrr#rGzInquirerPyUIListControl.choicescCrHrIrLrJrrrr#rKcCsD|jstdd}|jD] }t|dtsd}nq |s tddS)zValidate processed choices. Check if the choices are empty or if it only contains :class:`~InquirerPy.separator.Separator`. z argument choices cannot be emptyFrTzs $