o 3b^A@sdZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z e dZ dZdZgdZeeddZes?eed ZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZdade efddZddejdefddZdS)zLibvirt event loop implementation using asyncio Register the implementation of default loop: >>> import libvirtaio >>> libvirtaio.virEventRegisterAsyncIOImpl() .. seealso:: https://libvirt.org/html/libvirt-libvirt-event.html N)AnyCallableDict GeneratorOptionalTypeVar_Tz,Wojtek Porczyk z LGPL-2.1+)getCurrentImplvirEventAsyncIOImplvirEventRegisterAsyncIOImpl ensure_futureasyncc sfeZdZdZeZdddeee gdfde de de d df fd d Z d e fd d Z dddZZS)CallbackzBase class for holding callback :param virEventAsyncIOImpl impl: the implementation in which we run :param cb: the callback itself :param opaque: the opaque tuple passed by libvirt implr cbNopaqueargskwargsreturncs4tj|i|t|j|_||_||_||_dSN)super__init__next _iden_counteridenrrr)selfrrrrr __class__,/usr/lib/python3/dist-packages/libvirtaio.pyrEs   zCallback.__init__cCsd|jj|jS)Nz <{} iden={}>)formatr__name__rrrrr__repr__LszCallback.__repr__cCs(|jjd|j|j|j|jdS)zSchedule *ff* callbackz"callback %d close(), scheduling ffN)rlogdebugrschedule_ff_callbackrr"rrrcloseOszCallback.closerN)r! __module__ __qualname____doc__ itertoolscountrrintrrrstrr#r' __classcell__rrrrr:s 2rc@s^eZdZdZdddeddfddZd eddfd d Zdd d ZdddZdeddfddZ dS) DescriptorzManager of one file descriptor :param virEventAsyncIOImpl impl: the implementation in which we run :param int fd: the file descriptor rr fdrNcCs||_||_i|_dSr)rr2 callbacks)rrr2rrrr_s zDescriptor.__init__eventcCsBt|jD]}|jdur|j|@r||j|j||jqdS)zDispatch the event to the descriptors :param int event: The event (from libvirt's constants) being dispatched N)listr3valuesr4rrr2r)rr4callbackrrr_handleds zDescriptor._handlecCstdd|jDrtdttdd|jDr+|jj|j |j t j n|jj |j tdd|jDrM|jj|j |j t jdS|jj|j dS)zRegister or unregister callbacks at event loop This should be called after change of any ``.event`` in callbacks. css$|] }|jtjtjB@VqdSr)r4libvirtVIR_EVENT_HANDLE_READABLEVIR_EVENT_HANDLE_WRITABLE.0r7rrr xsz$Descriptor.update..zTThe only event supported are VIR_EVENT_HANDLE_READABLE and VIR_EVENT_HANDLE_WRITABLEcs|] }|jtj@VqdSr)r4r9r:r<rrrr>csr?r)r4r9r;r<rrrr>r@N)anyr3r6warningswarn UserWarningrloop add_readerr2r8r9r: remove_reader add_writerr; remove_writerr"rrrupdatems*   zDescriptor.updater7 FDCallbackcCs||j|j<|dS)zAdd a callback to the descriptor :param FDCallback callback: the callback to add :rtype: None After adding the callback, it is immediately watched. N)r3rrJ)rr7rrr add_handles  zDescriptor.add_handlercCs|j|}||S)aRemove a callback from the descriptor :param int iden: the identifier of the callback :returns: the callback :rtype: FDCallback After removing the callback, the descriptor may be unwatched, if there are no more handles for it. )r3poprJ)rrr7rrr remove_handles zDescriptor.remove_handler()r7rKrN) r!r)r*r+r.rr8rJrLrNrrrrr1Ys " r1cs4eZdZdZd fdd Zdedefd d ZZS) DescriptorDictzdDescriptors collection This is used internally by virEventAsyncIOImpl to hold descriptors. rr rNcst||_dSr)rrr)rrrrrrs  zDescriptorDict.__init__r2cCst|j|}|||<|Sr)r1r)rr2 descriptorrrr __missing__s zDescriptorDict.__missing__)rr rN) r!r)r*r+rr.r1rQr0rrrrrOsrOc sVeZdZdZdededededdf fdd Zdefd d Z deddfd d Z Z S)rKzCallback for file descriptor (watcher) :param Descriptor descriptor: the descriptor manager :param int event: bitset of events on which to fire the callback rrPr4rrNcs"tj|i|||_||_dSr)rrrPr4)rrPr4rrrrrr zFDCallback.__init__cCsd|jj|j|jj|jS)Nz<{} iden={} fd={} event={}>)r rr!rrPr2r4r"rrrr#szFDCallback.__repr__cCs||_|jdS)z1Update the callback and fix descriptor's watchersN)r4rPrJ)rr4rrrrJszFDCallback.update) r!r)r*r+rr1r.rr/r#rJr0rrrrrKs "rKcszeZdZdZdededdffdd Zdefdd Zej de eddffd d Z d e ddfd dZ dfdd ZZS)TimeoutCallbackzCallback for timerrrrNcs"tj|i|d|_d|_dS)N)rrtimeout_task)rrrrrrrrRzTimeoutCallback.__init__cCsd|jj|j|jS)Nz<{} iden={} timeout={}>)r rr!rrUr"rrrr#szTimeoutCallback.__repr__ccs z |jdkr|jd}|jjd|t|EdHndVWntjy6|jjd|jYdSw||j|j |jjd|jq)zQAn actual timer running on the event loop. This is a coroutine. TrgMbP?z sleeping %rNztimer %d cancelledztimer %r callback ended) rUrr$r%asynciosleepCancelledErrorrrrrrUrrr_timers   zTimeoutCallback._timerrUcCs||_|jdkr#|jdur#|jjd|jt||jjd|_dS|jdkr@|jdurB|jjd|j|j d|_dSdSdS)z-Start or the timer, possibly updating timeoutrNztimer %r startrEz timer %r stop) rUrVrr$r%rr r[rEcancelrZrrrrJs   zTimeoutCallback.updatecs|jddtt|dS)z#Stop the timer and call ff callbackrTrUN)rJrrSr'r"rrrr's zTimeoutCallback.closer()r!r)r*r+rrr/r#rW coroutinerr[r.rJr'r0rrrrrSsrSc @s<eZdZdZd+dejddfddZdefddZd,d d Z d,d d Z d-d dZ de de ddfddZejde de ddfddZejdeeddffddZdefddZde de dejde de f ddZde de ddfdd Zde de fd!d"Zd#e dejde de fd$d%Zd&e d#e ddfd'd(Zd&e de fd)d*ZdS).r zLibvirt event adapter to asyncio. :param loop: asyncio's event loop If *loop* is not specified, the current (or default) event loop is used. NrErcCsP|pt|_i|_t||_t|jj |_ d|_ tj |d|_ |j dS)Nrr\)rWget_event_looprEr3rO descriptorslogging getLoggerrr!r$_pendingEvent _finishedset)rrErrrrs zvirEventAsyncIOImpl.__init__cCsdt|j|j|jS)Nz <{} callbacks={} descriptors={}>)r typer!r3rar"rrrr# szvirEventAsyncIOImpl.__repr__cCs|jd7_|jdS)z;Increase the count of pending affairs. Do not use directly.N)rdrfclearr"rrr _pending_inc$sz virEventAsyncIOImpl._pending_inccCs8|jdksJ|jd8_|jdkr|jdSdS)z;Decrease the count of pending affairs. Do not use directly.rriN)rdrfrgr"rrr _pending_dec)s  z virEventAsyncIOImpl._pending_deccCs0|jdt|j|j|j|j|j|j |S)z3Register this instance as event loop implementationz register()) r$r%r9virEventRegisterImpl _add_handle_update_handle_remove_handle _add_timeout_update_timeout_remove_timeoutr"rrrregister0s   zvirEventAsyncIOImpl.registerrrcCst||||jddS)z8Schedule a ff callback from one of the handles or timersr\N)r _ff_callbackrErrrrrrr&9sz(virEventAsyncIOImpl.schedule_ff_callbackcCs$|jd|t||dS)zFDirectly free the opaque object This is a coroutine. z ff_callback(iden=%d, opaque=...)N)r$r%r9virEventInvokeFreeCallbackrlrvrrrru=s  z virEventAsyncIOImpl._ff_callbackccs4|jd|jr|jEdH|jddS)zRWait for the implementation to become idle. This is a coroutine. zdrain()Nz drain ended)r$r%rdrfwaitr"rrrdrainGs  zvirEventAsyncIOImpl.draincCs|j o|j S)zReturns False if there are leftovers from a connection Those may happen if there are sematical problems while closing a connection. For example, not deregistered events before .close(). )r3rdr"rrris_idleRszvirEventAsyncIOImpl.is_idler2r4rcCsft||||j||d}|j|jvsJ|jd|||j||j|j<|j||||jS)aRegister a callback for monitoring file handle events :param int fd: file descriptor to listen on :param int event: bitset of events on which to fire the callback :param cb: the callback to be called when an event occurrs :param opaque: user data to pass to the callback :rtype: int :returns: handle watch number to be used for updating and unregistering for events .. seealso:: https://libvirt.org/html/libvirt-libvirt-event.html#virEventAddHandleFuncFunc )rPr4z4add_handle(fd=%d, event=%d, cb=..., opaque=...) = %d)rKrarr3r$r%rLrk)rr2r4rrr7rrrrnZs  zvirEventAsyncIOImpl._add_handlewatchcC8|jd|||j|}t|tsJ|j|ddS)aChange event set for a monitored file handle :param int watch: file descriptor watch to modify :param int event: new events to listen on .. seealso:: https://libvirt.org/html/libvirt-libvirt-event.html#virEventUpdateHandleFunc z!update_handle(watch=%d, event=%d))r4N)r$r%r3 isinstancerKrJ)rr{r4r7rrrror z"virEventAsyncIOImpl._update_handlec Cs|jd|z|j|}Wnty,}z|jd|jdWYd}~dSd}~wwt|ts4J|j j }||j | |usDJt |j |jdkrR|j |=|dS)aUnregister a callback from a file handle. :param int watch: file descriptor watch to stop listening on :returns: -1 on error, 0 on success .. seealso:: https://libvirt.org/html/libvirt-libvirt-event.html#virEventRemoveHandleFunc zremove_handle(watch=%d)z#remove_handle(): no such handle: %rrNrT)r$r%r3rMKeyErrorwarningrr}rKrPr2rarNlenr')rr{r7errr2rrrrps z"virEventAsyncIOImpl._remove_handlerUcCsTt|||}|j|jvsJ|jd||j||j|j<|j|d||jS)a}Register a callback for a timer event :param int timeout: the timeout to monitor :param cb: the callback to call when timeout has expired :param opaque: user data to pass to the callback :rtype: int :returns: a timer value .. seealso:: https://libvirt.org/html/libvirt-libvirt-event.html#virEventAddTimeoutFunc z0add_timeout(timeout=%d, cb=..., opaque=...) = %dr^)rSrr3r$r%rJrk)rrUrrr7rrrrqs   z virEventAsyncIOImpl._add_timeouttimercCr|)zChange frequency for a timer :param int timer: the timer to modify :param int timeout: the new timeout value in ms .. seealso:: https://libvirt.org/html/libvirt-libvirt-event.html#virEventUpdateTimeoutFunc z$update_timeout(timer=%d, timeout=%d)r^N)r$r%r3r}rSrJ)rrrUr7rrrrrr~z#virEventAsyncIOImpl._update_timeoutc Csf|jd|z|j|}Wnty,}z|jd|jdWYd}~dSd}~ww|dS)zUnregister a callback for a timer :param int timer: the timer to remove :returns: -1 on error, 0 on success .. seealso:: https://libvirt.org/html/libvirt-libvirt-event.html#virEventRemoveTimeoutFunc zremove_timeout(timer=%d)z%remove_timeout(): no such timeout: %rrNrT)r$r%r3rMrrrr')rrr7rrrrrss z#virEventAsyncIOImpl._remove_timeoutrr()rr )r!r)r*r+rWAbstractEventLooprr/r#rkrlrtr.rr&r_rurrryboolrzr9_EventCBrnrorp_TimerCBrqrrrsrrrrr s&      r rcCstS)z@Return the current implementation, or None if not yet registered) _current_implrrrrr sr rEcCst|datS)zArrange for libvirt's callbacks to be dispatched via asyncio event loop The implementation object is returned, but in normal usage it can safely be discarded. r\)r rtrr\rrrr sr r)r+rWr,rbrBr9typingrrrrrrr __author__ __license____all__getattrr objectrr1dictrOrKrSr rr rr rrrrs0    P:B