fdZddlmZddlmZddlmZddlmZddlmZddl m Z Gdd e Z y ) z7Common stateful utilities for the gcloud dataproc tool.)absolute_import)division)unicode_literals)apis)base) resourcesceZdZdZej j ffd ZedZ edZ edZ edZ d dZ d dZxZS) DataproczStateful utility for calling Dataproc APIs. While this currently could all be static. It is encapsulated in a class to support API version switching in future. cbtt| ||_d|_d|_d|_y)Nv1)superr __init__ release_track api_version_client _resources)selfr __class__s /lib/googlecloudsdk/api_lib/dataproc/dataproc.pyrzDataproc.__init__"s. (D"$&DDDLDOc||j%tjd|j|_|jSNdataproc)rrGetClientInstancerrs rclientzDataproc.client)s0 ||++J8H8HIdl <<rcBtjd|jSr)rGetMessagesModulerrs rmessageszDataproc.messages/s  ! !*d.>.> ??rc|jItjj|_|jj d|j |jSr)rrREGISTRYCloneRegisterApiByNamerrs rrzDataproc.resources3sF !**002do oo'' D4D4DE ??rc|jjjj|jjjj|jjjj gSN)r JobStatusStateValueValuesEnum CANCELLEDDONEERRORrs rterminal_job_stateszDataproc.terminal_job_states:sY 44>> 4499 44:: rc||jj||||S|jdk(r td|jj|||||S)aUGets the CreateClusterRequest for the appropriate api version. Args : cluster : Dataproc cluster to be created. project_id: The ID of the Google Cloud Platform project that the cluster belongs to. region : The Dataproc region in which to handle the request. request_id : A unique ID used to identify the request. action_on_failed_primary_workers : Supported only for v1 api. Raises : ValueError : if non-None action_on_failed_primary_workers is passed for v1beta2 api. Returns : DataprocProjectsRegionsClustersCreateRequest )cluster projectIdregion requestIdv1beta2zAaction_on_failed_primary_workers is not supported for v1beta2 api)r-r.r/r0actionOnFailedPrimaryWorkers)r,DataprocProjectsRegionsClustersCreateRequestr ValueError)rr- project_idr/ request_id action_on_failed_primary_workerss rGetCreateClusterRequestz Dataproc.GetCreateClusterRequestBs.(/ ]] G G H   9$  M OO == E E%E F GGrc|j}|j|j}|rt||_|j j j|S)adGets workflow template from dataproc. Args: template: workflow template resource that contains template name and id. version: version of the workflow template to get. Returns: WorkflowTemplate object that contains the workflow template info. Raises: ValueError: if version cannot be converted to a valid integer. )name)r2DataprocProjectsRegionsWorkflowTemplatesGetRequest RelativeNameintversionr"projects_regions_workflowTemplatesGet)rtemplater>r get_requests rGetRegionsWorkflowTemplatez#Dataproc.GetRegionsWorkflowTemplateksY}}HMM  " " $N&KLk ;; 9 9 = =k JJrr%)__name__ __module__ __qualname____doc__r ReleaseTrackGArpropertyrrrr+r8rC __classcell__)rs@rr r s $(#4#4#7#7    @ @     @D 'GRKrr N) rG __future__rrrgooglecloudsdk.api_lib.utilrgooglecloudsdk.calliopergooglecloudsdk.corerobjectr rrrRs, >&',()bKvbKr