dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z d Zd ZGd d eZy)z5Utilities Assured Workloads API, Workloads Endpoints.)absolute_import)division)unicode_literals) list_pager) message_util)util)waiter) resourcesz&Creating Assured Workloads environmentcxtjj|jd}|j S)Nz2assuredworkloads.organizations.locations.workloads collection)r REGISTRYParseRelativeNamenameSelfLink)resourceworkloads /lib/googlecloudsdk/api_lib/assured/workloads.pyGetWorkloadURIrs8    1 1mmE2(    cJeZdZdZd dZd dZdZdZdZdZ d Z d Z d Z y)WorkloadsClientz.Client for Workloads in Assured Workloads API.ctj|||_tj||_||_|jj |_y)N)rGetClientInstanceclientGetMessagesModulemessages_release_track!organizations_locations_workloads_service)self release_trackno_https r__init__zWorkloadsClient.__init__*sA((@DK**=9DM'DKKAADMrNc|jj||}tj|j|d||dS)a+List all Assured Workloads environments belonging to a given parent organization. Args: parent: str, the parent organization of the Assured Workloads environment to be listed, in the form: organizations/{ORG_ID}/locations/{LOCATION}. limit: int or None, the total number of results to return. page_size: int, the number of entries in each batch (affects requests made, but not the yielded results). Returns: A list of all Assured Workloads environments belonging to a given parent organization. )parentpageSize workloadsN)field batch_sizelimitbatch_size_attribute)r:AssuredworkloadsOrganizationsLocationsWorkloadsListRequestr YieldFromListr )r!r&r+ page_sizelist_reqs rListzWorkloadsClient.List0sN}}WW XH  # # !  rctj||||j}|jjj |}|j |tS)aCreate a new Assured Workloads environment in the parent organization. Args: parent: str, the parent organization of the Assured Workloads environment to be created, in the form: organizations/{ORG_ID}/locations/{LOCATION}. external_id: str, the identifier that identifies this Assured Workloads environment externally. workload: Workload, new Assured Workloads environment containing the values to be used. Returns: The created Assured Workloads environment resource. )rCreateCreateRequestrrrCreateWaitForOperation%WORKLOAD_CREATION_IN_PROGRESS_MESSAGE)r!r& external_idr create_reqops rr4zWorkloadsClient.CreateJsS11VXt':':J  6 6 = =j IB  %J KKrc|jj||}|jjj |S)a[Delete an existing Assured Workloads environment. Args: name: str, name of the Assured Workloads environtment to be deleted, in the form: organization/{ORG_ID}/locations/{LOCATION}/workloads/{WORKLOAD_ID} etag: str, the etag string acquired by reading the Workload. Returns: Empty response message. )retag)r >z JJrc|jj|}|jjj |S)a)Enable resource violation monitoring for a workload. Args: name: str, name of the Assured Workloads for which Resource Monitoring is enabled, in the form: organization/{ORG_ID}/locations/{LOCATION}/workloads/{WORKLOAD_ID} Returns: Empty response message. r@)rNAssuredworkloadsOrganizationsLocationsWorkloadsEnableResourceMonitoringRequestrrEnableResourceMonitoring)r!r enable_reqs rrMz(WorkloadsClient.EnableResourceMonitoringsCmm nJ 55NN  rc|j|j}tj|jj |jj }tj|||S)aGWaits for the given google.longrunning.Operation to complete. Args: operation: The operation to poll. progress_message: String to display for default progress_tracker. Raises: apitools.base.py.HttpError: if the request returns an HTTP error Returns: The created Environment resource. )GetOperationResourcerr CloudOperationPollerrr"organizations_locations_operationsWaitFor)r! operationprogress_message operation_refpollers rr5z WorkloadsClient.WaitForOperationsX--inn=M  ( ( 55 66F >>&-1A BBrcDtjj|dS)Nz3assuredworkloads.organizations.locations.operationsr )r rr)r!rs rrPz$WorkloadsClient.GetOperationResources&    / / N 0 r)F)Nd) __name__ __module__ __qualname____doc__r$r1r4r=rDrJrMr5rPrrrr's86B 4L(L"K$K,(C(rrN)r] __future__rrrapitools.base.pyrgooglecloudsdk.api_lib.assuredrrgooglecloudsdk.api_lib.utilr googlecloudsdk.corer r6robjectrr^rrres><&''7/.))Q%[f[r