o €Ï1ã@sldZddlmZddlmZddlmZGdd„deƒZGdd„deƒZGd d „d eƒZGd d „d eƒZ d S)zIterable peek utilities.é)Úabsolute_import)Údivision)Úunicode_literalsc@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)ÚPeekeraPeeks the first element from an iterable. The returned object is another iterable that is equivalent to the original. If the object is not iterable then the first item is the object itself. Example: iterable = Peeker(iterable) first_item = iterable.Peek() assert list(iterable)[0] == first_item Attributes: _iterable: The original iterable. _peek: The first item in the iterable, or the iterable itself if its not iterable. _peek_seen: _peek was already seen by the first next() call. cCs||_| ¡|_d|_dS©NF)Ú _iterableÚ_PeekÚ_peekÚ _peek_seen)ÚselfÚiterable©r úC/tmp/google-cloud-sdk/lib/googlecloudsdk/core/util/peek_iterable.pyÚ__init__)s  zPeeker.__init__cCó|S©Nr ©r r r rÚ__iter__.ózPeeker.__iter__c Csfzt|jƒWStyYn tyd|_YdSwz|j d¡WSttttfy2Y|jSw)z'Peeks the first item from the iterable.TNr) ÚnextrÚ TypeErrorÚ StopIterationr ÚpopÚAttributeErrorÚ IndexErrorÚKeyErrorrr r rr1s   þýz Peeker._PeekcCó| ¡S©zFor Python 2 compatibility.©Ú__next__rr r rrCóz Peeker.nextc Csn|js d|_|jSzt|jƒWStyYnwz|j d¡WSty*Yt ‚ttttfy6t ‚w)z&Returns the next item in the iterable.Tr) r r rrrrrrrrrr r rrGs   ÿ ýÿzPeeker.__next__cCó|jS)z'Returns the first item in the iterable.)r rr r rÚPeek[óz Peeker.PeekN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrr"r r r rrs rc@s2eZdZdZd dd„Zedd„ƒZedd„ƒZd S) Ú TapInjectorzTap item injector.FcCs||_||_dSr)Ú_valueÚ_is_replacement)r ÚvalueÚreplacer r rrcs zTapInjector.__init__cCr!r)r)rr r rr+gr#zTapInjector.valuecCr!r)r*rr r rÚis_replacementkr#zTapInjector.is_replacementN)F)r$r%r&r'rÚpropertyr+r-r r r rr(`s  r(c@s eZdZdZdd„Zdd„ZdS)ÚTapzA Tapper Tap object.cCs|}dS)aûCalled on each item as it is fetched. Args: item: The current item to be tapped. Returns: True: The item is retained in the iterable. False: The item is deleted from the iterable. None: The item is deleted from the iterable and the iteration stops. Injector(): Injector.value is injected into the iterable. If Injector.is_replacement then the item is deleted from the iterable, otherwise the item appears in the iterable after the injected value. Tr )r ÚitemÚ_r r rr/sszTap.TapcCsdS)zCalled after the last item.Nr rr r rÚDone„szTap.DoneN)r$r%r&r'r/r2r r r rr/ps r/c@s8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) ÚTappera¢Taps an iterable by calling a method for each item and after the last item. The returned object is another iterable that is equivalent to the original. If the object is not iterable then the first item is the object itself. Tappers may be used when it is not efficient or possible to completely drain a resource generator before the resources are finally consumed. For example, a paged resource may return the first page of resources immediately but have a significant delay between subsequent pages. A tapper allows the first page to be examined and consumed without waiting for the next page. If the tapper is a filter then it can filter and display a page before waiting for the next page. Example: tap = Tap() iterable = Tapper(iterable, tap) # The next statement calls tap.Tap(item) for each item and # tap.Done() after the last item. list(iterable) Attributes: _iterable: The original iterable. _tap: The Tap object. _stop: If True then the object is not iterable and it has already been returned. _injected: True if the previous _call_on_each injected a new item. _injected_value: The value to return next. cCs"||_||_d|_d|_d|_dSr)rÚ_tapÚ_stopÚ _injectedÚ_injected_value)r r Útapr r rr¦s  zTapper.__init__cCrrr rr r rr­rzTapper.__iter__c Cs¶|jr d|_|jSzt|jƒWStyYn ty$|j ¡‚wz|j d¡WSt t tfy8Ynt yE|j ¡t‚w|jdusN|j rU|j ¡t‚d|_ |jS)z(Returns the next item in self._iterable.FrNT) r6r7rrrrr4r2rrrrr5rr r rÚ _NextItem°s0    þ  þ zTapper._NextItemcCrrrrr r rrÌr z Tapper.nextcCsX | ¡}|j |¡}|dur|j ¡t‚t|tƒr'|js$d|_||_ |j S|r+|Sq)z;Gets the next item, calls _tap.Tap() on it, and returns it.TN) r9r4r/r2rÚ isinstancer(r-r6r7r+)r r0Úinject_or_keepr r rrÐs   ôzTapper.__next__N) r$r%r&r'rrr9rrr r r rr3‰s r3N) r'Ú __future__rrrÚobjectrr(r/r3r r r rÚs   I