o I^:>@sbddlZddlmZddlmZzddlmZWney'ddlm ZYnwddl Z ddl Z ddl m Z mZddlmZed gd Zd d Zd dZddZe jddGdddeZddZddZddZddZddZe jddd Gd!d"d"eZe jddGd#d$d$eZe jddd Gd%d&d&eZ eZ!d'd(Z"Gd)d*d*eZ#dS)+Nwraps)count)getfullargspec) getargspec) Transitioner Automaton) preserveNameArgSpecargsvarargsvarkwdefaults kwonlyargskwonlydefaults annotationsc Cst|}tt|j|jtjr|jn|j|j r|j ndtjr"t|j ndtjr2|j r0t|j ndndtjr?t|j dSddS)z Normalize inspect.ArgSpec across python versions and convert mutable attributes to immutable types. :param Callable func: A function. :return: The function's ArgSpec. :rtype: ArgSpec r ) getArgsSpecr tupler rsixPY3rkeywordsrrritemsr)funcspecrr5/usr/lib/python3/dist-packages/automat/_methodical.py _getArgSpecs"  rcCs2t|j|j|jr dnd|jrdnd|jS)a0 Get the name of all arguments defined in a function signature. The name of * and ** arguments is normalized to "*args" and "**kwargs". :param ArgSpec spec: A function to interrogate for a signature. :return: The set of all argument names in `func`s signature. :rtype: Set[str] )z*argsr)z**kwargs)setr rrrr)rrrr _getArgNames2s   r cstfdd}|S)a Decorate a function so all its arguments must be passed by keyword. A useful utility for decorators that take arguments so that they don't accidentally get passed the thing they're decorating as their first argument. Only works for methods right now. cs|fi|SNr)selfkwfrrgOsz_keywords_only..gr)r%r&rr$r_keywords_onlyEs r'T)frozenc@sDeZdZdZejddZeZejddZe fddZ ddZ dS) MethodicalStatez- A state for a L{MethodicalMachine}. FreprcCsjt|j}|D]!}t|j}||s(tdj|jj|jjt|jt|jdq|j |||||dS)ac Declare a state transition within the :class:`automat.MethodicalMachine` associated with this :class:`automat.MethodicalState`: upon the receipt of the `input`, enter the `state`, emitting each output in `outputs`. :param MethodicalInput input: The input triggering a state transition. :param MethodicalState enter: The resulting state. :param Iterable[MethodicalOutput] outputs: The outputs to be triggered as a result of the declared state transition. :param Callable collector: The function to be used when collecting output return values. :raises TypeError: if any of the `outputs` signatures do not match the `inputs` signature. :raises ValueError: if the state transition from `self` via `input` has already been defined. zdmethod {input} signature {inputSignature} does not match output {output} signature {outputSignature})inputoutputinputSignatureoutputSignatureN) r argSpecissubset TypeErrorformatmethod__name__rmachine_oneTransition)r"r,enteroutputs collector inputArgsr- outputArgsrrrupon^s    zMethodicalState.uponcC|jjSr!r4r5r"rrr_namezMethodicalState._nameN) r5 __module__ __qualname____doc__attribr6r4 serializedlistr=rArrrrr)Us    "r)cCs0t||d}|durt||j}t||||S)z Get a L{Transitioner} N)getattrr initialStatesetattr)oselfsymbol automaton transitionerrrr_transitionerFromInstances  rQcCsdSr!rrrrr_emptysrRcCsdS) docstringNrrrrr _docstringsrTcCs$|jjtjjtjjfvrtddS)Nzfunction body must be empty)__code__co_coderRrT ValueError)inst attributer%rrr assertNoCodes rZc stt|jdd|}jr|}n fdd|D}t||D] \}}||f7q!t|jddd|jddd}fdd|D} | |jrT| } || fSjddjfdd| D} || fS) a Filter out arguments that were passed to input that output won't accept. :param tuple args: The *args that input received. :param dict kwargs: The **kwargs that input received. :param ArgSpec inputSpec: The input's arg spec. :param ArgSpec outputSpec: The output's arg spec. :return: The args and kwargs that output will accept. :rtype: Tuple[tuple, dict] rNcsg|] \}}|jvr|qSr)r .0nv) outputSpecrr z_filterArgs..csi|] \}}|vr||qSrrr[)passed_arg_namesrr raz_filterArgs..csi|] \}}|vr||qSrrr[)all_accepted_namesrrrds ) rzipr rrupdaterrr) r kwargs inputSpecr_ named_args return_argsnamevaluer full_kwargs return_kwargsr)rer_rcr _filterArgss   " rpF)eqhashc@sveZdZdZejddZejedZejddZ eje e ddZ ejdddZ e jddZdd d Zd d Zd S)MethodicalInputz. An input for a L{MethodicalMachine}. Fr*) validator)defaultr+initr+cC t|jSr!rr4r@rrr _buildArgSpec zMethodicalInput._buildArgSpecNcs8tjjtjtjfdd}|S)z Return a function that takes no arguments and returns values returned by output functions produced by the given L{MethodicalInput} in C{oself}'s current state. c sjg|Ri|j}\}}j|}g}|D]%}|r)||t||j|j\}} |g|Ri| } || q||Sr!)r4_state transition collectorsrArpr0append) r rh previousStater9 outTracerr:valuesr-akrmrMr"rPrrdoInputs   z(MethodicalInput.__get__..doInput)rQrNrOr r4r)r"rMtyperrrr__get__s zMethodicalInput.__get__cCr>r!r?r@rrrrArBzMethodicalInput._namer!)r5rCrDrErFrGrOrZr4rNFactorydictr~r0rurzrrArrrrrss      rsc@sZeZdZdZejddZeZejdddZej ddZ ddd Z d d Z d d Z dS)MethodicalOutputz/ An output for a L{MethodicalMachine}. Fr*rvcCrxr!ryr@rrrrzr{zMethodicalOutput._buildArgSpecNcCstdj|j|jjd)zX Outputs are private, so raise an exception when we attempt to get one. zf{cls}.{method} is a state-machine output method; to produce this output, call an input method instead.)clsr4)AttributeErrorr3r5r4r"rMrrrrr szMethodicalOutput.__get__cOs|j|g|Ri|S)z- Call the underlying method. )r4)r"rMr rhrrr__call__szMethodicalOutput.__call__cCr>r!r?r@rrrrArBzMethodicalOutput._namer!)r5rCrDrErFrGr6r4r0rurzrrrArrrrrs    rc@s.eZdZejddZejddZdddZdS)MethodicalTracerFr*Ncs t||j|jfdd}|S)Ncs|dSr!)setTrace)tracerrPrrr)sz*MethodicalTracer.__get__..setTrace)rQrNrO)r"rMrrrrrr&s  zMethodicalTracer.__get__r!)r5rCrDrFrGrOrNrrrrrr s  rcCsdtttS)z, Create a unique Python identifier. _symbol_)strnextcounterrrrrgensym0src@seZdZdZddZdddZe  ddd Zed d Zed d Z ddZ eddZ eddZ e ddZddZdS)MethodicalMachinezj A :class:`MethodicalMachine` is an interface to an `Automaton` that uses methods on a class. cCst|_i|_t|_dSr!)r _automaton _reducersr_symbolr@rrr__init__>s zMethodicalMachine.__init__NcCs|durtd|S)z L{MethodicalMachine} is an implementation detail for setting up class-level state; applications should never need to access it on an instance. Nz.MethodicalMachine is an implementation detail.)rrrrrrDs zMethodicalMachine.__get__Fcsfdd}|S)a Declare a state, possibly an initial state or a terminal state. This is a decorator for methods, but it will modify the method so as not to be callable any more. :param bool initial: is this state the initial state? Only one state on this :class:`automat.MethodicalMachine` may be an initial state; more than one is an error. :param bool terminal: Is this state a terminal state? i.e. a state that the machine can end up in? (This is purely informational at this point.) :param Hashable serialized: a serializable value to be used to represent this state to external systems. This value should be hashable; :py:func:`unicode` is a good type to use. cst|d}r |j_|S)N)r6r4rH)r)rrK) stateMethodstateinitialr"rHrr decoratorfsz*MethodicalMachine.state..decoratorr)r"rterminalrHrrrrrPszMethodicalMachine.statecfdd}|S)zM Declare an input. This is a decorator for methods. cstj|jdS)N)rOr4rN)rsrr) inputMethodr@rrrwsz*MethodicalMachine.input..decoratorrr"rrr@rr,ps zMethodicalMachine.inputcr)z Declare an output. This is a decorator for methods. This method will be called when the state machine transitions to this state as specified in the decorated `output` method. cs t|dS)N)r6r4)r) outputMethodr@rrrs z+MethodicalMachine.output..decoratorrrrr@rr-~s zMethodicalMachine.outputcCs$|j|||t|||j|<dS)z. See L{MethodicalState.upon}. N)r addTransitionrr~)r" startState inputTokenendState outputTokensr:rrrr7s z MethodicalMachine._oneTransitioncr) ctfdd}|S)Ncst|jj}||jjSr!)rQrrr|rH)rMrP decorateer"rr serializeszBMethodicalMachine.serializer..decorator..serializer)rrr@rrrsz/MethodicalMachine.serializer..decoratorrrrr@r serializers zMethodicalMachine.serializercr)rcr)NcsR|g|Ri|}i}jD]}|||j<qt|jj}|||_dSr!)rstatesrHrQrr|)rMr rhrmapping eachStaterPrrr unserializes   zFMethodicalMachine.unserializer..decorator..unserializer)rrr@rrrs z1MethodicalMachine.unserializer..decoratorrrrr@r unserializers  zMethodicalMachine.unserializercCst|j|jSr!)rrrr@rrr _setTraceszMethodicalMachine._setTracecCs*ddlm}||jdddddddS)a Generate a L{graphviz.Digraph} that represents this machine's states and transitions. @return: L{graphviz.Digraph} object; for more information, please see the documentation for U{graphviz} r) makeDigraphcSr>r!r?)rrrrz-MethodicalMachine.asDigraph..cSr>r!r?)r,rrrrrcSr>r!r?)r-rrrrr) stateAsString inputAsStringoutputAsString) _visualizerr)r"rrrr asDigraphs zMethodicalMachine.asDigraphr!)FFN)r5rCrDrErrr'rr,r-r7rrpropertyrrrrrrr8s(         r)$ collections functoolsr itertoolsrinspectrr ImportErrorrrFr_corerr _introspectionr namedtupler rr r'sobjectr)rQrRrTrZrprsrrrrrrrrrs@     . ( + "