.dZddlmZddlmZddlmZddlZddlmZddlmZddl m Z dd l m Z ddl Z d Zd Zd Zd ZdZe j&j(fdZe j&j(fdZe j&j(fdZ ddZdZdZy)z'Utilities for calling the Composer API.)absolute_import)division)unicode_literalsN)encoding) list_pager)apis)basecomposerv1v1beta1v1alpha2c|tjjk(rtS|tjjk(rt St SN)r ReleaseTrackBETACOMPOSER_BETA_API_VERSIONALPHACOMPOSER_ALPHA_API_VERSIONCOMPOSER_GA_API_VERSION release_tracks +lib/googlecloudsdk/api_lib/composer/util.py GetApiVersionr%s<d'',,, $$))/// %%  cJtjtt|SNr)rGetClientInstanceCOMPOSER_API_NAMErrs rrr-!  ]C EErcJtjtt|Sr)rGetMessagesModulerrrs rr"r"2r rc g}|D]_}|} t| ||jtj|t j || ||dn||t n|d}atj||S)anCollects the results of a List API call across a list of locations. Args: request_cls: class, the apitools.base.protorpclite.messages.Message class corresponding to the API request message used to list resources in a location. service: apitools.base.py.BaseApiService, a service whose list method to call with an instance of `request_cls` location_refs: [core.resources.Resource], a list of resource references to locations in which to list resources. field: str, the name of the field within the list method's response from which to extract a list of resources page_size: int, the maximum number of resources to retrieve in each API call limit: int, the maximum number of results to return. None if all available results should be returned. location_attribute: str, the name of the attribute in `request_cls` that should be populated with the name of the location Returns: A generator over up to `limit` resources if `limit` is not None. If `limit` is None, the generator will yield all resources in all requested locations. NpageSize)requestfieldlimit batch_sizebatch_size_attribute)setattr RelativeName itertoolschainr YieldFromListDEFAULT_PAGE_SIZEislice) request_clsservice location_refsr& page_sizer'location_attributeresults location_refr%s rAggregateListResultsr87s< '#lmG G')B)B)DEoo  -$U,5,=(9!+  -.G$   '5 ))rcf|s|Stj|tj|Sr)r JsonToMessage MessageToJson)metadata_value metadata_types rParseOperationJsonMetadatar>es2  ?    ( 6 6~ F HHrc |tj|Dcgc]\}}|j||c}}Scc}}w)N)keyvalue)additionalProperties)six iteritemsAdditionalProperty) dictionarymsg_typer@rAs r DictToMessagerHlsL  j1(1*#u!!c!71( (sA )Nparent)__doc__ __future__rrrr,apitools.base.pyrrgooglecloudsdk.api_lib.utilrgooglecloudsdk.callioper rCrrrrr/rGArrr"r8r>rHrrrQs.&'%',( %'!% 1 1 4 4!%)$5$5$8$8E %)$5$5$8$8E $,4 +*\Hr