o ˷e@sdZddlmZmZddlmZmZddlmZddl m Z ddl m Z m Z gdZGd d d eZGd d d ed ZGdddeZGdddeZGdddeZGdddeZGdddeZdS)zV Input validation for a `Buffer`. (Validators will be called before accepting input.) )ABCMetaabstractmethod)CallableOptionalrun_in_executor_with_context)Document) FilterOrBool to_filter)ConditionalValidatorValidationError ValidatorThreadedValidatorDummyValidatorDynamicValidatorcs>eZdZdZd dededdffdd Zdefd d ZZS) r z Error raised by :meth:`.Validator.validate`. :param cursor_position: The cursor position where the error occurred. :param message: Text. rcursor_positionmessagereturnNcst|||_||_dSN)super__init__rr)selfrr __class__P/var/www/ideatree/venv/lib/python3.10/site-packages/prompt_toolkit/validation.pyrs  zValidationError.__init__cCsd|jj|j|jS)Nz&{}(cursor_position={!r}, message={!r}))formatr__name__rrrrrr__repr__$s zValidationError.__repr__)rr) r __module__ __qualname____doc__intstrrr! __classcell__rrrrr sr c @sfeZdZdZededdfddZdeddfddZe dd e e ge fd e d e ddfddZ dS)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`. documentrNcCsdS)z Validate the input. If invalid, this should raise a :class:`.ValidationError`. :param document: :class:`~prompt_toolkit.document.Document` instance. Nrrr(rrrvalidate9szValidator.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)rrrvalidate_asyncCs  zValidator.validate_async Invalid inputF validate_func error_messagemove_cursor_to_endcCs 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-r.r/rrr from_callableNs zValidator.from_callable)r,F) rr"r#r$rr r*r+ classmethodrr&boolr2rrrrr,s"    r) metaclassc@sTeZdZdZdeegefdededdfddZdefd d Zd e ddfd d Z dS)r0z0 Validate input from a simple callable. funcr.r/rNcCs||_||_||_dSr)r6r.r/)rr6r.r/rrrrls z_ValidatorFromCallable.__init__cCsd|jdS)NzValidator.from_callable())r6r rrrr!tsz_ValidatorFromCallable.__repr__r(cCs4||js|jrt|j}nd}t||jddS)Nr)rr)r6textr/lenr r.)rr(indexrrrr*ws  z_ValidatorFromCallable.validate) rr"r#r$rr&r4rr!r r*rrrrr0gs  r0c@sFeZdZdZdeddfddZdeddfdd Zdeddfd 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.) validatorrNcC ||_dSr)r;)rr;rrrr zThreadedValidator.__init__r(cCs|j|dSr)r;r*r)rrrr*szThreadedValidator.validatecs$dfdd }t|IdHdS)z: Run the `validate` function in a thread. rNcs Sr)r*rr(rrrrun_validation_threadr=z?ThreadedValidator.validate_async..run_validation_thread)rNr)rr(r?rr>rr+sz ThreadedValidator.validate_async) rr"r#r$rrr r*r+rrrrrs rc@s"eZdZdZdeddfddZdS)rz1 Validator class that accepts any input. r(rNcCsdSrrr)rrrr*szDummyValidator.validate)rr"r#r$r r*rrrrrsrc@s8eZdZdZdededdfddZdeddfd d ZdS) r zq Validator that can be switched on/off according to a filter. (This wraps around another validator.) r;filterrNcCs||_t||_dSr)r;r r@)rr;r@rrrrszConditionalValidator.__init__r(cCs|r |j|dSdSr)r@r;r*r)rrrr*szConditionalValidator.validate) rr"r#r$rr rr r*rrrrr sr c@sReZdZdZdegeefddfddZdeddfdd Z deddfd d Z dS) rz Validator class that can dynamically returns any Validator. :param get_validator: Callable that returns a :class:`.Validator` instance. get_validatorrNcCr<r)rA)rrArrrrr=zDynamicValidator.__init__r(cCs|pt}||dSr)rArr*rr(r;rrrr*szDynamicValidator.validatecs$|pt}||IdHdSr)rArr+rBrrrr+szDynamicValidator.validate_async) rr"r#r$rrrrr r*r+rrrrrs rN)r$abcrrtypingrrprompt_toolkit.eventlooprr(r filtersr r __all__ Exceptionr rr0rrr rrrrrs   ;