R2dZddlZddlmZddlmZddlmZddlm Z ddl mZddl m Z ddl m Z e jd d Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$d Z%d!e&fd"Z'd!e&fd#Z(d(d$Z)d%Z*d&Z+d'Z,y))zDA library used to support Managed Service for Apache Kafka commands.N)encoding) exceptions)core)apis)log) properties managedkafkav190929192z /contexts/z /subjects/z/mode/z/config/c<|dkrtjdd|S)zValidate CPU >= 3.z--cpuzCPU must be at least 3)rBadArgumentException)cpus 4lib/googlecloudsdk/command_lib/managed_kafka/util.py ValidateCPUr&s"1W  ) )'3K LL *c|js|S|jsi|_d}t|j||_t |||S)zPrepare the update request with the information from the subnet flag. Args: _: resource parameter required but unused variable. args: list of flags. request: the payload to return. Returns: The updated request with the subnet. %gcpConfig.accessConfig.networkConfigs)subnetsclusterAppendUpdateMask updateMaskMapSubnetsToNetworkConfig)_argsrequestsubnet_update_masks rPrepareUpdateWithSubnetsr-sK  N GO>'(:(:G N    N GO:'(:(:'+N) 00AA(K+'0k9+$>+ ! X{#(:(:-6Kq *DK&' rctjj|j}dt j |t jji|_ d|_ |S)zLoad the topics JSON from the argument to the request. Args: _: resource parameter required but unused variable. args: list of flags. request: the payload to return. Returns: The updated request with topics. topics) ryamlload topics_filerr2r3 ConsumerGroup TopicsValue consumerGroupr)rrrr^s r UpdateTopicsresV 99>>$** +&&& (((44'  ' .rc~|js|js|S|jjsi|j_dgi|jj_|jsdgi}n|jggd}|j sg|_t |j |d<|jsg|_t |j|d<|jjjjjtj|tjt|tjrrtj dd|j"|_tj dd|j"|_tj dd|j"|_|S) a Maps subnets and DNS names to the network config API field. Args: _: resource parameter required but unused variable. args: list of flags. request: the payload to return. Returns: The updated request with networkConfig in the JSON format. r-dns_domain_names)primary_subnetadditional_subnetsrgriz7gcpConfig\.accessConfig\.networkConfigs\.dnsDomainNamesrz6gcpConfig\.accessConfig\.networkConfigs\.primarySubnetz:gcpConfig\.accessConfig\.networkConfigs\.additionalSubnets)rhdns_nameconnectClusterr/r0additional_subnetlistr-r1rr2r3ConnectNetworkConfig isinstance8ManagedkafkaProjectsLocationsConnectClustersPatchRequestresubr)rrrr5s rMapConnectParamsToNetworkConfigrss   T]] N    ) )')G$3CR2H'""/  ("-N-- N   D)-d.D.D)E.%& DM'+DMM':.#$ ""//>>EE^X-J-JK GG B/G A/G E/G .rcz|jrftjj|j}t j |t jj|j_ |jxs<tjjjjj}|j}t!j"d|jsA|j$xs|j&j)dd}d|d|d|j}||j_|S)zLoad the config JSON from the argument to the request and build the kafka cluster resource path. Args: _: resource parameter required but unused variable. args: list of flags. request: the payload to return. Returns: z$projects/.+/locations/.+/clusters/.+rGrz projects/z /locations/z /clusters/) config_filerr_r`rr2r3ConnectCluster ConfigValuerkconfigrPrrOrQ kafka_clusterrqmatchrYconnect_clusterrN kafkaCluster)rrrrxrPkafka_cluster_pathrYs rPrepareConnectClusterCreater~5s  YY^^D,, -F$,$:$:''33%G! LL EDOO2277??CCE')) 94;M;M N}}B 4 4 : :3 ? BH G9KzDG )3G )'*<*./ "" D ) ) 7 783>./*2*@*@   +G' .rcx|jsi|_|jrtjj |j}t j |tjj|j_ t|jd|_ i}|jr%t|jjdz|d<|j r%t|j jdz|d<|r3t j |tj"|j_|S)zLoad the config JSON from the argument to the request, and parse out the task restart policy. Args: _: resource parameter required but unused variable. args: list of flags. request: the payload to return. Returns: rrrr)rrurr_r`rr2r3rrrrrrrrrrrrs r-ConnectorUpdateReadConfigAndTaskRestartPolicyrs    G  YY^^D,, -F ( 6 6""//!G*'*<*./ "" D ) ) 7 783>./*2*@*@   +G' .rc|jr|jjd}|jdg}|jj D]\}}|j d|dt dj|dj||_|jrt |jd|_|S)zUnnest the configs dictionary to the update mask. Args: _: resource parameter required but unused variable. args: list of flags. request: the payload to return. Returns: The new update mask with the configs. r'rz configs["z"]) rrrNremoveitemsr1rjoinr)rrrr* configs_listkeys r PatchConfigsrs \\$$**3/Ky!L,,$$&QIcU"-.') sxx 5G )'*<* >>  x""66???  x""66===  rc|tjjjk(ry|tjjjk(ry|tjjj k(ry|tjjj k(ry|tjjjk(ry|tjjjk(ryy)zParse the compatibility enum to a string. Args: compatibility: The compatibility enum of the schema registry or subject. Returns: The compatibility string. BACKWARDBACKWARD_TRANSITIVEFORWARDFORWARD_TRANSITIVEFULLFULL_TRANSITIVEr) r3 SchemaConfigCompatibilityValueValuesEnumrrrrrr) compatibilitys rParseCompatibilityrs    ; ; D DE     ; ; O OP !    ; ; C CD     ; ; N NO --JJOOO     ; ; K KL  rcr|xs4tjjjj dS)NT)required)rrOrrPrQ) project_ids r ParseProjectrs-  Hz((--559949HHrcztjdd}tjdd}|}|t|}|ritj j d|z|j|} |jj|t j|dy tj j d|z|j%|} |j&j|t j|dy #tj$r}tj|t}tj j |j d|j vr&tj|d j#| Yd }~y d }~wwxYw#tj$r}tj|t}tj j |j d|j vr&tj|d j#| Yd }~y d }~wwxYw) aACalled when the user runs gcloud managed-kafka schema-registries subject delete ... Args: subject: The subject of the attribute to delete. subject_run_resource: The subject resource path. context: The context of the schema registry if provided. Returns: The updated subject with its mode deleted. r r zDeleting subject mode for [%s].rArzmode. It is now unset.detailsResource not foundSubject {} not found. error_formatN)rGetMessagesModuleGetClientInstanceSUBJECTS_MODE_RESOURCE_PATHrstatusPrintFManagedkafkaProjectsLocationsSchemaRegistriesContextsModeDeleteRequest1projects_locations_schemaRegistries_contexts_modeDeleteUpdatedResourceapitools_exceptionsHttpNotFoundErrorr HttpExceptionHTTP_ERROR_FORMATmessageformat>ManagedkafkaProjectsLocationsSchemaRegistriesModeDeleteRequest(projects_locations_schemaRegistries_mode) subjectsubject_run_resourcecontextrclientschema_registry_resourcere api_errors rDeleteSubjectModers   " ">4 8'  ! !.$ 7&1 ""#>"?yI  JJ6@A\\ %]G   >>EEF '+CDJJ6@ANN) O    55<4 8'  ! !.$ 7& $ %&C%DWI N$ JJ87BC^^ _G   @@GGH '+EFJJ87BCVV WG   77>>w>O '+EF  0 0 **1.?@i jjy(() !2!2 2&& 3::7C   3   0 0 **1.?@i jjy(() !2!2 2&& 3::7C   3 s203D 3F$ F! A7FF!$H87A7H33H8z;ResponseError: code={status_code}, message={status_message})N)-__doc__rqapitools.base.pyrrrgooglecloudsdkrgooglecloudsdk.api_lib.utilrgooglecloudsdk.calliopegooglecloudsdk.corerrrr3rVrXCONTEXTS_RESOURCE_PATHSUBJECTS_RESOURCE_PATHrrrrr%rrr#rCrEr@rersr~rrrrrrrrrrrr)rrrsK %>,.#* "4 ! !.$ 7   %%& * 54!1H C.:  :6*9x6@!H"J4s(%%PI7 t. dBr