dZddlmZddlmZddlmZddlZddlZddlmZddlm Z ddl m Z dd l m Z dd l mZdd lmZdd lmZd ZdZ ddZdZddZdZdZdZy)z%Client for interaction with Dataplex.)absolute_import)division)unicode_literalsN) storage_api) storage_util)apis)waiter) exceptions) resources)yamlc.tjddSNdataplexv1)rGetClientInstance+lib/googlecloudsdk/api_lib/dataplex/util.pyrr!   D 11rc.tjddSr)rGetMessagesModulerrrGetMessageModuler%rrctjj|jd}t j |t j}t j||dj|j||S)z=Waits for the given google.longrunning.Operation to complete.z&dataplex.projects.locations.operations) collectionzWaiting for [{0}] to finish)sleep_mspre_start_sleep_ms) r REGISTRYParsenamer CloudOperationPollerrprojects_locations_operationsWaitForformat RelativeName) operationresourcerr operation_refpollers rWaitForOperationr))s$$**nn!I+-  & &!#AA &  #**=+E+E+GH+  rc t|ddra|jt|jj Dcgc]"\}}|jj ||$c}}Sycc}}w)Nlabels)keyvalue)additionalProperties)getattr LabelsValuesortedr+itemsAdditionalProperty)dataplex_resourceargsr,r-s r CreateLabelsr6<s T8T"  ( ( %T[[%6%6%89  : U  ) ) < <u = :   )   s'A0 c|jdstj|S|xstj}t j j|} |j|}tj|d}tj|jS#t$r*}tjdj!||d}~wwxYw)axReads an object's content from GCS. Args: object_url: Can be a local file path or the URL of the object to be read from gcs bucket (must have "gs://" prefix). storage_client: Storage api client used to read files from gcs. Returns: A str for the content of the file. Raises: ObjectReadError: If the read of GCS object is not successful. zgs://zutf-8)encodingzSUnable to read file {0} due to incorrect file path or insufficient read permissionsN) startswithr load_pathr StorageClientrObjectReferenceFromUrl ReadObjectio TextIOWrapperloadread Exceptionr BadFileExceptionr#) object_urlstorage_clientclient object_refbytes_iowrapperes rr>r>Is   w ' >>* %%  8[668&++33J?*   ,Hx':G 99W\\^ $$   % % "F:.    sA B** C3%CCcbdjdt|jdDS)z)Converts snake case string to camel case.c3PK|]\}}|dkDr|jn| yw)rN) capitalize).0indwords r zSnakeToCamel..is.4)#t7doo,4s$&_)join enumeratesplit)arg_strs r SnakeToCamelrYgs.  s!34 rc 6t|tr1|Dcgc]%}t|ttfr t|n|'c}S|j Dcic]2\}}t |t|ttfr t|n|4c}}Scc}wcc}}w)zQReccursive method to convert all nested snake case dictionary keys to camel case.) isinstancelistdictSnakeToCamelDictr2rY)arg_typelist_valr,r-s rr^r^os$ !  !H ht - "  !  !(  )*#u3%/d|%D 5 !%P)      s *B7Bcvt}|D]}d|vr$tjdj||j dd}t |dks0t |dj dk(st |ddk(r$tjdj||d||d<|S)zxReturns Dataplex task execution spec args as a map of key,value pairs from an input list of strings of format key=value.=zCExecution spec argument '{}' should be of the type argKey=argValue.rzEExecution spec argument '{}' should be of the format argKey=argValue.)r]argparseArgumentTypeErrorr#rWlenstrip)args_map_as_listexecution_args_map arg_entryarg_entry_splits rFetchExecutionSpecArgsrmsv#i )  & & O 6)   ooc1-O Oq q!'') *a / q! "a '  & & $fY/ .=Q-?q)*!$" r)ii)N)__doc__ __future__rrrrer?googlecloudsdk.api_lib.storagerrgooglecloudsdk.api_lib.utilrr googlecloudsdk.callioper googlecloudsdk.corer r rrr)r6r>rYr^rmrrrrts\,&' 67,..)$22 <@&  <"r