o  @sPdZddlmZddlZdgZddZddZddd d d dd d def ddZdS)zGA helper function that executes a series of List queries for many APIs.)encodingN YieldFromListcCsBt|tjr t||St|dkr|Stt||d|ddS)a-Gets a possibly nested attribute. Same as getattr() if attribute is a string; if attribute is a tuple, returns the nested attribute referred to by the fields in the tuple as if they were a dotted accessor path. (ex _GetattrNested(msg, ('foo', 'bar', 'baz')) gets msg.foo.bar.baz rN) isinstancesix string_typesgetattrlen_GetattrNested)message attributer D/tmp/google-cloud-sdk/lib/third_party/apitools/base/py/list_pager.pyr s  r cCsft|tjr t|||St|dkrtdt|dkr$t||d|Stt||dd|d|S)a:Sets a possibly nested attribute. Same as setattr() if attribute is a string; if attribute is a tuple, sets the nested attribute referred to by the fields in the tuple as if they were a dotted accessor path. (ex _SetattrNested(msg, ('foo', 'bar', 'baz'), 'v') sets msg.foo.bar.baz to 'v' rzNeed an attribute to setrN)rrrsetattrr ValueErrorr )r r valuer r r_SetattrNested,s   rdListitems pageToken nextPageToken maxResultsc cst|}t||d|dus|rl| r(|durd} nt||p |} t|| | t||||d} | | |}|r?tt||}|D]}|V|durKqA|d8}|sTdSqAt| | }|s^dSt||||dus|sdSdS)aMake a series of List requests, keeping track of page tokens. Args: service: apitools_base.BaseApiService, A service with a .List() method. request: protorpc.messages.Message, The request message corresponding to the service's .List() method, with all the attributes populated except the .maxResults and .pageToken attributes. global_params: protorpc.messages.Message, The global query parameters to provide when calling the given method. limit: int, The maximum number of records to yield. None if all available records should be yielded. batch_size: int, The number of items to retrieve per request. method: str, The name of the method used to fetch resources. field: str, The field in the response that will be a list of items. predicate: lambda, A function that returns true for items to be yielded. current_token_attribute: str or tuple, The name of the attribute in a request message holding the page token for the page being requested. If a tuple, path to attribute. next_token_attribute: str or tuple, The name of the attribute in a response message holding the page token for the next page. If a tuple, path to the attribute. batch_size_attribute: str or tuple, The name of the attribute in a response message holding the maximum number of results to be returned. None if caller-specified batch size is unsupported. If a tuple, path to the attribute. get_field_func: Function that returns the items to be yielded. Argument is response message, and field. Yields: protorpc.message.Message, The resources listed by the service. N) global_paramsr)rCopyProtoMessagerminrlistfilterr )servicerequestrlimit batch_sizemethodfield predicatecurrent_token_attributenext_token_attributebatch_size_attributeget_field_funcrequest_batch_sizeresponseritemtokenr r rrAs8 (       )__doc__apitools.base.pyrr__all__r rrr r r rs