dZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd l m Z dd l m Z dd lmZdd lmZdd lmZddlmZddlmZddlZddlmZddlmZej4j6j8j:ZGddej>Z Gddej>Z!Gddej>Z"Gdde#Z$dZ%dZ&dZ'dOdZ(dOdZ)dZ*d Z+d!Z,d"Z-d#Z.d$Z/d%e j`jbd&fd'Z2d(Z3d)Z4e j`jbfd*Z5e j`jbfd+Z6e j`jbfd,Z7dPd-Z8dPd.Z9d/Z: dPd0Z;dPd1Zd4Z?d5Z@d6ZAd7ZBd8ZCd9ZDd:ZEd;ZF dQd<ZGd=ZHd>ZId?ZJd@ZKdAZLdBZMdCZNdDZOdEZPdFZQdGZRdHZSdIZTdJZUdKZVdLZWdMZXdNZYy)Rz;Utilities for parsing arguments to `gcloud tasks` commands.)absolute_import)division)unicode_literalsN)encoding)base) parser_errors)app) constants) exceptions) properties) resources) console_io) http_encoding)filter)mapceZdZdZy)NoFieldsSpecifiedErrorzAError for when calling an update method with no fields specified.N__name__ __module__ __qualname____doc__/lib/googlecloudsdk/command_lib/tasks/parsers.pyrr)sIrrceZdZdZy)FullTaskUnspecifiedErrorzCreates QueueUpdatableConfiguration from the given parameters.N maxAttemptsmaxRetryDuration) max_attemptsmax_retry_duration retryConfig maxDoublings minBackoff maxBackoffr%r& max_doublings min_backoff max_backoffmaxTasksDispatchedPerSecondmaxConcurrentTasks)max_tasks_dispatched_per_secondmax_concurrent_tasksrouting_overrideappEngineRoutingOverride uriOverride httpMethodheaderOverrideszoauthToken.serviceAccountEmailzoauthToken.scopezoidcToken.serviceAccountEmailzoidcToken.audiencehttp_uri_overridehttp_method_overridehttp_header_override)http_oauth_service_account_email_overridehttp_oauth_token_scope_override(http_oidc_service_account_email_override!http_oidc_token_audience_override rateLimitsappEngineHttpTarget httpTargetmaxDispatchesPerSecondmaxConcurrentDispatches maxBurstSizemax_dispatches_per_secondmax_concurrent_dispatchesmax_burst_sizelog_sampling_ratio samplingRatioappEngineHttpQueuestackdriverLoggingConfig)rGrH) retry_config rate_limitsapp_engine_routing_override http_targetstackdriver_logging_configretry_config_mask_prefixrate_limits_mask_prefix'app_engine_routing_override_mask_prefixhttp_target_mask_prefix&stackdriver_logging_config_mask_prefixr PULL_QUEUE PUSH_QUEUEr ReleaseTrackALPHABETA)cls queue_type release_trackconfigs rFromQueueTypeAndReleaseTrackz8QueueUpdatableConfiguration.FromQueueTypeAndReleaseTrack9s UFFF)+F&F(*F%&*F#%)F"59F2%)F"48F1Y)))' 2f)6f%h Mg y++ + $++11 1)"4+''  0M$8   :. * !0"/$  +8')5&9N6)5&` M_ D--22 2)"4+''  *B)B,   :. * !0"/$ " !/- )+8')5&9M6)5&8R5P MK*"4+''  *B)B   :. * !0"/$ " !/- )+8')5&9;6)5&8R5 Mrc*|j|jf|j|jf|j|j f|j |jf|j|jfg}|Dcgc] \}}|r||f c}}Scc}}w)zEReturns the initialized configs as a list of (config, prefix) tuples.) rOrTrPrUrQrVrRrWrSrX)selfall_configs_and_prefixesraprefixs r"_InitializedConfigsAndPrefixTuplesz>QueueUpdatableConfiguration._InitializedConfigsAndPrefixTupless   D99:  4778 ) ) 5 5 7  4778 ( ( 4 4 6  %= $< V $<  s;Bct}|jD]5}||}|rdj||||<n|||<|||t|<7|S)zEBuild a map from each arg and its clear_ counterpart to a mask field.z{}.{})dictkeysformat_EquivalentClearArg)rdrarffields_to_maskfield output_fields r_GetSingleConfigToMaskMappingz9QueueUpdatableConfiguration._GetSingleConfigToMaskMappingsaVNE]l  'v| Du ,u3A%3Hn(/0  rct}|jD]&\}}|j|j||(|S)zFBuilds mapping from config fields to corresponding update mask fields.)rirgupdaterp)rdconfig_to_maskrarfs rGetConfigToUpdateMaskMappingz8QueueUpdatableConfiguration.GetConfigToUpdateMaskMappingsCVN CCED>>vvNOF rcht|jjt|jjzt|jjzt|j jzt|j jzSN)listrOrjrPrQrRrS)rds r AllConfigsz&QueueUpdatableConfiguration.AllConfigss ""'') *T$2B2B2G2G2I-J J 11668 9 : !!&&( ) * 00557 8 9:rN) rrrr classmethodrr[GArbrgrprtrxrrrr!r!6s?3261B1B1E1EJJX  :rr!cptjjttj S)N collection)r REGISTRYParse_PROJECTr PROJECTS_COLLECTIONrrr ParseProjectrs.    ! !j.. " 00rcntjj|dtitj SN projectsIdparamsr})r r~rrr LOCATIONS_COLLECTION)locations r ParseLocationrs4    ! !H %// " 11rctjd|}|r0dj|jd|jdS|S)aParses a full queue reference and returns an abridged version. Args: queue_string: A full qualifying path for a queue which includes project and location, e.g. projects/PROJECT/locations/LOCATION/queues/QUEUE Returns: A shortened string for the full queue ref which has only the location and the queue (LOCATION/QUEUE). For example: 'projects/myproject/location/us-east1/queue/myqueue' => 'us-east1/myqueue' z@projects\/.*\/locations\/(?P.*)\/queues\/(?P.*)z{}/{}rqueue)rematchrkgroup) queue_stringrs rGetConsolePromptStringrsF ((I % >>%++j15;;w3G HH rc|syd} tjj|tj}|S#tj $rw|xst jt}t|}tjj||j|jdtj}Y|SwxYw)a Parses an id or uri for a queue. Args: queue: An id, self-link, or relative path of a queue resource. location: The location of the app associated with the active project. Returns: A queue resource reference, or None if passed-in queue is Falsy. Nr|)r locationsIdr) r r~rr QUEUES_COLLECTIONRequiredFieldOmittedExceptionr ResolveAppLocationrrrr)rr queue_ref app_location location_refs r ParseQueuers  ) 0""((4=4O4O)QI   0 00Es55lnEL .L""(( \%<%<&2&>&>@..)0I 0s/8BCCc|r|jnd} tjj|tj |S#tj $r tdwxYw)zParses an id or uri for a task.N)r}rzDMust specify either the fully qualified task path or the queue flag.)AsDictr r~rr TASKS_COLLECTIONrr)taskrrs r ParseTaskr,sq!*9   &P    # #D/8/I/I+1 $ 33  0 0P "N PPPs /AA%c6|jr |jSdS)zParses an id for a task.N)task_idargss r ParseTaskIdr8s/4/rc|sytjd|}|rB|jd|jd|jd|jdgSy)z9Parses and retrieves the segments of a full KMS key name.Nzjprojects\/(?P.*)\/locations\/(?P.*)\/keyRings\/(?P.*)\/cryptoKeys\/(?P.*)projectrkeyringkey)rrr) kms_key_namers rParseFullKmsKeyNamer=s`   ((s %  I J I E   rc|jr |jnd}d}t|j}||d}|j}nm|jra|jrU|jrIdj |j r |j n t ||j|j}t ||fS)zParses KMS key value.Nzbprojects/{kms_project_id}/locations/{location_id}/keyRings/{kms_keyring}/cryptoKeys/{kms_key_name})kms_project_id location_id kms_keyringr)rrrrrk kms_projectr)rrfull_kms_key_name parse_results rParseKmsUpdateArgsrPs!% D+$T%6%67, q/K))     --|DD+/+;+;t''$$&& D & 33rcR|jr |jnd}t}||fS)zParses KMS describe args.Nrr)rr project_ids rParseKmsDescribeArgsrks%!% D+z* [  rcN|jr |jnd}t|fS)zParses KMS clear args.Nr)rrs rParseKmsClearArgsrss !% D+ [  rc|j}|d=tjjd|tj }|S)NqueuesIdr)rr r~rr r)rrrs rExtractLocationRefFromQueueRefrzsG    & Z##)) 6i&D&D*F, rFTc |tjjk(rZt|||}|r t |||nd}|j t ||||dt||||t||||||S|tjjk(re|r t |||nd}|j t ||||dt||||t||||t||||t||||S|r t |||nd}|j t ||||dt||||t||||t||||S)zParses queue level args.NT)is_alpha)r'r@ pullTargetrArBF)r'r@rMrLrBtype)r'r@rMr4rB)rr[r\_ParseAppEngineHttpTargetArgs_ParseHttpTargetArgsQueue_ParseRetryConfigArgs_ParseAlphaRateLimitsArgs_ParsePullTargetArgsr]_ParseRateLimitsArgs"_ParseStackdriverLoggingConfigArgs_ParseAppEngineHttpQueueArgs_ParseQueueType"_ParseAppEngineRoutingOverrideArgs)rr_messages is_updater` http_queueapp_engine_http_targetrRs rParseCreateOrUpdateQueueArgsrsd''---: j(=GT:x8D >>) *h D - *h  (j(IN2   ))...>) *h E (j(IN!C *h " 8 *h  T:x C    =GT:x8D >>) *h E (j(IN!C *h " "D *h"    rc|j|jjnd}|j|jjnd}|j|jjnd}|j6|jj |jjj nd}|j6|jj |jjj nd}|j6|jj |jjj nd}|j6|jj |jjjnd}|||||||dS)zHReturns a pair of each http target attribute and its value in the queue.N)r9r:r;http_oauth_email_overridehttp_oauth_scope_overridehttp_oidc_email_overridehttp_oidc_audience_override) rBr5r6r7 oauthTokenserviceAccountEmailscope oidcTokenaudience) queue_configr9r:r;rrrrs rGetHttpTargetArgsrs  ,))   ,((   ,--   ! ! -%%00<((<<    ! ! -%%00<((..    ! ! -%%//;'';;    ! ! -%%//;''00  -22#<#<":%@ rc0d} |jjj}|r|Sd} |jjj}|sydj |}||vr|j |ddSdS#t$rYawxYw#t$rYIwxYw)aExtracts any target (service) if it exists in the appEngineRouting field. Args: job: An instance of job fetched from the backend. project: The base name of the project. Returns: The target if it exists in the URL, or if it is present in the service attribute of the appEngineRouting field, returns None otherwise. Some examples are: 'alpha.some_project.uk.r.appspot.com' => 'alpha' 'some_project.uk.r.appspot.com' => None Nz.{}.rr)rAappEngineRoutingserviceAttributeErrorhostrksplit)jobrtargethost_url delimiters r!ExtractTargetFromAppEngineHostUrlrs &  $ $ 5 5 = =F M( &&77<>@,.<=nsc"n+= T6tyyI,?,??v K !!>s B#cnfd}g}|r|Dcgc] }t|}}t|||zScc}w)z:Returns the list of known arguments in the specified list.cgd} jS#tj$rdvstfd|DrYywxYw)aWrapper function for Namespace.IsSpecified function. We need this function to be support being able to modify certain queue attributes internally using `gcloud app deploy queue.yaml` without exposing the same functionality via `gcloud tasks queues create/update`. Args: arg: The argument we are trying to check if specified. Returns: True if the argument was specified at CLI invocation, False otherwise. r8)rIclear_max_burst_sizec3&K|]}|v ywrvr).0flagrs r z>_SpecifiedArgs.._IsSpecifiedWrapper..psB"1$$#+/sF) IsSpecifiedrUnknownDestinationExceptionany)rhttp_queue_argsspecified_args_objects` r_IsSpecifiedWrapperz+_SpecifiedArgs.._IsSpecifiedWrapperVs\O " . .s 33  4 4 : :cB"1B?  s+AA)rlr)r args_listrrclear_args_listas` rrrSsF @/7@Ay!*1-yOA #Y%@ AABs2c.tt|||S)z@Returns whether there are known arguments in the specified list.)rr)rrrs rrr|s ^19j I JJrc$dj|S)Nzclear_{}rk)rs rrlrls   3 rc|tjk(r;t|ddg|r+|j|j}t ||||S|tj k(r]t|gd|rL|j|j|j|j|j}t ||||Syy)z6Parses the attributes of 'args' for Queue.retryConfig.r%r&r)r$r+)r$r(r)r*N) r rYr RetryConfigr&_AddMaxAttemptsFieldsFromArgsrZr,r-r.)rr_rrrrOs rrrsI(((~/CD#,.''00(2L!$ h? I(((N#,.''00''D4D4D##(%L"$ h? .)rc|jdr0|j|rd|_yd|_y|j|_yy)Nr%T)rr%unlimitedAttemptsr#)r config_objectrs rrrsB n%  *. '$& !"&"3"3m&rc|tjk(r8t|ddg|r'|j|j|j Syy)5Parses the attributes of 'args' for Queue.rateLimits.r1r2r)r/r0N)r rZr RateLimitsr1r2rr_rrs rrrs^I(((@57#,.   $($H$H44  66.)rc|tjk(rSt|gd|rBt|dr |jnd}|j |j |j|Syy)r"rFrrIN)rCrDrE)r rZrrrIr#rGrH)rr_rrrIs rrrsxI(((    't-=>D   #== $ > >#  %% )rc|tjk7r,t|dg|r|j|jSyy)zCParses the attributes of 'args' for Queue.stackdriverLoggingConfig.rJr)rKN)r rYrStackdriverLoggingConfigrJr$s rrrsNI(((34K  , ,-- - //L)rcP|tjk(r|s|jSyy)z5Parses the attributes of 'args' for Queue.pullTarget.N)r rY PullTarget) unused_argsr_rrs rrrs)9'''     1:'rct|dr|jtjk(s|tjk(r"|s |jj j S|jj jS)z/Parses the attributes of 'args' for Queue.type.r)rrr rYrTypeValueValuesEnumPULLPUSHr$s rrrs[tVi.B.B!BY)))) >> - - 2 22  + + 0 00rcn|tjk(r"t|||}|y|j|Sy)z>Parses the attributes of 'args' for Queue.appEngineHttpTarget.Nr4)r rZrAppEngineHttpTargetrr_rr3s rrrsK9'''9 j($   ' '!1 ( 33 (rcP|tjk(rt||}|jdr3|jj |j jnd}t||}t||}||||y|j |t|||||Sy)z5Parses the attributes of 'args' for Queue.HttpTarget.r:N)r5r7r6rr) r rZ_ParseHttpRoutingOverrideArgsr HttpTargetHttpMethodValueValuesEnumr:upper_ParseHttpTargetOAuthArgs_ParseHttpTargetOidcArgs_ParseHttpTargetHeaderArg)rr_r uri_override http_method oauth_token oidc_tokens rrrs9'''0x@L   2 3 55  % % + + - /9= ,D(;K)$9J            1$A  '(rch|tjk(rt|||}|j|Sy)z=Parses the attributes of 'args' for Queue.appEngineHttpQueue.r0N)r rZrAppEngineHttpQueuer2s rrrsD9'''9 j(  & &!1 '  (rc|tjk(r.|jdr|jdi|jSyy)z5Parses the attributes of 'args' for AppEngineRouting.r3Nr)r rZrAppEngineRoutingr3)rr_rs rrrsC9''' *+ &X & & ?)>)> ?? (rcT|jdrtdd|i|jSy)z1Parses the attributes of 'args' for HTTP Routing.r9rNr)r_ParseUriOverrider9rrs rr4r4s- )*  Ih I$2H2H II rc B|r)|jj|jnd}|r t|nd}|r)|jj |jnd}|j||||j ||j ||S)z1Parses the attributes of 'args' for URI Override.N)path) queryParams)schemerport pathOverride queryOverrideuriOverrideEnforceMode) UriOverrideSchemeValueValuesEnumr7int%UriOverrideEnforceModeValueValuesEnum PathOverride QueryOverride)rrIrrJrGquerymodeuri_override_enforce_modes rrDrDs 00@ T$ @@N      ((d(3**u*=6  88rcv|tjk(r&|jt||jSy)N)payloadtag)r r PullMessage_ParsePayloadArgsrY)rrrs rrr7s5)%%%   (9$(?TXX  NN&rc |tjk(r|jr|jdi|jnd}|j dr3|j j |jjnd}|j ||t||jt||j jSy)>Parses the attributes of 'args' for Task.appEngineHttpRequest.Nmethod)rr6rX relativeUrlheadersr) r APP_ENGINE_TASKroutingrBrAppEngineHttpRequestr6r^r7r[url_ParseHeaderArg HeadersValuerrrrbr<s rrr<s)+++59\\!!!1DLL1t !% 0 0 :00JJ @D  ( ( [!$'TXX ( = = J JL ) MM ,rc|jdr"tj|jd}n|jdr |j}nyt j |S)N payload_fileFbinarypayload_content)rrReadFromFileOrStdinrirlrEncode)rrXs rr[r[JsV n%,,T->->uMG )*""G    g &&rc |tjk(r|jr|jdi|jnd}|j dr3|j j |jjnd}|j ||t||jt||j jSy)r]Nr^)rr6body relativeUrir`r) r rarbrBrrcr6r^r7_ParseBodyArgs relative_urirerfrgs rrrTs)+++59\\!!!1DLL1t !% 0 0 :00JJ @D  ( ( [ D !t/@/@ ( = = J JL ) MM ,rc n|tjk(r|jdr3|jj |j j nd}|jt||jj|t||jt||t||Sy)z5Parses the attributes of 'args' for Task.httpRequest.r^N)r`r6rprdrr) r HTTP_TASKr HttpRequestr6r^r7rerfrrrd_ParseOAuthArgs_ParseOidcArgs)rrrr<s rrrbs)%%% $ 0 0 :''AA @D   h&:&:&G&GH^D%9txx"42 x0  22&rc|jdr"tj|jd}n|jdr |j}nyt j |S)N body_fileFrj body_content)rrrmrzr{rrn)rrps rrrrrnsS k"  ) )$.. GD '   D    d ##rct|jdr'|j|j|jSy)Noauth_service_account_emailrr)r OAuthTokenr}oauth_token_scoperEs rrwrwxs@ 34    <<$$  && rct|jdr'|j|j|jSy)Noidc_service_account_emailrr)r OidcTokenroidc_token_audiencerEs rrxrxs@ 23    ;;))  ++ rct|jdr'|j|j|jSy)Nr<r~)rrr<r=rEs rr8r8s@ AB    JJ22  44 rct|jdr'|j|j|jSy)Nr>r)rrr>r?rEs rr9r9s@ @A    II77  99 rc|jrCtt|jDcic]\}}|| }}}tj||Sycc}}wrv)headerr_SplitHeaderArgValuerDictToAdditionalPropertyMessage)r headers_valuekv headers_dicts rreresN [[%()=t{{%KL%KTQAqD%KLL  3 3L- PPLs AcP|jdd\}}||jfS)N:r)rlstrip)header_arg_valuervalues rrrs)%%c1-*#u elln rcxg}|jdrtt|jDcic]\}}|| }}}t |j }|D]U\}}|j |j|j|j} |j| W|Scc}}w)zCConverts header values into a list of headers and returns the list.r;)rr)r) rrrr;ritemsHeaderOverrideHeaderencoder) rrmap_rrrrrrheader_overrides rr:r:s $ ,-143L3LMMA1M <%%' (E U //SZZ\H0Jo kk/" +s B6c$dj|S)Nz{}sr)lease_durations rFormatLeaseDurationrs n %%rcv|jry|jdrdj|jSy)Nztag_function=oldest_tag()rYztag="{}") oldest_tagrrkrYrs rParseTasksLeaseFilterFlagsrs4 __ & e   TXX &&rctjj|jdtit j jSr)r r~rnamerr rSelfLink)rs r QueuesUriFuncrsA    ! ! jjH %,, " ./7hj9rctjj|jdtit j jSr)r r~rrrr rrrs r TasksUriFuncrsA    ! ! iiH %++ " -.6XZ8rctjj|jdtit j jSr)r r~rrrr rrrs rLocationsUriFuncrsA    ! ! iiH %// " 12:rsB&' %(106**)22     ! ! ) ) 3 3JZ--JEz//E)9)9 v:&v:r0 1*6 P0 &46!!##&& ?D;|"LL'+&7&7&:&:F$,0+<+<+?+?M*.):):)=)= " &BRK  $). 46%"/! 13:"  80O M' M 2$Q  "&'98