3 YN. @s,ddlmZddlmZyddlmZWn ek rHddlmZYnXddl Z ddl m Z m Z ddl mZd d Ze jd d Gd ddeZddZddZddZddZe jdddGdddeZe jd d GdddeZe jdddGdddeZeZdd ZGd!d"d"eZdS)#)wraps)count)getfullargspec) getargspecN) Transitioner Automaton) preserveNamecstfdd}|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 |f|S)N)selfkw)fr 5/usr/lib/python3/dist-packages/automat/_methodical.pygsz_keywords_only..g)r)r rr )r r_keywords_onlys rT)frozenc@sDeZdZdZejddZejZejddZe fddZ ddZ dS) MethodicalStatez- A state for a L{MethodicalMachine}. F)reprcCs`t|j}x<|D]4}t|j}||krtdj|jj|jj||dqW|jj|||||dS)z Declare a state transition within the L{MethodicalMachine} associated with this L{MethodicalState}: upon the receipt of the input C{input}, enter the state C{enter}, emitting each output in C{outputs}. zdmethod {input} signature {inputSignature} does not match output {output} signature {outputSignature})inputoutputZinputSignatureZoutputSignatureN) getArgsSpecmethod TypeErrorformat__name__machine_oneTransition)r rZenteroutputs collectorZ inputSpecrZ outputSpecr r rupon+s   zMethodicalState.uponcCs|jjS)N)rr)r r r r_name@szMethodicalState._nameN) r __module__ __qualname____doc__attribrr serializedlistrr r r r rr"s    rcCs0t||d}|dkr,t||j}t||||S)z Get a L{Transitioner} N)getattrr initialStatesetattr)oselfsymbol automaton transitionerr r r_transitionerFromInstanceDs  r/cCsdS)Nr r r r r_emptyRsr0cCsdS)Z docstringNr r r r r _docstringUsr1cCs$|jjtjjtjjfkr tddS)Nzfunction body must be empty)__code__co_coder0r1 ValueError)ZinstZ attributer r r r assertNoCodeXs  r5F)Zcmphashc@sZeZdZdZejddZejedZejddZ ejej e ddZ d ddZ d d ZdS) MethodicalInputz. An input for a L{MethodicalMachine}. F)r)Z validator)defaultrNcs8tjjtjtjfdd}|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 stjf||j}j\}}j|}g}x6|D].}|rN||j|f||}|j|q:W||S)N)r_stateZ transition collectorsr append) argskwargsZ previousStaterZ outTracerrvaluesrvalue)r+r r.r rdoInput|s   z(MethodicalInput.__get__..doInput)r/r,r-r rr)r r+typer@r )r+r r.r__get__ts  zMethodicalInput.__get__cCs|jjS)N)rr)r r r rr szMethodicalInput._name)N)rr!r"r#r$r%r-r5rr,ZFactorydictr:rBr r r r rr7is    r7c@s>eZdZdZejddZejZd ddZddZ d d Z dS) MethodicalOutputz/ An output for a L{MethodicalMachine}. F)rNcCstdj|j|jjddS)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.)clsrN)AttributeErrorrrr)r r+rAr r rrBszMethodicalOutput.__get__cOs|j|f||S)z- Call the underlying method. )r)r r+r<r=r r r__call__szMethodicalOutput.__call__cCs|jjS)N)rr)r r r rr szMethodicalOutput._name)N) rr!r"r#r$r%rrrBrGr r r r rrDs   rDc@s.eZdZejddZejddZdddZdS)MethodicalTracerF)rNcs t||j|jfdd}|S)Ncsj|dS)N)setTrace)Ztracer)r.r rrIsz*MethodicalTracer.__get__..setTrace)r/r,r-)r r+rArIr )r.rrBs zMethodicalTracer.__get__)N)rr!r"r$r%r-r,rBr r r rrHs  rHcCsdtttS)z, Create a unique Python identifier. Z_symbol_)strnextcounterr r r rgensymsrMc@s|eZdZdZddZdddZeddd Zed d Zed d Z ddZ eddZ eddZ e ddZddZdS)MethodicalMachineze A L{MethodicalMachine} is an interface to an L{Automaton} that uses methods on a class. cCst|_i|_t|_dS)N)r _automatonZ _reducersrM_symbol)r r r r__init__szMethodicalMachine.__init__NcCs|dk rtd|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.)rF)r r+rAr r rrBszMethodicalMachine.__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 initial: is this state the initial state? Only one state on this L{MethodicalMachine} may be an initial state; more than one is an error. @type initial: L{bool} @param 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.) @type terminal: L{bool} @param serialized: a serializable value to be used to represent this state to external systems. This value should be hashable; L{unicode} is a good type to use. @type serialized: a hashable (comparable) value cst|d}r|j_|S)N)rrr&)rrOr))Z stateMethodstate)initialr r&r r decorators z*MethodicalMachine.state..decoratorr )r rSZterminalr&rTr )rSr r&rrRszMethodicalMachine.statecsfdd}|S)zM Declare an input. This is a decorator for methods. cstj|jdS)N)r-rr,)r7rOrP)Z inputMethod)r r rrTsz*MethodicalMachine.input..decoratorr )r rTr )r rrs zMethodicalMachine.inputcsfdd}|S)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 L{MethodicalMachine.output} method. cs t|dS)N)rr)rD)Z outputMethod)r r rrTsz+MethodicalMachine.output..decoratorr )r rTr )r rrs zMethodicalMachine.outputcCs$|jj|||t|||j|<dS)z. See L{MethodicalState.upon}. N)rOZ addTransitiontupler:)r Z startStateZ inputTokenZendStateZ outputTokensrr r rrs  z MethodicalMachine._oneTransitioncsfdd}|S)z cstfdd}|S)Ncst|jj}||jjS)N)r/rPrOr9r&)r+r.) decorateer r r serialize;szBMethodicalMachine.serializer..decorator..serialize)r)rVrW)r )rVrrT:sz/MethodicalMachine.serializer..decoratorr )r rTr )r r serializer5s zMethodicalMachine.serializercsfdd}|S)z cstfdd}|S)NcsP|f||}i}xjjD]}|||j<q Wt|jj}|||_dS)N)rOZstatesr&r/rPr9)r+r<r=rRmappingZ eachStater.)rVr r r unserializeIs zFMethodicalMachine.unserializer..decorator..unserialize)r)rVrZ)r )rVrrTHs z1MethodicalMachine.unserializer..decoratorr )r rTr )r r unserializerCs zMethodicalMachine.unserializercCst|j|jS)N)rHrOrP)r r r r _setTraceVszMethodicalMachine._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) makeDigraphcSs|jjS)N)rr)rRr r rgsz-MethodicalMachine.asDigraph..cSs|jjS)N)rr)rr r rr^hscSs|jjS)N)rr)rr r rr^is)Z stateAsStringZ inputAsStringZoutputAsString)Z _visualizer]rO)r r]r r r asDigraphZs zMethodicalMachine.asDigraph)N)FFN)rr!r"r#rQrBrrRrrrrXr[propertyr\r_r r r rrNs      rN) functoolsr itertoolsrinspectrr ImportErrorrr$Z_corerrZ_introspectionr rsobjectrr/r0r1r5r7rDrHrLrMrNr r r rs0   !&