ZdZddlmZddlmZddlmZddlmZddlmZ Gdde Z y ) z0Jobs client for Faultinjectiontesting Cloud SDK.)absolute_import)division)unicode_literals) list_pager)utilsc6eZdZdZddZdZ d dZdZdZy) JobsClientz.Client for Jobs in Faultinjection Testing API.Nc|xstj|_|xstj|_|jj |_y)N) api_lib_utilsGetClientInstanceclientGetMessagesModulemessagesprojects_locations_jobs _jobs_client)selfr rs 2lib/googlecloudsdk/api_lib/fault_injection/jobs.py__init__zJobsClient.__init__s>=M;;=DKA ? ? ADM ;;Dcp|jj|}|jj|S)zDescribe a job in the Project/location. Args: job: str, the name for the job being described. Returns: Described Job Resource. name)r4FaultinjectiontestingProjectsLocationsJobsGetRequestrGet)rjob describe_reqs rDescribezJobsClient.Describe!s= JJ K      ..rc|jj|}tj|j|d||dS)aYList Jobs in the Projects/Location. Args: parent: str, projects/{projectId}/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: Generator of matching jobs. )parentjobspageSize)field batch_sizelimitbatch_size_attribute)r5FaultinjectiontestingProjectsLocationsJobsListRequestr YieldFromListr)rrr$ page_sizelist_reqs rListzJobsClient.List1sQ$ KK L   # # '  rcg}|D]-}|j|jj|/|jj||}|jj ||||} |j j | S)a2Create a job in the Project/location. Args: job_id: str, the name for the job being created experiment_id: str, name of the experiment fault_targets: targets for the faults dry_run: Boolean value for dry-run parent: parent for fault resource Returns: Job. )service) experiment faultTargets)jobIdr validateOnlyr)appendrFaultInjectionTargetMatcherJob7FaultinjectiontestingProjectsLocationsJobsCreateRequestrCreate) rjob_id experiment_id fault_targetsdry_runrtargets fault_targetr create_reqs rr5zJobsClient.CreatePsG%  nn -- 3 3L 3 I & --  }7  KC MM  N     # #J //rcp|jj|}|jj|S)zDelete a Job in the Project/location. Args: job: str, the name for the job being deleted Returns: Empty Response Message. r)r7FaultinjectiontestingProjectsLocationsJobsDeleteRequestrDelete)rr delete_reqs rr?zJobsClient.Deleteps= MM N     # #J //r)NN)Nd) __name__ __module__ __qualname____doc__rrr*r5r?rrr r s'6< /&  >0@0rr N) rE __future__rrrapitools.base.pyr&googlecloudsdk.api_lib.fault_injectionrr objectr rFrrrKs'7&''Ie0e0r