B 't\# @sdddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZddlmZddlmZdd lmZd d d d ddddddddg ZGdd d eZeZeeddZddZeddddZGdd d eZGdddeZeZ d d Z!d!d Z"d"dZ#d#d$Z$d%dZ%d&dZ&d'd(Z'd)dZ(d*dZ)d+dZ*d,dZ+dS)-)unicode_literalsN)deque)partial)PY2 text_type)range)wcwidth)memoizedEvent DummyContext get_cwidthsuspend_to_background_supportedis_conemu_ansi is_windowsin_main_threadtake_using_weightstest_callable_argsto_strto_intto_floatc@sJeZdZdZdddZddZddZd d Zd d Zd dZ ddZ 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() NcCs ||_g|_|dk r||7}dS)N)sender _handlers)selfrhandlerr3lib/python3.7/site-packages/prompt_toolkit/utils.py__init__4szEvent.__init__cCsx|jD]}||jqWdS)z Fire event. N)rr)rrrrr__call__;s zEvent.__call__cCs |dS)z# Alias for just calling the event. Nr)rrrrfire@sz Event.firecCs0t|s tt|s td||j|dS)z Add another handler to this callback. (Handler should be a callable that takes exactly one parameter: the sender object.) z%%r doesn't take exactly one argument.N)callableAssertionError_func_takes_one_arg TypeErrorrappend)rrrrr add_handlerDs  zEvent.add_handlercCs||jkr|j|dS)z6 Remove a handler from this callback. N)rremove)rrrrrremove_handlerRs zEvent.remove_handlercCs|||S)z3 `event += handler` notation for adding a handler. )r%)rrrrr__iadd__Ys zEvent.__iadd__cCs|||S)z5 `event -= handler` notation for removing a handler. )r')rrrrr__isub__^s zEvent.__isub__)N) __name__ __module__ __qualname____doc__rrrr%r'r(r)rrrrr s Z signaturecCst|ttfsttdk rvy t|}Wn$tk rJt|}|t|<YnXy|j|Wntk rndSXdSnlt|t rdSt |}dd}||}|j dk rdSt |jt |jpgt |kot |jkSSdS)zP Return True when this function can be called with the given arguments. NFTcSs:|\}}}}|dddgkr*|dd}t||||S)Nrr r)inspectZArgSpec)specargsvarargsZvarkwdefaultsrrr drop_selfs  z%test_callable_args..drop_self) isinstancelisttupler!_inspect_signature_signatures_cacheKeyErrorZbindr#rr.Z getargspecr1lenr0r2)funcr0Zsigr/r3rrrrjs(    i)maxsizecCs t|dgS)zR Test whether the given function can be called with exactly one argument. N)r)r;rrrr"sr"c@s eZdZdZddZddZdS)r z5 (contextlib.nested is not available on Py3) cCsdS)Nr)rrrr __enter__szDummyContext.__enter__cGsdS)Nr)rarrr__exit__szDummyContext.__exit__N)r*r+r,r-r=r?rrrrr scs0eZdZdZdZdZfddZddZZS)_CharSizesCachez" Cache for wcwidth sizes. @cstt|t|_dS)N)superr@rr _long_strings)r) __class__rrrsz_CharSizesCache.__init__cst|dkrtdt|}ntfdd|D}||<t|jkr|j}||t|jkr||}|kr||=|S)Nr rc3s|]}|VqdS)Nr).0c)rrr sz._CharSizesCache.__missing__..) r:maxrsumLONG_STRING_MIN_LENrDr$MAX_LONG_STRINGSpopleft)rstringresultZ long_stringsZ key_to_remover)rr __missing__s  z_CharSizesCache.__missing__) r*r+r,r-rKrLrrP __classcell__rr)rErr@s  r@cCst|S)z? Return width of a string. Wrapper around ``wcwidth``. )_CHAR_SIZES_CACHE)rNrrrr scCs ttdS)z Returns `True` when the Python implementation supports suspend-to-background. This is typically `False' on Windows systems. SIGTSTP)hasattrsignalrrrrrscCs tjdS)z) True when we are using Windows. win)sysplatform startswithrrrrrscCsddlm}to|S)zS True when we are using Windows, but VT100 escape sequences are supported. r)is_win_vt100_enabled)Zprompt_toolkit.output.windows10rZr)rZrrris_windows_vt100_supporteds r[cCstotjdddkS)z7 True when the ConEmu Windows console is used. Z ConEmuANSIZOFFZON)rosenvirongetrrrrrscCstjjdkS)z: True when the current thread is the main thread. Z _MainThread) threadingZcurrent_threadrEr*rrrrrscCs tjdd}tr|d}|S)z( Return the $TERM environment variable. ZTERMzutf-8)r\r]r^rdecode)Ztermrrrget_term_environment_variables rbc cs@t|tstt|tsttdd|Ds2tt|t|ksFtt|dksVtg}g}x2t||D]$\}}|dkrj||||qjW|}|}|stddd|D}t|}t|}d}xrd} x`| r.d} xPtt |||D]<\} } } || || t |kr| V|| d 7<d} qWqW|d 7}qWd S) 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.) css|]}t|tVqdS)N)r4int)rFirrrrHsz%take_using_weights..rz+Did't got any items with a positive weight.cSsg|]}dqS)rr)rFrdrrr -sz&take_using_weights..TFr N) r4r5r!allr:zipr$ ValueErrorrIrfloat) itemsZweightsZitems2Zweights2rdwZ already_takenZ item_countZ max_weightZaddingZitem_iitemZweightrrrr s:   cCst|rt|St|SdS)z& Turn callable or string into string. N)r rr)valuerrrrAs cCst|rt|St|SdS)z Turn callable or int into int. N)r rrc)rmrrrrIs cCst|rt|St|SdS)z$ Turn callable or float into float. N)r rri)rmrrrrQs ),Z __future__rr.r\rUrWr_weakref collectionsr functoolsrZsixrrZ six.movesrrcacher __all__objectr WeakKeyDictionaryr8getattrr7rr"r dictr@rRr rrr[rrrbrrrrrrrrsV      E . ' 7