o j@sdZddlmZddlmZmZddlmZddlm Z ddl m Z ddl m Z mZgd ZGd d d eZGd d d edZGdddeZGdddeZGdddeZGdddeZGdddeZdS)zV Input validation for a `Buffer`. (Validators will be called before accepting input.) ) annotations)ABCMetaabstractmethod)Callablerun_in_executor_with_context)Document) FilterOrBool to_filter)ConditionalValidatorValidationError ValidatorThreadedValidatorDummyValidatorDynamicValidatorcs.eZdZdZddfd d Zdd d ZZS)r z Error raised by :meth:`.Validator.validate`. :param cursor_position: The cursor position where the error occurred. :param message: Text. rcursor_positionintmessagestrreturnNonecst|||_||_dSN)super__init__rr)selfrr __class__/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/prompt_toolkit/validation.pyr"s  zValidationError.__init__cCs|jjd|jd|jdS)Nz(cursor_position=z , message=))r__name__rrrrrr __repr__'szValidationError.__repr__)rr)rrrrrrrr)r" __module__ __qualname____doc__rr$ __classcell__rrrr r sr c@s<eZdZdZedddZddd Ze ddddZdS)ra Abstract base class for an input validator. A validator is typically created in one of the following two ways: - Either by overriding this class and implementing the `validate` method. - Or by passing a callable to `Validator.from_callable`. If the validation takes some time and needs to happen in a background thread, this can be wrapped in a :class:`.ThreadedValidator`. documentr rrcCsdS)z Validate the input. If invalid, this should raise a :class:`.ValidationError`. :param document: :class:`~prompt_toolkit.document.Document` instance. Nrrr*rrr validate8szValidator.validatecs$z||WdStyw)z Return a `Future` which is set when the validation is ready. This function can be overloaded in order to provide an asynchronous implementation. N)r,r r+rrr validate_asyncBs  zValidator.validate_async Invalid inputF validate_funcCallable[[str], bool] error_messagermove_cursor_to_endboolcCs t|||S)aF Create a validator from a simple validate callable. E.g.: .. code:: python def is_valid(text): return text in ['hello', 'world'] Validator.from_callable(is_valid, error_message='Invalid input') :param validate_func: Callable that takes the input string, and returns `True` if the input is valid input. :param error_message: Message to be displayed if the input is invalid. :param move_cursor_to_end: Move the cursor to the end of the input, if the input is invalid. )_ValidatorFromCallable)clsr/r1r2rrr from_callableMs zValidator.from_callableNr*r rr)r.F)r/r0r1rr2r3rr) r"r&r'r(rr,r- classmethodr6rrrr r+s   r) metaclassc@s.eZdZdZdd d Zdd d ZdddZdS)r4z0 Validate input from a simple callable. funcr0r1rr2r3rrcCs||_||_||_dSr)r:r1r2)rr:r1r2rrr rks z_ValidatorFromCallable.__init__cCsd|jdS)NzValidator.from_callable(r!)r:r#rrr r$rsz_ValidatorFromCallable.__repr__r*r cCs4||js|jrt|j}nd}t||jddS)Nr)rr)r:textr2lenr r1)rr*indexrrr r,us  z_ValidatorFromCallable.validateN)r:r0r1rr2r3rrr%r7)r"r&r'r(rr$r,rrrr r4fs   r4c@.eZdZdZdddZdd d Zdd d ZdS)rz Wrapper that runs input validation in a thread. (Use this to prevent the user interface from becoming unresponsive if the input validation takes too much time.) validatorrrrcC ||_dSr)r?)rr?rrr r zThreadedValidator.__init__r*r cCs|j|dSr)r?r,r+rrr r,szThreadedValidator.validatecs$dfdd }t|IdHdS)z: Run the `validate` function in a thread. rrcs Sr)r,rr*rrr run_validation_threadrAz?ThreadedValidator.validate_async..run_validation_threadN)rrr)rr*rCrrBr r-sz ThreadedValidator.validate_asyncN)r?rrrr7r"r&r'r(rr,r-rrrr rs   rc@seZdZdZd ddZdS) rz1 Validator class that accepts any input. r*r rrcCsdSrrr+rrr r,szDummyValidator.validateNr7)r"r&r'r(r,rrrr rsrc@s$eZdZdZddd Zdd d ZdS)r zq Validator that can be switched on/off according to a filter. (This wraps around another validator.) r?rfilterr rrcCs||_t||_dSr)r?r rE)rr?rErrr rszConditionalValidator.__init__r*r cCs|r |j|dSdSr)rEr?r,r+rrr r,szConditionalValidator.validateN)r?rrEr rrr7)r"r&r'r(rr,rrrr r s r c@r>)rz Validator class that can dynamically returns any Validator. :param get_validator: Callable that returns a :class:`.Validator` instance. get_validatorCallable[[], Validator | None]rrcCr@r)rF)rrFrrr rrAzDynamicValidator.__init__r*r cCs|pt}||dSr)rFrr,rr*r?rrr r,szDynamicValidator.validatecs$|pt}||IdHdSr)rFrr-rHrrr r-szDynamicValidator.validate_asyncN)rFrGrrr7rDrrrr rs   rN)r( __future__rabcrrcollections.abcrprompt_toolkit.eventlooprr*r filtersr r __all__ Exceptionr rr4rrr rrrrr s     ;