5RddZddlmZddlmZddlmZddlmZddlZddlmZddl m Z dd l m Z dd l m Z dd l mZdd lmZddlZdd lmZdZdZdZdZdZdZdZGdde j6ZGdde j6ZGdde j6ZGdde j6ZGdde j6Z dZ!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,Z.d-Z/d.Z0d/Z1d0Z2d1Z3d2Z4d3Z5d4Z6d5Z7d6Z8d7Z9d8Z:d9Z;d:Zd=Z?d>Z@d?ZAd@ZBdAZCdCdBZDy)Dz>A library that is used to support Cloud Pub/Sub Lite commands.)absolute_import)division)print_function)unicode_literalsN)apis)util) exceptions)log) properties) console_io)urlparsez projects/z locations/z reservations/ztopics/zsubscriptions/ pubsublitev1ceZdZdZy)UnexpectedResourceFieldz,Error for having and unknown resource field.N__name__ __module__ __qualname____doc__2lib/googlecloudsdk/command_lib/pubsub/lite_util.pyrr+4rrceZdZdZy)InvalidPythonVersionz$Error for an invalid python version.Nrrrrrr/s,rrc"eZdZdZfdZxZS)NoGrpcInstalledz8Error that occurs when the grpc module is not installed.c,tt| dy)NzPlease ensure that the gRPC module is installed and the environment is correctly configured. Run `sudo pip3 install grpcio` and set the environment variable CLOUDSDK_PYTHON_SITEPACKAGES=1.)superr__init__)self __class__s rr!zNoGrpcInstalled.__init__6s /4) ?@r)rrrrr! __classcell__)r#s@rrr3s@@@rrceZdZdZy)InvalidSeekTargetz,Error for specifying an invalid seek target.Nrrrrr&r&=rrr&ceZdZdZy)InvalidResourcePathz>Error for specifying an invalid fully qualified resource path.Nrrrrr(r(AsFrr(c>tjttS)z*Returns the Pub/Sub Lite v1 client module.)rGetClientInstancePUBSUBLITE_API_NAMEPUBSUBLITE_API_VERSIONrrrPubsubLiteClientr-E   35K LLrc>tjttS)z,Returns the Pub/Sub Lite v1 messages module.)rGetMessagesModuler+r,rrrPubsubLiteMessagesr1Jr.rcXtjt|jdzS)zCConvert Duration object to total seconds for backend compatibility.s)six text_typeint total_seconds)durations rDurationToSecondsr9Os! s8112 3c 99rcL|jd}dj|ddS)aReturns the region from a location string. Args: location: A str of the form `-` or ``, such as `us-central1-a` or `us-central1`. Any other form will cause undefined behavior. Returns: The str region. Example: `us-central1`. -N)splitjoin)locationsplitss rDeriveRegionFromLocationrATs' >># & &!* rct|}|jjd}t|dkDr|ddz|dzSy)aDReturns the region from a endpoint string. Args: endpoint: A str of the form `https://base.url.com/`. Example `https://us-central-base.url.com/`, `https://us-central-autopush-base.url.com/`, or `https://base.url.com/`. Returns: The str region if it exists, otherwise None. r;r<rN)r netlocr=len)endpointparsed dash_splitss rDeriveRegionFromEndpointrIcsJ H & ##C(+ q>C +a. 00 rcP|jd}|d}|dz|zdz|dzS)zTReturns a new endpoint string with the defined `region` prefixed to the netlocation.z://rr;rC)r=)regionurl url_parts url_schemes rCreateRegionalEndpointrOvs7ii)|* e f $s *Yq\ 99rcJt|}|r|j|dzd}|S)zCReturns a new endpoint string stripped of the region if one exists.r;)rIreplace)rFrKs rRemoveRegionFromEndpointrS}s* #H -&  b1H /rcd}d}t|dr|j}d}||fSt|dr|j}d}||fSt|dr|j}d}||fSt d)aReturns a tuple of the resource and resource name from the `request`. Args: request: A Request object instance. Returns: A tuple of the resource string path and the resource name. Raises: UnexpectedResourceField: The `request` had a unsupported resource. rQparentname subscriptionz3The resource specified for this command is unknown!)hasattrrUrVrWr)requestresource resource_names rGetResourceInfor\s(- Wh~~HM =  w||HM =  w'##H"M =   "= ??rcddlm}t|jddk(r|jj |S|j j |S)Nr)typesr;)google.cloud.pubsubliter^rEr= CloudZoneparse CloudRegion) location_idr^s rLocationToZoneOrRegionresI+  3  A% ??  --    " "; //rc~||jtttzd}|jdd}|S)z,Returns the location from a resource string.N/r)indexLOCATIONS_RESOURCE_PATHrEr=)rZr?s rDeriveLocationFromResourcerjsC hnn%<=12345( ^^C  #( /rc~||jtttzd}|jdd}|S)z+Returns the project from a resource string.Nrgr)rhPROJECTS_RESOURCE_PATHrEr=)rZprojects rDeriveProjectFromResourcernsC X^^$:;/012 3' MM# q !' .rclt|\}}||jtd}t||||S)z;Returns an updated `request` with the resource path parsed.N)r\rindexrlsetattr)rYrZr[ new_resources r ParseResourcerss9+G4(M(//*@ABC, '=,/ .rct|\}}tt|}tjt t }t|}t||}tjjjj|y)zx$H I&  ) )*=*@B( &h /(,VX> **5599:KLrctjdd}tjdd}|j|}|jj |}|j S)zBReturns the Cloud project number associated with the `project_id`.cloudresourcemanagerr) projectId)rr0r*&CloudresourcemanagerProjectsGetRequestprojectsGet projectNumber) project_idcrm_message_moduleresource_managerreqrms rProjectIdToProjectNumberrse--.DdK++,BDIAA B #  % % ) )# .'   rc t|\}}t|}t|}t|||j |t j ||S)z;Returns an updated `request` with the Cloud project number.)r\rnrrqrRr4r5)rYrZr[rproject_numbers r OverrideProjectIdToProjectNumberrsR+G4(M(2*+J7. '=   :s}}^'D EG .rcL~~t|}t|}t||S)zCReturns an updated `request` with values for a valid Admin request.)rsrr{ resource_refargsrYs rUpdateAdminRequestrs*D ' "' ,W 5'W% .rc~~t|}|jjxjdz c_t ||S)zAUpdates a CommitCursorRequest by adding 1 to the provided offset.rC)rscommitCursorRequestcursoroffsetr{rs rUpdateCommitCursorRequestrs?D ' "' $$++q0+W% .rcZ|jyt|jjS)z0Returns whether the topic has a reservation set.F)reservationConfigboolthroughputReservationtopics r_HasReservationr s( $  e%%;; <x$H I&''==+ '+BF $k3 --!!7 .rc~~t|\}}dj|t|jj|j_|S)z?Returns an updated `request` with a resource path on the topic.z{}/{}{})r\rTOPICS_RESOURCE_PATHrWr)rrrYrZrys rAddSubscriptionTopicResourcer9sLD(+(A(//:N070D0D0J0J L' .rc\~d|vs |js|Stjddd|S)z:Prompts to confirm an update to a topic's partition count. partitionszWarning: The number of partitions in a topic can be increased but not decreased. Additionally message order is not guaranteed across a topic resize. See https://cloud.google.com/pubsub/lite/docs/topics#scaling_capacity for more detailsT)messagedefault cancel_on_no)rr PromptContinuers rConfirmPartitionsUpdaterEs:T__ N    .rcH~t|dr|jdk(|_|S)Nstarting_offsetend)rXr skipBacklogrs rUpdateSkipBacklogFieldrVs( T$%//58G .rc6|jxs |jS)z"Returns the raw location argument.)r?zone)rs rGetLocationValuer`s  #$))#rc~t|}t|vr|St|}|stdj ||S)zReturns the resource location (zone or region) extracted from arguments. Args: args: argparse.Namespace, the parsed commandline arguments. Raises: InvalidResourcePath: if the location component in a fully qualified path is invalid. zGThe location component in the specified location path is invalid: `{}`.)rrirjr(r)rr?parsed_locations r GetLocationresJd #(H, O.x8/  Q    rct|}|jts@|jxs2tj j jjSt|}|stdj||S)zReturns the project from either arguments or attributes. Args: args: argparse.Namespace, the parsed commandline arguments. Raises: InvalidResourcePath: if the project component in a fully qualified path is invalid. zFThe project component in the specified location path is invalid: `{}`.) r startswithrlrmr rvcore GetOrFailrnr(r)rr?parsed_projects r GetProjectr{svd #(   3 4 << E:,,1199CCEE,X6.  P    rc|jdk(r |jjjS|jjjS)z4Returns the DeliveryRequirement enum from arguments.zdeliver-after-stored)delivery_requirementDeliveryConfig"DeliveryRequirementValueValuesEnumDELIVER_AFTER_STOREDDELIVER_IMMEDIATELYrpsls rGetDeliveryRequirementrsA "88    @ @ U UU   > > R RRrc|jdk(r |jjjS|jjjS)z4Returns the export DesiredState enum from arguments.paused)export_desired_state ExportConfigDesiredStateValueValuesEnumPAUSEDACTIVErs rGetDesiredExportStaters@ (*    7 7 > >>   5 5 < <>++D,=,=>"@ ' AA   & &>>))$//:"< ' ==  {*  ( (S5P5P*D*DTT)KK   &  ( (S5P5P*D*DTT)KK  L NN < ==rcr|jrT|j}|jtsdjt|t|}|j ||_|jrK|j}|jts"djt|t|t|}||_ yy)z8Sets fully qualified resource paths for an ExportConfig. {}{}/{}{}rrN) export_pubsub_topicrrlrr PubSubConfig pubsubConfigexport_dead_letter_topicrideadLetterTopic)rrrmr? export_configrs rSetExportConfigResourcesrs   $ $E   2 3!!"8'"6?e!$!1!1!1!>M ""  ) )E   2 3&&'=w'>';UDe%*M! #rc|jyt||}|r |jjj}|j|}t ||||||S)z'Returns an ExportConfig from arguments.N desiredState)rrrrrr)rrrmr? requires_seek desired_staters rGetExportConfigrs` % 'c2-$$@@GGM"" ">-4gxG rct}t|}t|}tjt |}|j xs |j}|jdjt|t||j|j|jt||t!||||||j"}t%|t'||||s t)|||t+}|j,j/|} |rB|j1| j2t5||} |j,j7| |r|j"j8r|j:dk(rw|j=| j2d|j|j?|j>j@jB  } |j,jE| } | S) a,Issues a CreateSubscriptionRequest and potentially other requests. Args: resource_ref: resources.Resource, the resource reference for the resource being operated on. args: argparse.Namespace, the parsed commandline arguments. Returns: The created Pub/Sub Lite Subscription. r)deliveryRequirement)rdeliveryConfig exportConfig)rUrWsubscriptionId)rVseekSubscriptionRequestactivezexport_config.desired_stater)r)rV updateMaskrW)#r1rrr4r5rrr:PubsubliteAdminProjectsLocationsSubscriptionsCreateRequestrrlri SubscriptionrrrrrWr{rrr-&admin_projects_locations_subscriptionsCreate8PubsubliteAdminProjectsLocationsSubscriptionsSeekRequestrVrSeekrr9PubsubliteAdminProjectsLocationsSubscriptionsPatchRequestrrrPatch) rrrr?rrrcreate_requestclientresponse seek_requestupdate_requests r ExecuteCreateSubscriptionRequestr s #  ($*==!9*!EF.##6t-QQ  !7!8(D## ++"8s"C,E&tS.('46 $7 &&R (.^,|T>B <~>  &  : : A A(OO ]]N44MPOL 1166|D~22??DD]D]aiDiRR ]]0%%))s7G7G+F+Fvv*O&PSQN <<BBH /rc~|jj|St|\}}t|}t |}t }t |||||jj|S)aISets export resource paths for an UpdateSubscriptionRequest. Args: resource_ref: resources.Resource, the resource reference for the resource being operated on. args: argparse.Namespace, the parsed commandline arguments. request: An UpdateSubscriptionRequest. Returns: The UpdateSubscriptionRequest. )rWrr\rnrjr1r)rrrYrZryrmr?rs rAddExportResourcesr sl &&. N(+(A %h /' ' 1(#4gx"//<<> .rch~t}t|||_tjd|S)z.Sets the target for a SeekSubscriptionRequest.zThe seek operation will complete once subscribers react to the seek. If subscribers are offline, `pubsub lite-operations describe` can be used to check the operation status later.)r1rrr warning)rrrYrs r SetSeekTargetr.s6#$24$='!++23 .rcb~|jr|S|jr0dj|jt|j|_|j rW|jr|xjdz c_nd|_|xjdj|j z c_|S)z/Updates the filter for a ListOperationsRequest.ztarget={}/{}{}z AND rQzdone={})filterrWrrVSUBSCRIPTIONS_RESOURCE_PATHdoners rUpdateListOperationsFilterr<s [[ N &,,W\\-H-1->->@GN YY~~ nnngn NNi&&tyy11N .rctjjdks:tjjdk(r8tjjdkrt dj |yy)zVerifies that the python version is 3.6+. Args: cmd: The string command that requires python 3.6+. Raises: InvalidPythonVersion: if the python version is not 3.6+. r_znThe `{}` command requires python 3.6 or greater. Please update the python version to use this command.N)sys version_infomajorminorrr)cmds rRequirePython36rVse aC$4$4$:$:a$?$'$4$4$:$:Q$>  //5vc{ <<%?%@r)gcloud)Er __future__rrrrrgooglecloudsdk.api_lib.utilr!googlecloudsdk.command_lib.pubsubrgooglecloudsdk.corer r r googlecloudsdk.core.consoler r4six.moves.urllib.parser rlrirrrr+r,Errorrrrr&r(r-r1r9rArIrOrSr\rerjrnrsr{rrrrrrrrrrrrrrrrrrr r rrrrrrr$suE&%' ,2*#*2 +%&, ."5j..5-:++-@j&&@5 ((5G***GM M : &:!>0M"   =(* "$ ,,S=>4*$ 6r6 4