oF"dZddlmZddlmZddlmZddlZddlZddlmZddl m Z ddl m Z d Z d d d d ZdddZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZd Zd!Z d"Z!d#Z"d$Z#d%Z$d&Z%d'Z&d(Z'd)Z(d*Z)d+Z*d,Z+y)-zResource parsing helpers.)absolute_import)division)unicode_literalsN)Any) exceptions)parsersiDATAPROC_METASTOREDATAPLEXBIGQUERY)dpmsdataplexbigqueryserviceslakes)r r cd|dks|dkDr%tjddj||S)zKPython hook to validate that the port is between 1024 and 65535, inclusive.iiz--portz-Port ({0}) is not in the range [1025, 65535].rBadArgumentExceptionformat)ports 6lib/googlecloudsdk/command_lib/metastore/validators.py ValidatePortr's: D[D5L  ) )AHHN  +ct|dS)z+Python hook to validate the scaling factor.z--scaling-factorValidateScalingFactorFloatscaling_factors rValidateScalingFactorr0s #N4F GGrct|dS)z/Python hook to validate the min scaling factor.z--min-scaling-factorrrs rValidateMinScalingFactorr 5 #N4J KKrct|dS)z/Python hook to validate the max scaling factor.z--max-scaling-factorrrs rValidateMaxScalingFactorr#:r!rc2|dks|dkDr%tj|dj||dkr0|dzdzdk7r%tj|dj||dk\r-|dzdk7r%tj|d j||S) z(Validate the scaling factor float value.g?z4Scaling factor ({0}) is not in the range [0.1, 6.0]. rzUScaling factor less than 1.0 ({0}) should be a multiple of 0.1 (e.g. (0.1, 0.2, 0.3))g?zQScaling greater than 1.0 ({0}) should be a multiple of 1.0 (e.g. (1.0, 2.0, 3.0))r)r flag_names rrr?sc^a/  ) )>EE    nr1A5:  ) ) 228&2H  ~3q8  ) ) ))/)?  rcfd}|S)z-Validates the gcs uri is formatted correctly.ct|jds%tjdj||S)Nzgs://z'Expected URI {0} to start with `gs://`.) startswithrrr)gcs_uriarg_names rProcesszValidateGcsUri..ProcessZs<   g &  + + =DDWM  Nrr-r.s` rValidateGcsUrir1Ws .rctjd}|j|s&tjddj |||S)Nz^(.+)/(.+)@(.+)$--kerberos-principalz0Kerberos Principal {0} does not match ReGeX {1}.)recompilematchrrr)kerberos_principalpatterns rValidateKerberosPrincipalr9dsO JJ* +' ) *  ) ):AA    rcd|dks|dkDr%tjddj||S)z:Validates that the hour falls between 0 and 23, inclusive.rz --maintenance-window-hour-of-dayz$Hour of day ({0}) is not in [0, 23].r)hours rValidateHourOfDayr=ps: AX  ) )*.55d;  +rc|t|tkDr)tj|dj t|S)z\Validates that the string field is not longer than STRING_MAX_LENGTH, to avoid abuse issues.z7The string field can not be longer than {0} characters.)lenSTRING_MAX_LENGTHrrr)r-s rValidateStringFieldrAzs=]&&  ) )AHH    /rc~tjd}|j|stjdd|S)aAValidates the connection name of a CloudSQL instance, must be in the form '{project_id}:{region}:{instance_id}'. Args: connection_name: The CloudSQL instance connection name string. Returns: The connection name string. Raises: BadArgumentException: when the input string does not match the pattern. z^([^:]+:){2}[^:]+$z--instance-connection-namezRThe instance connection name should be in the format project_id:region:instance_idr4r5r6rr)connection_namer8s r&ValidateCloudSqlInstanceConnectionNamerEsB JJ, -'  '  ) )$ )  rcfd}|S)z{Validates the resource name of a compute network, must be in the form 'projects/{project_id}/global/networks/{network_id}'.ctjd}|j|stjd|S)Nz&^projects/[^/]+/global/networks/[^/]+$zdThe network resource name should be in the format projects//global/networks/rC resource_namer8r-s rr.z,ValidateNetworkResourceName..ProcesssBjjBCG == '  + +  @  rr/r0s` rValidateNetworkResourceNamerJ .rcfd}|S)zValidates the resource name of a compute subnetwork, must be in the form 'projects/{project_id}/regions/{region_id}/subnetworks/{subnetwork_id}'.ctjd}|j|stjd|S)Nz0^projects/[^/]+/regions/[^/]+/subnetworks/[^/]+$zzThe subnetwork resource name should be in the format projects/{project_id}/regions/{region_id}/subnetworks/{subnetwork_id}rCrHs rr.z/ValidateSubnetworkResourceName..ProcesssBjjLMG == '  + +  S  rr/r0s` rValidateSubnetworkResourceNamerNrKrctjd}|j|stjddt |dkDrtjdd|S)zValidates the hive database name. Args: db_name: the hive database name. Returns: the hive database name. Raises: BadArgumentException: when the database name doesn't conform to the pattern or is longer than 64 characters. z^[0-9a-zA-Z$_-]+$z--hive-database-namezhive database name must start with an alphanumeric character, and contain only the following characters: letters, numbers, dashes (-), and underscores (_).@z3hive database name must be less than 64 characters.)r4r5r6rrr?)db_namer8s rValidateHiveDatabaseNamerRsh JJ+ ,' w   ) )    \B  ) )=  .rcx tj||S#t$rtjddwxYw)zValidates the Cloud SQL IP address. Args: ip_address: the Cloud SQL IP address. Returns: the IP address. Raises: BadArgumentException: when the IP address is invalid. z --ip-addresszInvalid IP address.) ipaddress IPv4Address ValueErrorrr) ip_addresss rValidateCloudSqlIpAddressrXsC *%    ) ) s 9cd} ||stjdd |S#t$rtjddwxYw)zValidates the subnet IP range. Args: cidr: the CIDR range for the subnet. Returns: the CIDR range. Raises: BadArgumentException: when the CIDR range is invalid. cgd}dg}dg}gd}dg}dg}||z|z|z|z|z}tj|}|D](} |jtj| s(yy) zMChecks if a given CIDR block is contained within a list of valid CIDR ranges.)z 10.0.0.0/8z 172.16.0.0/12z192.168.0.0/16z 100.64.0.0/10z 192.0.0.0/24)z 192.0.2.0/24z198.51.100.0/24z203.0.113.0/24z192.88.99.0/24z 198.18.0.0/15TF)rT IPv4Network subnet_of) cidrrfc_1918_spacesrfc_6598_spacesrfc_6890_spacesrfc_5737_spacesrfc_7526_spacesrfc_2544_spacesvalid_cidr_ranges cidr_block valid_ranges rIsCidrWithinValidRangesz6ValidateSubnetIpRange..IsCidrWithinValidRangessHO&'O%&OKO'(O&'O             &&t,J(   i33K@ A) rz--subnet-ip-rangez]The subnet IP range is invalid, see https://cloud.google.com/vpc/docs/subnets.md#valid-rangeszInvalid CIDR address block.)rrrV)r]rgs rValidateSubnetIpRangerhsf2  "4 (  + +  G  ) +   ) )% s & Ac~tjd}|j|stjdd|S)a;Validates the Cloud Storage bucket name used for CDC during migration, should not start with 'gs://'. Args: bucket_name: the Cloud Storage bucket name. Returns: the Cloud Storage bucket name. Raises: BadArgumentException: when the Cloud Storage bucket name doesn't conform to the pattern. z^(?!gs://)([a-z0-9\._-]+)$z--bucketzInvalid bucket namerC) bucket_namer8s rValidateMigrationBucketNamerk#s@ JJ4 5' { #  ) )  rc~tjd}|j|stjdd|S)a3Validates the root path inside the Cloud Storage bucket used for CDC during migration, must start with a forward slash ('/') character. Args: root_path: the root path inside the Cloud Storage bucket. Returns: the root path. Raises: BadArgumentException: when the root path is invalid. z ^/([^\n\r]*)$z --root-pathzInvalid root pathrC) root_pathr8s rValidateMigrationRootPathrn9s@ JJ' (' y !  ) )  rct|||S)a^Validates that the mutual exclusive configurations of Dataproc Metastore service are not set at the same time. Args: req: A request with `service` field. Returns: A request without service mutex configuration conflicts. Raises: BadArgumentException: when mutual exclusive configurations of service are set at the same time. )ValidateServiceMutexConfigForV1 unused_ref unused_argsreqs rValidateServiceMutexConfigruNs )[# FFrc|jjru|jjjrU|jjjjr+t |jrt j dd|S)axValidates exclusively for V1 fields that the mutual exclusive configurations of Dataproc Metastore service are not set at the same time. Args: req: A request with `service` field. Returns: A request without service mutex configuration conflicts. Raises: BadArgumentException: when mutual exclusive configurations of service are set at the same time. r3zoKerberos configuration cannot be used in conjunction with --network-config-from-file or --consumer-subnetworks.)servicehiveMetastoreConfigkerberosConfig principal _IsNetworkConfigPresentInServicerrrqs rrprp]sl kk%% ++ ) ) 8 8 ++ ) ) 8 8 B B *3;; 7  ) ) A  *rc t|j}|jjjrd|vrt j dd|jjjrd|vrt j dd|S)aValidates that the cron_schedule and backup_location are set when the scheduled backup is enabled. Args: unused_ref: A resource ref to the parsed metastore service resource. args: The parsed args namespace from CLI. req: A request with `service` field. Returns: A request with service scheduled backups configurations required. Raises: BadArgumentException: when cron_schedule and backup_location are not set when the scheduled backup is enabled. z--scheduled-backup-cronzI--scheduled-backup-cron must be set when the scheduled backup is enabled.z--scheduled-backup-locationzM--scheduled-backup-location must be set when the scheduled backup is enabled.)setGetSpecifiedArgNamesrwscheduledBackupenabledrr)rrargsrtargs_sets rValidateScheduledBackupConfigsrws**, -( kk!!)) #8 3  ) )!    kk!!)) 'x 7  ) )%   *rcJ|jxr|jjSN) networkConfig consumers)rws rr{r{s    B7#8#8#B#BBrctt|j}d|vrd|vrtjdd|S)aValidate if users run update federation command with --clear-backends arg only. Args: unused_ref: A resource ref to the parsed Federation resource. args: The parsed args namespace from CLI. update_federation_req: The request for the API call. Returns: String request Raises: BadArgumentException: When users run update federation command with --clear-backends arg only. z--clear-backendsz--update-backendsz4--clear-backends must be used with --update-backends)r}r~rr)rrrupdate_federation_reqrs rValidateClearBackendsrsH**, -(8#(;8(K  ) )>  rc@|jrt|dk\Sy)NrF)isdigitint)strings r_IsZeroOrPositiveNumberrs ^^ v;!  rcNdjd|jDS)N|c3&K|] \}}| ywrr/).0keyvalues r z,_GetMetastoreTypeFromDict..s=*=:,... For example: 1=dpms:dpms1,2=dataplex:lake1 Returns: Backend metastore dict Raises: BadArgumentException: When the input backends is invalid or duplicated keys rzCannot be empty,=rrz8Invalid backends format or key of backend is less than 0r&:zDuplicated keys of backendszInvalid backends type)name metastoreType)rrsplitr?rMETASTORE_TYPE_DICTkeysr) backends backend_dictbackenddatarank_and_metastorerrrgenerated_namebackend_metastores_dicts r&ValidateBackendsAndReturnMetastoreDictrsl,   ) ),8I JJ NN3 'dC !#  + + 1  Q C "3 '  + +  D  q !E#kk#. "#q(  + + 1  l  + + 5 q!)<)A)A)CC  + + / 66MNN,-DQ-GH0L?@ rc|jjjD]R}|jjj |j j|j_T|S)zGenerate the long backend name of Dataproc Metastore federation requests. Args: job_ref: A resource ref to the parsed Federation resource. request: The request for the API call. Returns: Modified request for the API call. ) federationbackendMetastoresadditionalPropertiesrrrParent RelativeName)job_refrequestprops rParseBackendsIntoRequestr!sT  22GGdjjoo,,W^^-=-J-J-LMDJJOH .rc|jj |jjj|S|jjjD]}tj||S)aValidates that the kms keys are valid. Args: req: A request with `service` field. Returns: The unchaged request. Raises: InvalidResourceException: If the line is invalid. RequiredFieldOmittedException: If resource is underspecified. UnknownCollectionException: If no collection is provided or can be inferred. WrongResourceCollectionException: If the provided URL points into a collection other than the one specified. )rwencryptionConfigkmsKeysrParseCloudKmsKey)rrrsrtkms_keys rValidateKmsKeysr1s\" kk""* kk""**2 J--55g W%6 *r),__doc__ __future__rrrrTr4typingrgooglecloudsdk.callioper$googlecloudsdk.command_lib.metastorerr@rrrrr r#rr1r9r=rArErJrNrRrXrhrkrnrurprr{rrrrrrrr/rrrs '' .8  )3HH L L 0   *   :,2j,* G 4" JC0 >%P4n   r