dZddlmZddlmZddlmZddlZddlmZddlmZddl m Z dd l m Z dd l m Z d Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!y)z/A library that is used to support our commands.)absolute_import)division)unicode_literalsN)apis)waiter) exceptions) properties) resourcesc.tjddS)z1Shortcut to get the latest Bigtable Admin client. bigtableadminv2)rGetClientInstance+lib/googlecloudsdk/api_lib/bigtable/util.pyGetAdminClientr   66rc.tjddS)z3Shortcut to get the latest Bigtable Admin messages.r r )rGetMessagesModulerrrGetAdminMessagesr$rrcdjdtjjjj gS)N/projects)joinr VALUEScoreprojectGetrrr ProjectUrlr)s1 :z0055==AACD EErc:djtd|gS)Nr locations)rr)locations r LocationUrlr#-s :<h7 88rct}tj||j}tj|||S)N)rrCloudOperationPoller operationsWaitFor)result_service operation_refmessageclientpollers r_Awaitr-1s5  &  & &~v7H7H I&  w 77rcDt}t|j||S)z5Waits for cluster long running operation to complete.)rr-projects_instances_clustersr)r*r+s r AwaitClusterr17s  & 22M7 KKrcDt}t|j||S)z6Waits for instance long running operation to complete.)rr-projects_instancesr0s r AwaitInstancer4=s  & ))=' BBrcDt}t|j||S)z9Waits for app profile long running operation to complete.)rr-projects_instances_appProfilesr0s rAwaitAppProfiler7Cs  & 55}g NNrcDt}t|j||S)z3Waits for table long running operation to complete.)rr-projects_instances_tablesr0s r AwaitTabler:Is  & 00- IIrcDt}t|j||S)z4Waits for backup long running operation to complete.)rr-#projects_instances_clusters_backupsr0s r AwaitBackupr=Os$  &  00- rcDt}t|j||S)z:Waits for logical view long running operation to complete.)rr-projects_instances_logicalViewsr0s rAwaitLogicalViewr@Ws  & 66 w OOrcDt}t|j||S)z?Waits for materialized view long running operation to complete.)rr-$projects_instances_materializedViewsr0s rAwaitMaterializedViewrC]s$  &  11=' rctjj|tjj j j|ddS)z+Get a resource reference to an app profile. projectsId instancesIdz,bigtableadmin.projects.instances.appProfilesparams collectionr REGISTRYParser rrr GetOrFail)instance app_profiles rGetAppProfileRefrQesL    ! !"))..66@@!@ " rctjj|tjj j j|ddS)z&Get a resource reference to a cluster.rEz)bigtableadmin.projects.instances.clustersrHrK)rOclusters r GetClusterRefrTqsK    ! ! "))..66@@!= " rctjj|tjj j j|ddS)z+Get a resource reference to a logical view.rEz-bigtableadmin.projects.instances.logicalViewsrHrK)rO logical_views rGetLogicalViewRefrW}sL    ! !"))..66@@!A " rcVtjj|jdS)z5Get a resource reference to a long running operation.zbigtableadmin.operations)r rLParseRelativeNamename) operations rGetOperationRefr\s%    - -nn0 rctjj|dtjj j jidS)z(Get a resource reference to an instance.rFz bigtableadmin.projects.instancesrHrK)rOs rGetInstanceRefr^sJ    ! !  ))..66@@4 " rctjj|tjj j j|ddS)z$Get a resource reference to a table.rEz'bigtableadmin.projects.instances.tablesrHrK)rOtables r GetTableRefrasK    ! ! "))..66@@!; " r"CLOUD_BIGTABLE_APP_PROFILE_WARNINGctj|j}|jd |jdjd|dg}dg}|ddD]}|jdg}|D]g}|jdj t r!|j |jdH|j |jdid}t|d kDr|d j|z }t|d kDr|d j|z }|s|tj|d j|dd |) aFormat app profile error message from API and raise new exception. The error messages returned from the backend API are not formatted well when using the default format. This raises a new generic exception with a well formatted error message built from the original response. Args: exception: HttpError raised by API. Raises: exceptions.HttpException: Reformatted error raised by API. errordetailszErrors:z!Warnings (use --force to ignore): violationstype descriptionz z{} {}r*) jsonloadscontentget startswithWARNING_TYPE_PREFIXappendlenrr HttpExceptionformat) exceptionresponseerrorswarningsdetailrf violation error_msgs rFormatErrorMessagesr|s>ZZ )) *(ll7# g  " "9 - 5 O ;&1 2(!),fL"-J v  ) )*= > m45 immM23  -)]Q X&&I[1_ V$$I  O  '!29!=yI r)"__doc__ __future__rrrrkgooglecloudsdk.api_lib.utilrrgooglecloudsdk.calliopergooglecloudsdk.corer r rrrr#r-r1r4r7r:r=r@rCrQrTrWr\r^rarpr|rrrrs 6&' ,..*)7 7 F98 L C O J P     ;'r