dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z d ejd d d e d edef dZd ejd dfdZ dd ejd ddeeddfdZdejd e j&j(deddfdZdejd e j&j(ddfdZy)z(The BigQuery CLI routine client library.)absolute_import)division)print_function)Optional) discovery)bq_error) bq_id_utilsroutines_api_client referencez,bq_id_utils.ApiClientHelper.DatasetReference max_results page_tokenfilter_expressionc|jj|j|j|||j S)aLists routines for the given dataset reference. Arguments: routines_api_client: the api client used to make the request. reference: Reference to the dataset. max_results: Number of results to return. page_token: Token to retrieve the next page of results. filter_expression: An expression for filtering routines. Returns: A dict that contains entries: 'routines': a list of routines. 'token': nextPageToken for the last page, if present. ) projectId datasetId maxResults pageTokenfilter)routineslistrrexecute)r r r r rs %platform/bq/clients/client_routine.py ListRoutinesrsH,""$ t'''' "  wy z,bq_id_utils.ApiClientHelper.RoutineReferencec2tj|tjjd |j j |j |j|jjS#tj$rYywxYw)z#Returns true if the routine exists. RoutineExistsmethod)rr routineIdF) r typecheckApiClientHelperRoutineReferencergetrrrrrBigqueryNotFoundError)r r s rrr2s !!22  $$& ))))))    ' ' sA BBBignore_not_foundreturnNc tj|tjjd |j j dit |jy#tj$r|sYywxYw)aDeletes RoutineReference reference. Args: routines_api_client: the api client used to make the request. reference: the RoutineReference to delete. ignore_not_found: Whether to ignore "not found" errors. Raises: BigqueryTypeError: if reference is not a RoutineReference. bq_error.BigqueryNotFoundError: if reference does not exist and ignore_not_found is False. DeleteRoutinerN) r r r!r"rdeletedictrrr$)r r r%s rr(r(Jsv"!!22   )  "))>+;|<wyrctj|tjjd|j j |j jS)aFGets IAM policy for the given routine resource. Arguments: apiclient: the apiclient used to make the request. reference: the RoutineReference for the routine resource. Returns: The IAM policy attached to the given routine resource. Raises: BigqueryTypeError: if reference is not a RoutineReference. GetRoutineIAMPolicyr)r1)r r r!r"r getIamPolicyr3r)r,r s rr6r6sV !!22 "     * *INN4D * E M M OOr)F)__doc__ __future__rrrtypingrgoogleapiclientrutilsrr Resourceintstrrrboolr(r!r"r/r6r)rrrAs.&%%"++=   D"++=6(- "++ = tn   :!!**;;  >P!!P**;;P Pr