z;dZddlmZddlmZddlmZddlZddlZddlmZddlm Z ddlm Z dd lm Z dd l m Z dd l mZddlZdd lmZd ZdZdZdZdZdZdZdZ ddZy)zModule for making API requests.)absolute_import)division)unicode_literalsN) batch_helper)single_request_helper)utils)waiters)log) properties)zipc|Dcgc] \}}}|dv }}}t|ryt|sytdcc}}w)z,Checks if all requests are of list requests.)ListAggregatedList ListInstancesListManagedInstancesTFz?All requests must be either list requests or non-list requests.)allany ValueError)requests_method list_requestss 4lib/googlecloudsdk/api_lib/compute/request_helper.py_RequestsAreListRequestsr#se !)!) 61   (    }   I KKs;cg}tj|}|dvr|jdg}n|dk(r|jdg}n|dk(r|jdjj dd}t j|dD]U}|jdd }|r|d d k(r|jd |d f|j|j|gW||jd d fS)aExtracts data from one *List response page as JSON and stores in dicts. Args: response: str, The *List response in JSON service: The service which responded to *List request method: str, Method used to list resources. One of 'List' or 'AggregatedList'. errors: list, Errors from response will be appended to this list. Returns: Pair of: - List of items returned in response as dicts - Next page token (if present, otherwise None). rritemsrmanagedInstancesr/warningNcode UNREACHABLEmessage nextPageToken) jsonloadsgetGetMethodConfig relative_pathsplitsix itervaluesappendextendresponseservicererrorsritems_field_name scope_resultr!s r_HandleJsonListr63s % ZZ !( (( LL" %E '' LL+R 0E !!..'-c 27x'89   D1g WV_ 5 tWY/01 ll<##$4b9::  _d3 33cg}|dvr |j}n|dk(r |j}n|jdjj dd}|jj D]}|j j}|r@|j|jjk(r|jd|jf|jt|j |||jfS)z=Extracts data from one *List response page as Message object.rrrrr N)rrr)r*r+additionalPropertiesvaluer!r"CodeValueValuesEnumr#r.r$r/getattrr%r0s r_HandleMessageListr=_s % (( NNE ''  % %E ..'-c 27 ;; ""**g W\\W%@%@%L%LL tW__-. ll7<--/?@A< && &&r7c#K|rtsCt|dk(r5|d\}}}tj|||\}} |j | n,t j |||\}} |j | g} t|D]]\} } | s || \}}} || |||\}}|D]}||s.tj| }||_ | j|||f_| }|ryyw)aMakes a series of list and/or aggregatedList batch requests. Args: requests: A list of requests to make. Each element must be a 3-element tuple where the first element is the service, the second element is the method ('List' or 'AggregatedList'), and the third element is a protocol buffer representing either a list or aggregatedList request. http: An httplib2.Http-like object. batch_url: The handler for making batch requests. errors: A list for capturing errors. If any response contains an error, it is added to this list. response_handler: The function to extract information responses. Yields: Resources encapsulated in format chosen by response_handler as they are received from the server. rrhttp batch_urlN) _ForceBatchRequestlenrMakeSingleRequestr/r MakeRequests enumeratecopydeepcopy pageTokenr.)rrArBr3response_handlerr2r request_body responsesrequest_errors new_requestsir1request_protobufrnext_page_tokenitemnew_request_protobufs r _ListCorerU{s$   CMQ$6&.qk#gv|"7"I"I 6<#i mmN#".";";$)#i mmN#L + 8 *21+'gv'/'606 8e_$  #}}-=>)8&Wf.BCD, H? sB6C296C20C2c(t||||tS)aMakes a series of list and/or aggregatedList batch requests. Args: requests: A list of requests to make. Each element must be a 3-element tuple where the first element is the service, the second element is the method ('List' or 'AggregatedList'), and the third element is a protocol buffer representing either a list or aggregatedList request. http: An httplib2.Http-like object. batch_url: The handler for making batch requests. errors: A list for capturing errors. If any response contains an error, it is added to this list. Returns: Resources encapsulated as protocol buffers as they are received from the server. )rUr=rrArBr3s r_ListrXs" 8T9f6H IIr7ct||jjjs t d|j D]:}|j dk7s|j dk7s#t||j :yy)zChecks whether operation argument is empty. Args: operation: Operation thats checked for emptyness. service: Variable used to access service.client.MESSAGES_MODULE.Operation. Returns: True if operation is empty, False otherwise. zEoperation must be instance ofservice.client.MESSAGES_MODULE.OperationkindwarningsFT) isinstanceclientMESSAGES_MODULE Operationr all_fieldsnamer<) operationr2fields r_IsEmptyOperationrdsx Iw~~==GG H B CC##%e fz!9 5::&2 & r7cftjjjj S)z5Check if compute/force_batch_request property is set.)r VALUEScomputeforce_batch_requestGetBoolr7rrCrCs#    " " 6 6 > > @@r7c#K|ddjj5t||||tD]}| dddy#1swYyxYww)aGMakes a series of list and/or aggregatedList batch requests. This function does all of: - Sends batch of List/AggregatedList requests - Extracts items from responses - Handles pagination All requests must be sent to the same client - Compute. Args: requests: A list of requests to make. Each element must be a 3-element tuple where the first element is the service, the second element is the method ('List' or 'AggregatedList'), and the third element is a protocol buffer representing either a list or aggregatedList request. http: An httplib2.Http-like object. batch_url: The handler for making batch requests. errors: A list for capturing errors. If any response contains an error, it is added to this list. Yields: Resources in dicts as they are received from the server. rN)r]JsonResponseModelrUr6)rrArBr3rSs rListJsonrmsJ2{1~..0(D)V_M jN100s!AA  A AAc #Kt|rt||||D]} | yts3t|dk(r%|d\} }}t j | ||\}}nt j|||\}}|j|d}d}g}|s|Dcgc]}d}}t|||D]t\}}}| |\} }}t|| jjjr5t|| s(| jj dvr| }d}t#|drn|r|}n |j$}|j&r| jj(}nP|j*r| jj,}n-| jj.}n| jj0}t#|d rB|j3t5j6|||||j8j:||| D|j3t5j6||||||| q|w|rXg}t5j<||||||| | D]}||r-| r*t?j@tCjDd |yyyycc}ww)aMakes one or more requests to the API. Each request can be either a synchronous API call or an asynchronous one. For synchronous calls (e.g., get and list), the result from the server is yielded immediately. For asynchronous calls (e.g., calls that return operations like insert), this function waits until the operation reaches the DONE state and fetches the corresponding object and yields that object (nothing is yielded for deletions). Currently, a heterogeneous set of synchronous calls can be made (e.g., get request to fetch a disk and instance), however, the asynchronous requests must be homogenous (e.g., they must all be the same verb on the same collection). In the future, heterogeneous asynchronous requests will be supported. For now, it is up to the client to ensure that the asynchronous requests are homogenous. Synchronous and asynchronous requests can be mixed. Args: requests: A list of requests to make. Each element must be a 3-element tuple where the first element is the service, the second element is the string name of the method on the service, and the last element is a protocol buffer representing the request. http: An httplib2.Http-like object. batch_url: The handler for making batch requests. errors: A list for capturing errors. If any response contains an error, it is added to this list. project_override: The override project for the returned operation to poll from. progress_tracker: progress tracker to be ticked while waiting for operations to finish. no_followup: If True, do not followup operation with a GET request. always_return_operation: If True, return operation object even if operation fails. followup_overrides: A list of new resource names to GET once the operation finishes. Generally used in renaming calls. log_result: Whether the Operation Waiter should print the result in past tense of each request. log_warnings: Whether warnings for completed operation should be printed. timeout: The maximum amount of time, in seconds, to wait for the operations to reach the DONE state. Yields: A response for each request. For deletion requests, no corresponding responses are returned. rWNr?r)r2rrLr@)GlobalOperationsServiceRegionOperationsServiceZoneOperationsService#GlobalOrganizationOperationsServiceGlobalAccountsOperationsServiceproject!instanceGroupManagerResizeRequest)rtresize_request_name no_followupfollowup_overridealways_return_operation)rtrwrxry)operations_datarArBr[progress_trackerr3 log_resulttimeoutz!Some requests generated warnings:)#rrXrCrDrrErrFr/r r\r]r^r_rd __class____name__hasattrrtzonezoneOperationsregionregionOperationsglobalOperationsglobalOrganizationOperationsr.r OperationDataruraWaitForOperationsr r!r ConstructList)rrArBr3project_overrider{rwryfollowup_overridesr| log_warningsr}rSr2rrLrM new_errorsoperation_serviceresource_servicerzrrequestr1rxrtr[s rrFrFsvh' & j   #h-1"4$,QK!GV\1CC\Iz)55 ;Iz--  / (011$1.1(I2D/F*gx*&GQ 8W^^;;EEF h 0""+/ / !g y ) $' ((' ==%nn;;  __%nn== %nn== #NNGG B C  ! !! $0$R$R$W$W'"3(?     ! !! '"3(? A B no/FrH--' )  n L kk   A8 LN!xu2sBJ  J%G'J )NNFFNTTN)__doc__ __future__rrrrHr&googlecloudsdk.api_lib.computerrrr googlecloudsdk.corer r r, six.movesr rr6r=rUrXrdrCrmrFrjr7rrs&&' 7@02#*  K )4X'81hJ(*A F! aNr7