dZddlZddlmZddlmZddlmZddlZddlm Z ddlm Z ddl m Z dd l mZddl m Zdd l mZdd l mZGd d e j$Zy)zThe BigQuery CLI wait command.N)Optional)app)flags) client_job)utils)bigquery_command)bq_cached_client)bq_error) bq_id_utilscleZdZdZdedej ffd Zdejfde e fdZ xZ S)Waitzwait [] []namefvctt| ||tjddd|tj ddd||j |y)N fail_on_errorTzsWhen done waiting for the job, exit the process with an error if the job is still running, or ended with a failure.) flag_valueswait_for_statusDONEz;Wait for the job to have a certain status. Default is DONE.)superr __init__rDEFINE_boolean DEFINE_string_ProcessCommandRc)selfrr __class__s $platform/bq/frontend/command_wait.pyrz Wait.__init__s] $tR(   @  E   2returnc X tj|}t j j}|sZtj|ddg}t|dk7r"tjdt|fz|j}n0tj||tj j"} tj$||||j&}t)j*|t-j.j0j2di|d d |j4rtj6|rdSd S#t$rtjd|wxYw#t8$r}t;t;|Yd }~nd }~wwxYw|j4rdSd S)aFWait some number of seconds for a job to finish. Poll job_id until either (1) the job is DONE or (2) the specified number of seconds have elapsed. Waits forever if unspecified. If no job_id is specified, and there is only one running job, we poll that job. Examples: bq wait # Waits forever for the currently running job. bq wait job_id # Waits forever bq wait job_id 100 # Waits 100 seconds bq wait job_id 0 # Polls if a job is done, then returns immediately. # These may exit with a non-zero status code to indicate "failure": bq wait --fail_on_error job_id # Succeeds if job succeeds. bq wait --fail_on_error job_id 100 # Succeeds if job succeeds in 100 sec. Arguments: job_id: Job ID to wait on. secs: Number of seconds to wait (must be >= 0). zInvalid wait time: PENDINGRUNNING)bqclient state_filterz)No job_id provided, found %d running jobs) id_fallbacks identifierdefault_location)r# job_referencewaitstatus jobReferenceshow) custom_formatrN)bq_client_utils NormalizeWait ValueErrorr UsageErrorr ClientGetr ListJobRefslenr BigqueryErrorpopGetJobReferencebq_flagsLOCATIONvalueWaitJobrfrontend_utilsPrintObjectInfor ApiClientHelper JobReferenceCreater IsFailedJob StopIterationprint)rjob_idsecsclient running_jobsr)jobes r RunWithArgszWait.RunWithArgs,s,>  * *4 0d $ $ ( ( *F ++I(>l \ a $$ 73|;L:N N  #&&(m%55#,,22m    %%%  c $$  % % 2 2 9 9 N#  $$)D)DS)IQPqPC > NNd< ==>D  g Ahh""1))s*E BE1 E1 "E.1 F:FF)__name__ __module__ __qualname__usagestrr FlagValuesrsysmaxsizerintrM __classcell__)rs@rr r s< (%3E$4$4" " >* >*rr )__doc__rTtypingrabslrrr;clientsrrr0frontendrr r?r r BigqueryCmdr r/rrr^sA$ ,%%,R*  ' 'R*r