BzdZddlmZddlmZddlmZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl m ZddlmZddlm Z ddlm Zdd lmZdd lmZdd lmZdd lmZddlZe j:d Ze j:dZdZ ddZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*dZ+ ddZ,dZ-dZ.y) z%Bigtable authorized views API helper.)absolute_import)division)unicode_literalsN)encoding) exceptions)util)yaml) console_io) resource_diff)edita To create an authorized view, specify a JSON or YAML formatted representation of a valid authorized view protobuf. Lines beginning with "#" are ignored. Example: { "subsetView": { "rowPrefixes": ["store1#"], "familySubsets": { "column_family_name": { "qualifiers":["address"], "qualifierPrefixes":["tel"] } } }, "deletionProtection": true } a Please pecify a JSON or YAML formatted representation of the new authorized view. Lines beginning with "#" are ignored. Example: { "subsetView": { "rowPrefixes": ["store1#"], "familySubsets": { "column_family_name": { "qualifiers":["address"], "qualifierPrefixes":["tel"] } } }, "deletionProtection": true } Current authorized view: c|jr&t|j|j|_nt d|j|_d|j_|S)a Parse argument and construct create authorized view request. Args: unused_ref: the gcloud resource (unused). args: input arguments. req: the real request to be sent to backend service. Returns: The real request to be sent to backend service. T) is_create pre_encodedN)definition_file/ParseAuthorizedViewFromYamlOrJsonDefinitionFilerauthorizedView!PromptForAuthorizedViewDefinitionname) unused_refargsreqs 7lib/googlecloudsdk/api_lib/bigtable/authorized_views.py!ModifyCreateAuthorizedViewRequestrXs\ H d..C ;D$4$4C !# *cltjj}|r t}n t ||} t j |} tj|}|s t|tj||}|S#t j$r tdt j$r}tdj|d}~wwxYw#tj $r}t#dj|d}~wt$$r}t#dj|d}~wwxYw)aPrompt user to fill out a JSON/YAML format representation of an authorized view. Returns the parsed authorized view proto message from user's response. Args: is_create: True if the prompt is for creating an authorized view. False if the prompt is for updating an authorized view. pre_encoded: True if all binary fields in the authorized view definition are already Base64-encoded. We skip the step of applying Base64 encoding in this case. current_authorized_view: The current authorized view definition. Only used in the update case to be included as part of the initial commented prompt. Returns: an authorized view proto message with fields filled accordingly. Raises: ChildProcessError if the user did not save the temporary file. ChildProcessError if there is a problem running the editor. ValueError if the user's response does not follow YAML or JSON format. ValueError if the YAML/JSON object cannot be parsed as a valid authorized View. zEdit aborted by user.z1There was a problem applying your changes. [{0}].NzGProvided response is not a properly formatted YAML or JSON file. [{0}].zEProvided response cannot be parsed as a valid authorized view. [{0}].)rGetAdminMessagesAuthorizedView%BuildCreateAuthorizedViewFileContents%BuildUpdateAuthorizedViewFileContentsr OnlineEditNoSaveExceptionChildProcessErrorEditorExceptionformatr load*Base64EncodingYamlAuthorizedViewDefinitionrPyValueToMessageYAMLParseError ValueErrorAttributeError) rrcurrent_authorized_viewauthorized_view_message_type help_textcontenteauthorized_view_to_parseauthorized_views rrrys'4"&!6!6!8!G!G57I5Iooi(G#yy1 01IJ//$&>O /   5 3 44   ;BB1E      &)   O  s;B 8C 0C9CCD3.D D3D..D3ctjj} tj|}|s t |t j||}|S#tjtjf$r}tjd|d}~wt$r }tdj||d}~wwxYw)aCreate an authorized view proto message from a YAML or JSON formatted definition file. Args: file_path: Path to the YAML or JSON definition file. pre_encoded: True if all binary fields in the authorized view definition are already Base64-encoded. We skip the step of applying Base64 encoding in this case. Returns: an authorized view proto message with fields filled accordingly. Raises: BadArgumentException if the file cannot be read. BadArgumentException if the file does not follow YAML or JSON format. ValueError if the YAML/JSON object cannot be parsed as a valid authorized view. z--definition-fileNz>File [{0}] cannot be parsed as a valid authorized view. [{1}].)rrrr load_pathr&rr' FileLoadErrorr(calliope_exceptionsBadArgumentExceptionr*r)r$) file_pathrr,r0r1r/s rrrs$"&!6!6!8!G!G#~~i8 01IJ//$&>O    d11 2K  2 23F JJ  HOO q  s#8A#B?=B B?B::B?c|rd|vr|S|d}d|vr,|jdgDcgc] }t|c}|d<d|vr|djD]m\}}d|vr,|jdgDcgc] }t|c}|d<d|vr,|jdgDcgc] }t|c}|d<||d|<o|Scc}wcc}wcc}w)z\Apply base64 encoding to all binary fields in the authorized view definition in YAML format. subsetView rowPrefixes familySubsets qualifiersqualifierPrefixes)get Utf8ToBase64itemsyaml_authorized_viewyaml_subset_views family_name family_subsets rr&r& 5I!I ),7&&!1!5!5mR!H'!HA Q!H']#((&6&G&M&M&O" ]  &%2%6%6|R%H' %HLO%H' l#  -%2%6%67JB%O. %OLO%O. )*8E' 4'P ' ' . C3C #Cc|rd|vr|S|d}d|vr,|jdgDcgc] }t|c}|d<d|vr|djD]m\}}d|vr,|jdgDcgc] }t|c}|d<d|vr,|jdgDcgc] }t|c}|d<||d|<o|Scc}wcc}wcc}w)z\Apply base64 decoding to all binary fields in the authorized view definition in YAML format.r9r:r;r<r=)r> Base64ToUtf8r@rAs r*Base64DecodingYamlAuthorizedViewDefinitionrKrGrHcxtjtjtj|S)z)Encode a utf-8 string as a base64 string.)six ensure_textbase64 b64encode ensure_binary)rDs rr?r? s' ))#*;*;A*>? @@rc tjtj|S#tt j f$r }tdj||d}~wwxYw)z)Decode a base64 string as a utf-8 string.z_Error decoding base64 string [{0}] in the current authorized view definition into utf-8. [{1}].N) rMrNrO b64decode TypeErrorbinasciiErrorr)r$rDerrors rrJrJsX ??6++A. // X^^ $  ))/5)9 s'*A#AA#cb| |jy|j}|j|jD] }t||j`|jjD]F}|j }|j D] }t||jD] }t|Hyy)z>Raises a ValueError if the view contains non-ascii characters.N)r9r: CheckAsciir;additionalPropertiesvaluer<r=)r1 subset_view row_prefixadditional_propertyrF qualifierqualifier_prefixs r(CheckOnlyAsciiCharactersInAuthorizedViewrbs : : B **+(!-- .**88MM)//m$//)90+== #$> N+rc |jdy#t$r }tdj||d}~wwxYw)zCheck if a string is ascii.asciiztNon-ascii characters [{0}] found in the current authorized view definition, please use --pre-encoded instead. [{1}].N)decode UnicodeErrorr)r$rWs rrZrZ.sBHHW   @@Fq%@P s =8=c*tj}tjD]H}|j d|r|j d|j ||j dJ|j d|j S)zQBuilds the help text for creating an authorized view as the initial file content.#  )ioStringIO CREATE_HELP splitlineswritegetvalue)buflines rrr9sh #$$&dIIcN  iinIIdOIIdO ' ))D/ rc d}|jr&t|j|j|_nBt |j |j }t d|j||_|jj td|}|jj td|}|jr"|t |j d}tj|}|jj|jj|_|jj|jj|_ tj}tj ||}|j#d| |j%r)t'j(d |j%zd nt'j(d d d|j_|j,rd |_|S)aParse argument and construct update authorized view request. Args: original_ref: the gcloud resource. args: input arguments. req: the real request to be sent to backend service. Returns: The real request to be sent to backend service. NF)rrr+r]deletion_protection) check_ascii)originalchangeddefault)outzLDifference between the current authorized view and the new authorized view: T)message cancel_on_noz5The authorized view will NOT change with this update.)rrrrGetCurrentAuthorizedView RelativeNamerr9AddFieldToUpdateMaskdeletionProtection interactivecopydeepcopyrkrlr ResourceDiffPrintrpr PromptContinuerignore_warningsignoreWarnings) original_refrrr+new_authorized_viewrqdiffers r!ModifyUpdateAuthorizedViewRequestrFs! H d..C 7!!#)9)9%9;$$ 7C  "". }c 2C**6 4c :C & 8  # # %5! --(?@ $$0'*'9'9'D'D$ ,,8   / /, ++-C  ' '(2EF LLL$ ||~$ LLN   I!# C *rc(tj}tjj|} |jj |}|r t ||S#tj$r}tj|d}~wwxYw)aGet the current authorized view resource object given the authorized view name. Args: authorized_view_name: The name of the authorized view. check_ascii: True if we should check to make sure that the returned authorized view contains only ascii characters. Returns: The view resource object. Raises: ValueError if check_ascii is true and the current authorized view definition contains invalid non-ascii characters. )rN) rGetAdminClientr=BigtableadminProjectsInstancesTablesAuthorizedViewsGetRequest)projects_instances_tables_authorizedViewsGetrbapi_exceptions HttpErrorr HttpException)authorized_view_nameruclientrequestr1rXs rr|r|s    &  ! ! # a a  b '*FFJJO.?   ! !*  " "5 ))*s)A$$B7B  Bctj|}|s t|}|dk(r*tjt j |dS|dk(r(tjtj|Sy)z=Serializes a authorized view protobuf to either JSON or YAML.json)indentr N) r MessageToDictrKrM text_typerdumpsr dump)r1rserialized_formatauthorized_view_dicts rSerializeToJsonOrYamlrsq"//@ E& ==$8C DD& ==#78 99!rctj}tjD]H}|j d|r|j d|j ||j dJt ||}|jD]H}|j d|r|j d|j ||j dJ|j d|j S)a1Builds the help text for updating an existing authorized view. Args: current_authorized_view: The current authorized view resource object. pre_encoded: When pre_encoded is False, user is passing utf-8 values for binary fields in the authorized view definition and expecting us to apply base64 encoding. Thus, we should also display utf-8 values in the help text, which requires base64 decoding the binary fields in the `current_authorized_view`. Returns: A string containing the help text for update authorized view. rhrirj)rkrl UPDATE_HELPrnrorrp)r+rrqrr#serialized_original_authorized_views rrrs #$$&dIIcN  iinIIdOIIdO ')>{)%2<<>dIIcN  iinIIdOIIdO ? ))D/ rcv|j}|r#|j|dk(r |dz|z|_|S||_|S)zAdding a new field to the update mask of the UpdateAuthorizedViewRequest. Args: field: the field to be updated. req: the original UpdateAuthorizedViewRequest. Returns: req: the UpdateAuthorizedViewRequest with update mask refreshed. r,) updateMaskcount)fieldr update_masks rr~r~sI+1$"S(50cn *CN *r)N)r)/__doc__ __future__rrrrOrUrrkrtextwrapapitools.base.pyrrrgooglecloudsdk.api_lib.bigtablergooglecloudsdk.api_lib.utilgooglecloudsdk.callioper5googlecloudsdk.corer googlecloudsdk.core.consoler googlecloudsdk.core.resourcer googlecloudsdk.core.utilr rMdedentrmrrrrr&rKr?rJrbrZrrr|rrr~rrrs ,&'  %902E$26) hoo .hoo 2 D59:z"J00A %& M `*>5; : F r