tdZddlmZddlmZddlmZddlmZddlmZddlm Z dZ dd Z dd Z dd Z dd Zy )z-Utils for common operations API interactions.)absolute_import)division)unicode_literals)apis)waiter) resourcesz/transferappliance.projects.locations.operationsNc|tjdd}|j}|tj|}nW|dk(r!tj |j |}n1|dk(r!tj |j|}n tdtjj|jt}tj||dj|S)aPBlocks execution until an operation completes and handles the result. Args: operation (messages.Operation): The operation to wait on. verb (str): The verb to use in messages, such as "delete order". result_type (str|none): Type of result for retrieving once operation completes. Will skip returning result if None. client (apitools.base.py.base_api.BaseApiService): API client for loading the results and operations clients. Returns: poller.GetResult(operation). Raises: InternalError if provided `result_type` is not `appliance` or `order`. transferappliancev1alpha1 applianceorderz7The `result_type` must be `order`, `appliance` or None.) collectionz$Waiting for {} operation to complete)rGetClientInstanceprojects_locations_operationsrCloudOperationPollerNoResourcesCloudOperationPollerprojects_locations_appliancesprojects_locations_orders ValueErrorrREGISTRYParsename COLLECTIONWaitForformat) operationverb result_typeclientoperations_servicepoller operation_refs Alib/googlecloudsdk/api_lib/transfer/appliances/operations_util.py_wait_for_operationr$s  ^  # #$7 DF;;  3 34F GFk!  ( (,,.@BFg  ( (((*<>F N OO$$**nn+--  mCJJ4P c t||d|S)aBlocks execution until an operation completes and does not yield a result. Args: operation (messages.Operation): The operation to wait on. verb (str): The verb to use in messages, such as "delete order". client (apitools.base.py.base_api.BaseApiService): API client for loading the results and operations clients. Returns: poller.GetResult(operation). N)rrr$rrrs r#wait_then_yield_nothingr)As Y$v NNr%c*|dz }t||d|S)a|Blocks execution until an operation completes and returns an appliance. Args: operation (messages.Operation): The operation to wait on. verb (str): The verb to use in messages, such as "create". client (apitools.base.py.base_api.BaseApiService|None): API client for loading the results and operations clients. Returns: poller.GetResult(operation). z appliancer rr'r(s r#wait_then_yield_appliancer,Ps,$ Yk& IIr%c*|dz }t||d|S)axBlocks execution until an operation completes and returns an order. Args: operation (messages.Operation): The operation to wait on. verb (str): The verb to use in messages, such as "create". client (apitools.base.py.base_api.BaseApiService|None): API client for loading the results and operations clients. Returns: poller.GetResult(operation). z orderr r+r'r(s r#wait_then_yield_orderr.`s($ Ygf EEr%)NN)N)__doc__ __future__rrrgooglecloudsdk.api_lib.utilrrgooglecloudsdk.corerrr$r)r,r.r%r#r4s;4&',.)? !H O J Fr%