;dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l mZdd l mZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZdZGddej<ZGddeZdZ dZ!dZ"dZ#dZ$dZ%dZ&d6dZ'd Z(d!Z)d"Z*d#Z+d$Z,d6d%Z-d6d&Z.d'Z/d(Z0d)Z1d*Z2d+Z3d,Z4d-Z5d.Z6 d7d0Z7 d7d1Z8d2Z9d3Z:d4Z;d8d5Z  ' ' 44 >  81 < ==>s 3.3c |j}|jdddtjdj ||jddddj | |s$|jd dd d j | t j d j |j|y)zAdds arguments for parent of the entities. Args: parser: parser to which arguments are added. help_string: text that is prepended to help for each argument. exclude_billing_account: whether to exclude the billing account argument. z--organizationFORGANIZATION_IDzOrganization of the {0}.)r-metavar completerhelpz--folder FOLDER_IDzFolder of the {0}.)r-rVrXz--billing-accountBILLING_ACCOUNT_IDzBilling account of the {0}.zProject of the {0}.)help_text_to_prependN)add_mutually_exclusive_group add_argumentr OrganizationCompleterformatrProjectArgument AddToParser)parser help_stringexclude_billing_account entity_groups r AddParentArgsrfls446,00 % , ,[ 9   & &{ 3  !$ * 1 1+ > 077 DK rc X|jd|dtjdd|y)zAdds a location argument. Args: parser: parser to which to add args. required: whether the arguments is required. help_string: the help string. z --locationLOCATIONz.+zmust be non-empty)r-rVtyperXN)r]r RegexpValidator)rbr-rcs rAddBucketLocationArgrks4   & &u.A B  rctjj|xs4tjj j jddS)z-Returns the resource for the current project.Tr,r.r/)rr5r6rr1r2r3r4)r3s rGetProjectResourcermsK    ! ! B""''//33T3B0 " rcDtjj|dS)ztReturns the resource for the organization. Args: organization: organization. Returns: The resource. z"cloudresourcemanager.organizationsr/rr5r6) organizations rGetOrganizationResourcerqs(    ! !C " rcLtjj|dS)zbReturns the resource for the folder. Args: folder: folder. Returns: The resource. zcloudresourcemanager.foldersr/)rFoldersRegistryr6)folders rGetFolderResourcerus+  " ( ( 7 ) rcDtjj|dS)z}Returns the resource for the billing_account. Args: billing_account: billing account. Returns: The resource. zcloudbilling.billingAccountsr/ro)billing_accounts rGetBillingAccountResourcerxs(    ! !"@ " rc|jrt|jS|jrt|jS|s!|jrt |jSt |jS)zReturns the parent resource derived from the given args. Args: args: command line args. exclude_billing_account: whether to exclude the billing account argument. Returns: The parent resource. )rprqrtrurwrxrmr3rCrds rr?r?s_  "4#4#4 55 {{ T[[ )) "t';'; $T%9%9 :: dll ++rc6t||jS)zReturns the relative path to the parent from args. Args: args: command line args. exclude_billing_account: whether to exclude the billing account argument. Returns: The relative path. e.g. 'projects/foo', 'folders/1234'. )r?r7rzs rGetParentFromArgsr|s #4)@ A N N PPrcd|jr |j}nd}tt|d|S)zReturns the relative path to the bucket location from args. Args: args: command line args. Returns: The relative path. e.g. 'projects/foo/locations/bar'. - locations)rGCreateResourceNamer|)rCrGs rGetBucketLocationFromArgsrs. ]]}}HH -d3[( KKrcR|jry|jry|jryy)zReturns the id to be used for constructing resource paths. Args: args: command line args. Returns: The id to be used.. organizationsId foldersIdbillingAccountsId projectsId)rprtrw)rCs rr>r>s)   {{    rc|jrd}n |jrd}n|jrd}nd}dj||S)zReturns the collection derived from args and the suffix. Args: args: command line args. collection_suffix: suffix of collection Returns: The collection. zlogging.organizationszlogging.folderszlogging.billingAccountszlogging.projectsz{0}.{1})rprtrwr_)rCcollection_suffixprefixs rrArAsE  $F {{ F  &F F   &"3 44rcHdj|||jddS)aWCreates the full resource name. Args: parent: The project or organization id as a resource name, e.g. 'projects/my-project' or 'organizations/123'. collection: The resource collection. e.g. 'logs' resource_id: The id within the collection , e.g. 'my-log'. Returns: resource, e.g. projects/my-project/logs/my-log. z {0}/{1}/{2}/%2F)r_replace)parentr0 resource_ids rrr0s+    j+--c59 rc(d|vr|St|d|S)aCreates the full log resource name. Args: parent: The project or organization id as a resource name, e.g. 'projects/my-project' or 'organizations/123'. log_id: The log id, e.g. 'my-log'. This may already be a resource name, in which case parent is ignored and log_id is returned directly, e.g. CreateLogResourceName('projects/ignored', 'projects/bar/logs/my-log') returns 'projects/bar/logs/my-log' Returns: Log resource, e.g. projects/my-project/logs/my-log. /logs/logs)r)rlog_ids rCreateLogResourceNamerDs  M FFF 33rcP|jddd}|jddS)zExtracts only the log id and restore original slashes. Args: log_resource: The full log uri e.g projects/my-projects/logs/my-log. Returns: A log id that can be used in other commands. rrr)splitr) log_resourcers r ExtractLogIdrWs,   h *1 -& s ##rcntj|tjjddgS)a$Converts an Index Type String Literal to an Enum. Args: index_type: The index type e.g INDEX_TYPE_STRING. Returns: A IndexConfig.TypeValueValuesEnum mapped e.g TypeValueValuesEnum(INDEX_TYPE_INTEGER, 2) . Will return a Parser error if an incorrect value is provided. INDEX_TYPE_STRINGINDEX_TYPE_INTEGER) valid_choices)r ChoiceToEnumr* IndexConfigTypeValueValuesEnum) index_types rIndexTypeToEnumrds4   m33(*>? rc|rdj|}nd}|jdr/tjj dj|n|jdr/tjj dj|n?|jdr.tjj dj|tjj d y ) zPrints a message to remind the user to set up permissions for a sink. Args: destination: the sink destination (either bigquery or cloud storage). writer_identity: identity to which to grant write access. z`{0}`z!the group `cloud-logs@google.com`bigqueryzJPlease remember to grant {0} the BigQuery Data Editor role on the dataset.storagezKPlease remember to grant {0} the Storage Object Creator role on the bucket.pubsubzEPlease remember to grant {0} the Pub/Sub Publisher role on the topic.zjMore information about sinks can be found at https://cloud.google.com/logging/docs/export/configure_exportN)r_ startswithsdk_logstatusPrint) destinationwriter_identitygrantees rPrintPermissionInstructionsrwsnn_-G1GJ' NN &&,fWoi( NN &&,fWoh' NN #VG_ ..>rNct}|r>tj|}tj||j }||_|S|j ||||}|S)aReturns a LogMetric message based on a data stream or a description/filter. Args: metric_name: str, the name of the metric. description: str, a description. log_filter: str, the filter for the metric's filter field. bucket_name: str, the bucket name which ownes the metric. data: str, a stream of data read from a config file. Returns: LogMetric, the message representing the new metric. )name descriptionfilter bucketName)r*rloadr DictToMessage LogMetricr) metric_namer log_filter bucket_namedatamessagescontents metric_msgs rCreateLogMetricrsm]( yyH''(2D2DEJ!JO  ##  $J rc t}|r||_|r||_|r||_|rSt j |}t j||j}|D]}t||t|||S)aUpdates a LogMetric message given description, filter, and/or data. Args: metric: LogMetric, the original metric. description: str, updated description if any. log_filter: str, updated filter for the metric's filter field if any. bucket_name: str, the bucket name which ownes the metric. data: str, a stream of data read from a config file if any. Returns: LogMetric, the message representing the updated metric. ) r*rrrrrrrrsetattrgetattr) metricrrrrr update_data metric_diff field_names rUpdateLogMetricrsu]($FFM#F ))D/K((h6H6HIK!  fj'+z"BC" -rc|tj|}tjj |S)z!Get IAM policy, for a given view.)resource)r*7LoggingProjectsLocationsBucketsViewsGetIamPolicyRequestr& projects_locations_buckets_views GetIamPolicy)viewget_iam_policy_requests rrrsBmKKL  5 5 B B rct}|j||j|}tjj |S)z!Set IAM policy, for a given view.)policy)rsetIamPolicyRequest)r*7LoggingProjectsLocationsBucketsViewsSetIamPolicyRequestSetIamPolicyRequestr&r SetIamPolicy)rrrpolicy_requests rrrsX ](FF&::&:IG  5 5 B B rc gd}tjddtjtjdj |dS)z(Makes the base.Argument for --tags flag.)z%List of tags KEY=VALUE pairs to bind.zEach item must be expressed asz4`=`. z?Example: `123/environment=production,123/costCenter=marketing` z--tagsz KEY=VALUE T)rVriactionrXhidden)r Argumentr ArgDict UpdateActionjoin) help_partss r GetTagsArgrsF*      % % 99Z   rc t||}|sy|t|jDcgc]\}}|j||c}}Scc}}w)zMakes the tags message object.N)keyvalue)additionalProperties)rsorteditemsAdditionalProperty)rC tags_message tags_arg_nametagsrrs rGetTagsFromArgsrsb } %$  #4::<00jc5  ) )c ) ?0 sA )F)NNNN)r)=r __future__rrrapitools.base.pyrr'googlecloudsdk.api_lib.resource_managerrgooglecloudsdk.api_lib.utilr r#googlecloudsdk.callioper r +googlecloudsdk.command_lib.resource_managerr $googlecloudsdk.command_lib.util.apisr $googlecloudsdk.command_lib.util.argsrgooglecloudsdk.corerrrrrrr%rr r&r*r9rDrIrMrSrfrkrmrqrurxr?r|rr>rArrrrrrrrrrrrrrrs :&'%(;9/(B:<*.*)$$J  $"E"E E $  5>!H(   ,( QL"&5*(4& $&DLP>GK>  $ r