VdZddlmZddlmZddlmZddlZddlZddlmZddlm Z ddl m Z ddl mZ dd l mZdd l mZdd l mZej"d Ze j&Zej"d ZdZdj-eZdZddZddZ ddZdZdZdZy)z(Utilities for building the dataflow CLI.)absolute_import)division)unicode_literalsN) exceptions) list_pager)apis)log) properties) resourcesz ^dataflowz^(.*)-windmill-(.*)-watermarkz dataflow.projects.locations.jobszDefaults to '{0}'.c tj|j}|jdijddS#t$r,t j j|jYywxYw)zExtract the error message from an HTTPError. Args: error: The error exceptions.HttpError thrown by the API client. Returns: A string describing the error. errormessagez Unknown error) jsonloadscontentget ValueErrorr errPrintresponse)r content_objs 8lib/googlecloudsdk/command_lib/dataflow/dataflow_util.pyGetErrorMessager2s]**U]]+K ??7B ' + +Ir :: GGMM%..! sAA2A87A8c|rdj|}|rdj|}|rdj|}dj|||t|S)a\Create a standard error message across commands. Args: error: The error exceptions.HttpError thrown by the API client. job_id: The job ID that was used in the command. project_id: The project ID that was used in the command. region_id: The region ID that was used in the command. Returns: str, a standard error message. z with job ID [{0}]z in project [{0}]z in regional endpoint [{0}]zFailed operation{0}{1}{2}: {3})formatr)r job_id project_id region_ids rMakeErrorMessager Cs` ! ( ( 0F$++J7J-44Y?I ) 0 0Y1@1G IIc #K |D]}|y#tj$r&}tjt ||||d}~wwxYww)aWraps a generator to catch any exceptions. Args: generator: The error exceptions.HttpError thrown by the API client. job_id: The job ID that was used in the command. project_id: The project ID that was used in the command. region_id: The region ID that was used in the command. Yields: The generated object. Raises: dataflow_exceptions.ServiceException: An exception for errors raised by the service. N)r HttpErrordataflow_exceptionsServiceExceptionr ) generatorrrrxes rYieldExceptionWrapperr)YsP <  g   <  . .FJ : <<r?r rArBrCrDrErF)rrrHs r JobsUriFromIdrKsX    !((--55??! ! "# r!c|j}|s0t}djt}tj||S)zGet region to be used in Dataflow services. Args: args: Argument passed in when running gcloud dataflow command Returns: Region specified by user from --region flag in args, then fall back to 'us-central1'. z`--region` not set; defaulting to '{0}'. In an upcoming release, users must specify a region explicitly. See https://cloud.google.com/dataflow/docs/concepts/regional-endpoints for additional details.)regionDATAFLOW_API_DEFAULT_REGIONrr warning)argsrMmsgs r GetRegionrRs? ;;&  (F % F. /  KK -r!)rrr)Nditems maxResultsNrrr)__doc__ __future__rrrrreapitools.base.pyrrgooglecloudsdk.api_lib.dataflowrr$googlecloudsdk.corer r r compileDATAFLOW_METRICS_RErNWINDMILL_WATERMARK_RErErDEFAULT_REGION_MESSAGErr r)r5rIrKrRr!rras/&' ''0M#*) !bjj-">># #BC4/66!"I,<4 '3 0Ef(*r!