o Nh% @sddlmZddlmZddlmZmZmZmZddl m Z eGdddZ GdddZ e Z d ed eegeffd d Zd dZededZded efddZdS)) dataclasswraps)CallableListTypeTypeVar) EventEmitterc@seZdZUeed<eed<dS)HandlereventmethodN)__name__ __module__ __qualname__str__annotations__rrrE/home/air/yokohama/back/venv/lib/python3.10/site-packages/pyee/cls.pyr s  r c@s,eZdZddZddZddZddZd S) HandlerscC g|_dSN _handlersselfrrr__init__ zHandlers.__init__cCs|j|dSr)rappend)rhandlerrrrrszHandlers.appendcCs t|jSr)iterrrrrr__iter__rzHandlers.__iter__cCrrrrrrrresetrzHandlers.resetN)r rrrrr r!rrrrrs  rr returncsdtdtffdd }|S)zr Register an event handler on an evented class. See the `evented` class decorator for a full example. r r"cstt|d|S)N)r r )rrr )r r rr decorator%szon..decorator)r)r r$rr#ronsr%cstfdd}|S)Ncsg|Ri|Srr)argskwargsr rrrbound-sz_bind..boundr)rr r)rr(r_bind,sr*Cls)r)clscs8ttt|jt|jfdd}||_|S)aQ Configure an evented class. Evented classes are classes which use an EventEmitter to call instance methods during runtime. To achieve this without this helper, you would instantiate an `EventEmitter` in the `__init__` method and then call `event_emitter.on` for every method on `self`. This decorator and the `on` function help make things look a little nicer by defining the event handler on the method in the class and then adding the `__init__` hook in a wrapper: ```py from pyee.cls import evented, on @evented class Evented: @on("event") def event_handler(self, *args, **kwargs): print(self, args, kwargs) evented_obj = Evented() evented_obj.event_emitter.emit( "event", "hello world", numbers=[1, 2, 3] ) ``` The `__init__` wrapper will create a `self.event_emitter: EventEmitter` automatically but you can also define your own event_emitter inside your class's unwrapped `__init__` method. For example, to use this decorator with a `TwistedEventEmitter`:: ```py @evented class Evented: def __init__(self): self.event_emitter = TwistedEventEmitter() @on("event") async def event_handler(self, *args, **kwargs): await self.some_async_action(*args, **kwargs) ``` csN|g|Ri|t|dst|_D]}|j|jt||jqdS)N event_emitter)hasattrr r-r%r r*r )rr&r'hhandlersog_initrrinitis  zevented..init)listrr!rr)r,r3rr0revented7s-r5N) dataclassesr functoolsrtypingrrrrpyeer r rrrr%r*r+r5rrrrs