a Oc6@sBdZddlZddlZGdddejZddZddZd d ZdS) z0 Wrapper module for asynchronous task handling. Ncs*eZdZdZdfdd ZddZZS) UpdateTaskzWrap a coroutine in a futureNcs(t|sJ|||_t|dS)zl Arguments: - coro: awaitable object - desc: (Resource, key) N)asyncio iscoroutine_descsuper__init__)selfcorodescloop __class__M/var/www/html/gsl.tools/lib/python3.9/site-packages/peeringdb/_tasks_async.pyr szUpdateTask.__init__cCs|j\}}d|jd|dS)Nz)rtag)rrespkrrr__repr__s zUpdateTask.__repr__)N)__name__ __module__ __qualname____doc__rr __classcell__rrr rrs rcCstj|ddiS)zAggregate and collect jobsZreturn_exceptionsF)rgather)jobsrrrrsrcsfdd}|S)z Decorator to convert a generator function to an async function which collects and returns generator results, returning a list if there are multiple results cs|dg}}|i|}z||}Wnty<YqdYn0t|rT|IdH}n|}||qt|dkrx|S|S)N)send StopIterationinspect isawaitableappendlenpop)akrretgenitemfuncrr_wrapped%s       z wrap_generator.._wrappedrr*r+rr)rwrap_generators r-csfdd}|S)zh Decorator to wrap an async function in an event loop. Use for main sync interface methods. cst}||i|S)N)rZget_event_loopZrun_until_complete)r#r$r r)rrr+@szrun_task.._wrappedrr,rr)rrun_task:s r.)rrrZTaskrrr-r.rrrrs