o j!@stddlmZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z ddl mZmZddlmZgdZd ejvZed d d ZGd ddeeZGddde dZGdddeeefZeZd?ddZd@ddZd@ddZd@ddZd@d d!Z d@d"d#Z!d@d$d%Z"dAd&d'Z#ed(Z$dBd.d/Z%dCd2d3Z&dDd5d6Z'e ge(fe(BZ)dEd9d:Z*dFdGd=d>Z+dS)H) annotationsN)deque)Callable Generator)AbstractContextManager)GenericTypeVar)wcwidth)Event DummyContext get_cwidthsuspend_to_background_supportedis_conemu_ansi is_windowsin_main_threadget_bell_environment_variableget_term_environment_variabletake_using_weightsto_strto_intAnyFloatto_floatis_dumb_terminalzsphinx.ext.autodoc_SenderT) covariantc@sZeZdZdZ ddd d Zdd d Zdd dZdddZdddZdddZ dddZ dS)r a Simple event to which event handlers can be attached. For instance:: class Cls: def __init__(self): # Define event. The first parameter is the sender. self.event = Event(self) obj = Cls() def handler(sender): pass # Add event handler by using the += operator. obj.event += handler # Fire event. obj.event() Nsenderrhandler Callable[[_Sender], None] | NonereturnNonecCs$||_g|_|dur||7}dSdSN)r _handlers)selfrrr#/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/prompt_toolkit/utils.py__init__?s  zEvent.__init__cCs|jD]}||jqdS)z Fire event.N)r!rr"rr#r#r$__call__Hs  zEvent.__call__cCs |dS)z!Alias for just calling the event.Nr#r"r#r#r$fireMs z Event.fireCallable[[_Sender], None]cCs|j|dS)z Add another handler to this callback. (Handler should be a callable that takes exactly one parameter: the sender object.) N)r!appendr&r#r#r$ add_handlerQszEvent.add_handlercCs||jvr |j|dSdS)z6 Remove a handler from this callback. N)r!remover&r#r#r$remove_handlerZs zEvent.remove_handlerEvent[_Sender]cC|||S)zC `event += handler` notation for adding a handler. )r,r&r#r#r$__iadd__a zEvent.__iadd__cCr0)zE `event -= handler` notation for removing a handler. )r.r&r#r#r$__isub__hr2zEvent.__isub__r )rrrrrrrr)rr*rr)rr*rr/) __name__ __module__ __qualname____doc__r%r'r)r,r.r1r3r#r#r#r$r *s   r c@s$eZdZdZd ddZd dd Zd S) r z5 (contextlib.nested is not available on Py3) rrcCdSr r#r(r#r#r$ __enter__uzDummyContext.__enter__aobjectcGr9r r#)r"r<r#r#r$__exit__xr;zDummyContext.__exit__Nr4)r<r=rr)r5r6r7r8r:r>r#r#r#r$r ps r cs4eZdZdZdZdZd fdd Zdd d ZZS)_CharSizesCachez" Cache for wcwidth sizes. @rrcstt|_dSr )superr%r _long_stringsr( __class__r#r$r%s  z_CharSizesCache.__init__stringstrintcst|dkrtdt|}n tfdd|D}||<t|jkr>j}||t|jkr>|}|vr>|=|S)Nrc3s|]}|VqdSr r#).0cr(r#r$ sz._CharSizesCache.__missing__..) lenmaxr sumLONG_STRING_MIN_LENrCr+MAX_LONG_STRINGSpopleft)r"rFresult long_strings key_to_remover#r(r$ __missing__s  z_CharSizesCache.__missing__r4rFrGrrH) r5r6r7r8rPrQr%rV __classcell__r#r#rDr$r?|s r?rFrGrrHcCst|S)z? Return width of a string. Wrapper around ``wcwidth``. )_CHAR_SIZES_CACHE)rFr#r#r$r sr boolcCs ttdS)z Returns `True` when the Python implementation supports suspend-to-background. This is typically `False' on Windows systems. SIGTSTP)hasattrsignalr#r#r#r$r s r cCs tjdkS)z) True when we are using Windows. win32)sysplatformr#r#r#r$rs rcCs tjdkrddlm}|SdS)zS True when we are using Windows, but VT100 escape sequences are supported. r^ris_win_vt100_enabledF)r_r`prompt_toolkit.output.windows10rbrar#r#r$is_windows_vt100_supporteds  rdcCstjdko tjdddkS)z7 True when the ConEmu Windows console is used. r^ ConEmuANSIOFFON)r_r`osenvirongetr#r#r#r$rsrcCstjjdkS)z: True when the current thread is the main thread. _MainThread) threadingcurrent_threadrEr5r#r#r#r$rsrcCstjdd}|dvS)zD True if env variable is set to true (true, TRUE, True, 1). PROMPT_TOOLKIT_BELLtrue)1ro)rhrirjlowervaluer#r#r$rs rcCstjddS)z&Return the $TERM environment variable.TERM)rhrirjr#r#r#r$rsr_Titemslist[_T]weights list[int]Generator[_T, None, None]c cst|t|ks Jt|dksJg}g}t||D]\}}|dkr.||||q|}|}|s9tddd|D}t|}t|}d} d} | rzd} tt|||D]\} }} || | | t|krw|V|| d7<d} qY| sO| d7} qK)a Generator that keeps yielding items from the items list, in proportion to their weight. For instance:: # Getting the first 70 items from this generator should have yielded 10 # times A, 20 times B and 40 times C, all distributed equally.. take_using_weights(['A', 'B', 'C'], [5, 10, 20]) :param items: List of items to take from. :param weights: Integers representing the weight. (Numbers have to be integers, not floats.) rz+Did't got any items with a positive weight.cSsg|]}dqS)rr#)rJir#r#r$ sz&take_using_weights..TFrI)rMzipr+ ValueErrorrNrangefloat) rwryitems2weights2itemw already_taken item_count max_weightr|addingitem_iweightr#r#r$rs>   rrsCallable[[], str] | strcCt|r t|St|S)z$Turn callable or string into string.)callablerrGrrr#r#r$r rCallable[[], int] | intcCr)zTurn callable or int into int.)rrrHrrr#r#r$r'rrrrcCr)z"Turn callable or float into float.)rrrrrr#r#r$r2rrterm str | NonecCs&|dur ttjddS|dvS)z True if this terminal type is considered "dumb". If so, we should fall back to the simplest possible form of line editing, without cursor positioning and color support. Nrtru)dumbunknown)rrhrirjrq)rr#r#r$r:s rrW)rrZ)rrG)rwrxryrzrr{)rsrrrG)rsrrrH)rsrrrr )rrrrZ), __future__rrhr]r_rl collectionsrcollections.abcrr contextlibrtypingrrr __all__modulesSPHINX_AUTODOC_RUNNINGrr r dictrGrHr?rYr r rrdrrrrrvrrrrrrrr#r#r#r$s@      F )         6