TdZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd l m Z dd l mZe jj d e jj"d e jj$d iZdZe jj"fdZe jj"fdZe jj"fdZe jj"fdZGddZy)zIAPI wrapper for `gcloud network-security intercept-deployments` commands.)absolute_import)division)unicode_literalsN) list_pager)apis)waiter)base) resourcesv1alpha1v1beta1v1networksecuritycLt|}tjt|SN) GetApiVersionrGetMessagesModule _API_NAME release_track api_versions Hlib/googlecloudsdk/api_lib/network_security/intercept_deployments/api.pyrr&m,+   ; 77cLt|}tjt|Sr)rrGetClientInstancerrs rrr+rrcLt|}tjt|Sr)rrGetEffectiveApiEndpointrrs rrr0sm,+ % %i ==rc,tj|Sr)_API_VERSION_FOR_TRACKget)rs rrr5s  # #M 22rcreZdZdZdZ ddZdZdZdZ ddZ d Z d e jd fd Z y)ClientzAPI client for Intercept Deployments commands. Attributes: messages: API messages class, The Network Security API messages. c4t||_|jj|_|jj|_t ||_tj|_ |jjtt|yr)r_client'projects_locations_interceptDeployments_deployments_clientprojects_locations_operations_operations_clientrmessagesr Registry_resource_parserRegisterApiByNamerr)selfrs r__init__zClient.__init__@sr$]3DL << #llHHD%m4DM%..0D++=/rNc|jj|||}t|dr||_|jj |||}|j j |S)aCalls the CreateInterceptDeployment API. Args: parent: The parent of the deployment, e.g. "projects/myproj/locations/us-central1" forwarding_rule: The forwarding rule of the deployment, e.g. "projects/myproj/regions/us-central1/forwardingRules/my-rule" intercept_deployment_group: The deployment group of the deployment, e.g. "projects/myproj/locations/global/interceptDeploymentGroups/my-group" description: The description of the deployment. deployment_id: The ID of the deployment, e.g. "my-deployment". labels: A dictionary with the labels of the deployment. Returns: NetworksecurityProjectsLocationsInterceptDeploymentsCreateResponse )forwardingRuleinterceptDeploymentGrouplabels description)interceptDeploymentinterceptDeploymentIdparent)r)InterceptDeploymenthasattrr3ANetworksecurityProjectsLocationsInterceptDeploymentsCreateRequestr&Create) r-r6forwarding_ruleintercept_deployment_groupr3 deployment_idr2 deploymentcreate_requests rCreateDeploymentzClient.CreateDeploymentLsv422&!;3Jz=)*j]]dd&+eN  # # * *> ::rcp|jj|}|jj|S)z(Calls the DeleteInterceptDeployment API.name)r)ANetworksecurityProjectsLocationsInterceptDeploymentsDeleteRequestr&Delete)r-rCdelete_requests rDeleteDeploymentzClient.DeleteDeploymentws8]]dd eN  # # * *> ::rc,|jj|jdd}t|dr||_|jj ||dj |j}|jj|S)a9Calls the UpdateInterceptDeployment API. Args: name: The name of the deployment. description: The description of the deployment. update_fields: A dictionary of the fields to update mapped to their new values. Returns: Operation ref to track the long-running process. r2N)r2r3,)rCr4 updateMask) r)r7r r8r3@NetworksecurityProjectsLocationsInterceptDeploymentsPatchRequestjoinkeysr&Patch)r-rCr3 update_fieldsr>update_requests rUpdateDeploymentzClient.UpdateDeployment~s"22  403J z=)*j]]cc &88M..01dN  # # ) ). 99rcp|jj|}|jj|S)z%Calls the GetInterceptDeployment API.rB)r)>NetworksecurityProjectsLocationsInterceptDeploymentsGetRequestr&Get)r-rC get_requests rDescribeDeploymentzClient.DescribeDeployments8--^^ _K  # # ' ' 44rc|jj||}tj|j|||ddS)z'Calls the ListInterceptDeployments API.)r6filterinterceptDeploymentspageSize) batch_sizelimitfieldbatch_size_attribute)r)?NetworksecurityProjectsLocationsInterceptDeploymentsListRequestr YieldFromListr&)r-r6r\ page_size list_filter list_requests rListDeploymentszClient.ListDeploymentssP==``kaL  # #   $'  rcN|jj|jdS)zAConverts an Operation to a Resource to use with `waiter.WaitFor`.z-networksecurity.projects.locations.operations)r+ParseRelativeNamerC)r- operations rGetOperationRefzClient.GetOperationRefs%  2 2G rTiX)secondsc |r+tj|j|j}ntj|j}tj |||t |jdzS)aWaits for an operation to complete. Polls the Network Security Operation service until the operation completes, fails, or max_wait_seconds elapses. Args: operation_ref: A Resource created by GetOperationRef describing the Operation. message: The message to display to the user while they wait. has_result: If True, the function will return the target of the operation (the Intercept Deployment) when it completes. If False, nothing will be returned (useful for Delete operations). max_wait: The time to wait for the operation to succeed before timing out. Returns: If has_result = True, an Intercept Deployment entity. Otherwise, None. i) max_wait_ms)rCloudOperationPollerr&r(CloudOperationPollerNoResourcesWaitForint total_seconds)r- operation_refmessage has_resultmax_waitpollers rWaitForOperationzClient.WaitForOperationsp2**  " "D$;$;f55d6M6MNf >>..01D8  r)NN)NNN)__name__ __module__ __qualname____doc__r.r@rGrQrVrdrhdatetime timedeltarvrrr"r"9sT $ );V;:@5=A !x!!#. %rr")rz __future__rrrr{apitools.base.pyrgooglecloudsdk.api_lib.utilrrgooglecloudsdk.callioper googlecloudsdk.corer ReleaseTrackALPHABETAGArrrrrrr"r}rrrsP&'',.() ZI$  %)$5$5$:$:8 %)$5$5$:$:8 +/*;*;*@*@> !% 1 1 6 63ggr