a Oci@sJdZddlmZGdddZddZddZd d Zd d Zd dZdS)z Wrapper module for sequential task (continuation) handling. Use as fallback for _tasks_async when async/await is not available. ) GeneratorTypec@s8eZdZdZddZddZddZdd Zd d Zd S) UpdateTaskz)Wrap a generator in a task-like interfacecCst|tsJ||_||_dS)zk Arguments: - gen: generator object - desc: (Resource, key) N) isinstancer_gen_desc)selfgendescr R/var/www/html/gsl.tools/lib/python3.9/site-packages/peeringdb/_tasks_sequential.py__init__ szUpdateTask.__init__cCs|j\}}d|jd|dS)Nz)rtag)rrespkr r r __repr__s zUpdateTask.__repr__cCsdSNr rr r r cancelszUpdateTask.cancelcCs|jSr)rrr r r __iter__szUpdateTask.__iter__cCs |j|Sr)rsend)rxr r r rszUpdateTask.sendN) __name__ __module__ __qualname____doc__r rrrrr r r r rs  rccs|D]}|EdHqdS)zAggregate and collect jobsNr )jobsZjobr r r gather#srcCs|S)z8Identity decorator (only needed for async compatibility)r funcr r r wrap_generator)srcCst|ttfrt|S|SdSr)rrr _consume_task)itemr r r _consume_task_or_generator.sr"cCs\dg}}z||}Wnty.YqDYn0t|}||q t|dkrX|S|S)N)r StopIterationr"appendlenpop)rrretr!r r r r 5s     r csfdd}|S)zo Decorator to collect and return generator results, returning a list if there are multiple results cs|i|}t|Sr)r )akrrr r _wrappedJszrun_task.._wrappedr )rr,r rr run_taskDs r-N) rtypesrrrrr"r r-r r r r s