B3dZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd l m Z dd l mZdd l mZdd l mZdd l mZddl mZddlmZddlmZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%d6d Z& d7d!Z'd"Z(d#Z)d$Z*d%Z+d&Z,d'Z-d(Z.d)Z/d*Z0d+Z1d,Z2d-Z3d.Z4d/Z5d0Z6d1Z7d2Z8d3Z9d4Z:d5Z;y)8z,General utilties for Cloud Domains commands.)absolute_import)division)unicode_literalsN)encoding) operations) registrations)flags) exceptions)log) properties) resources)yaml) console_io)fileszdomains.projects.locationsz%domains.projects.locations.operationsz(domains.projects.locations.registrationscjtjjjj dS)NT)required)r VALUEScoreprojectGet.lib/googlecloudsdk/command_lib/domains/util.pyr's$:$$))1155t5Drcfd}|S)NcLt|jjSN)ParseRegistrationnameSelfLink)resource api_versions rUriFuncz%RegistrationsUriFunc..UriFunc,s [(-- 8 A A CCrr)r#r$s` rRegistrationsUriFuncr%+sD .rc"tj|}|j|jjj |jjj fvr.tjdj|jy)NzIThe registration resource must be in state ACTIVE or SUSPENDED, not '{}'.) rGetMessagesModulestate RegistrationStateValueValuesEnumACTIVE SUSPENDEDr Errorformat)r# registrationmessagess rAssertRegistrationOperationalr11s~  , ,[ 9(007700::     f\//0 22 rc8|ytj|} tj||}g}tj|D]J}dj|dDcgc]}|jdzc}} ||dD cgc]} | | z c} z }L|j|r4t j dj|dj||S#t$r*}t j dj||d}~wwxYwcc}wcc} w) aParse a Yaml file. Args: path: Yaml file path. If path is None returns None. message_type: Message type to parse YAML into. error_message: Error message to print in case of parsing error. Returns: parsed message of type message_type. Nz{}: {}r.z{}. Problematic fields: '{}'z, ) r load_pathrPyValueToMessage Exceptionr r-r.UnrecognizedFieldIterjoinfieldsort) path message_type error_message raw_messageparsed_messageeunknown_fieldsmessageedge outer_messager;s rParseMessageFromYamlFilerG=s \ t$+>..|[IN .//?gGG'!*E*$TZZ#-*EFM'!*E*}u,*EEN@    *22=3799^3L N OO  >   8??=!< ==> FEs#CD  D D%%D  Dch|jd}t|d|d<dj|S)z(Normalizes domain name in resource name./)splitNormalizeDomainNamer:)domainpartss rNormalizeResourceNamerO^s0 ,,s %!%),%) %rc|stjd |jdj}|j j d}|S#t $r*}tjdj||d}~wwxYw)z.Normalizes domain name (including punycoding).zEmpty domain nameidnar4zInvalid domain name '{}': {}.N)r r-encodedecodelowerrstrip UnicodeErrorr.)rM normalizedrBs rrLrLes    . //Pv&--/J!!#**3/J  P   <CCFAN OOPs>A B "%BB cB|jdjdS)Nzutf-8rQ)rRrSrMs rPunycodeToUnicoderZqs w  & &v ..rcD|syd}tj||rd|vryy)NFz"^[a-z0-9-]+(\.[a-z0-9-]+)+\.{0,1}$z..T)rematch)rMpatterns rValidateDomainNamer_us'   1' '6 "dfn  rc@|duxrt|jSr)boolstrip)ss rValidateNonEmptyrds $ *4 ?*rcr|duxr2t|dk(xr"|jxr|jS)N)lenisalphaisupper)rcs rValidateRegionCoderks/ 4 JCGqL JRZZ\ JbjjlJrcJ|syd}ttj||S)NFz^[^@\s]+@[^@\s]+\.[^@\s]+$)rar\r])emailr^s r ValidateEmailrns#   )' bhhw& ''rcn|rtjj|tj|S)zPrompt for user input. Args: prompt_string: Message to print in the line with prompt. message: Optional message to print before prompt. Returns: User provided value. )r statusPrintrPromptResponse) prompt_stringrDs rPromptrts* JJW  " "= 11rc|rtjj| |tj||}|s|Stj |}||r|Stjj|\)aPrompt for user input and validate output. Args: prompt_string: Message to print in the line with prompt. validator: Validation function (str) -> bool. error_message: Message to print if provided value is not correct. message: Optional message to print before prompt. default: Optional default value. Returns: Valid user provided value or default if not None and user chose it. )rDdefault)r rprqrPromptWithDefaultrr)rs validatorr?rDrvanswers rPromptWithValidatorrzss" JJW ++2f ((7f m jj}% rcftjj}|jd||S)Ndomains)r REGISTRYCloneRegisterApiByName)r#registrys r GetRegistryrs+    % % '( Y 4 /rcRt|j|tddtSNglobal) projectsId locationsId)params collection)rParse_PROJECTREGISTRATIONS_COLLECTION)r#r/s rrrs2 [ ! ' ' !* ( ++rcRt|j|tddtSr)rrrOPERATIONS_COLLECTION)r# operations rParseOperationrs2 [ ! ' ' !' ( ((rc*||jddS)Nr4)findrYs rDomainNamespacers  C ! ""rcztj|}|ytj|j |Sr)rr'r TransferLockEnumMapperGetEnumForChoice)r#transfer_lock_stater0s rParseTransferLockStaters<  , ,[ 9(   % %h / @ @ rct|j}tjd|d|ddd}|syd}t |D]\}}||j |k(s|}tj ||d | }||S) aPrompts the user for the new enum_type value. Args: enum_mapper: Instance of the EnumMapper. enum_type: A string with enum type name to print. current_value: Current value of the enum. Returns: The new enum choice or None if the enum shouldn't be updated. z Your current z is: r4zDo you want to change itF)rvNrz Specify new )optionsrvrD)listchoicesrPromptContinue enumerateGetChoiceForEnum PromptChoice) enum_mapper enum_type current_valuerupdatecurrent_choiceienumindexs r PromptForEnumrs $$ %'  $ $i[m_A6  &  .7#ga {++M::n$  ! !YK( % rctj|}tj|}t |d|}|yt ||S)z-Prompts the user for new transfer lock state.zTransfer Lock stateN)rr'r rrr)r# transfer_lockr0rresults rPromptForTransferLockStatersF  , ,[ 9(,,X6+ &;] K& ^   V 44rcztj|}|ytj|j |Sr)rr'r RenewalMethodEnumMapperr)r#renewal_methodr0s rParseRenewalMethodrs<  , ,[ 9(   & &x 0 A A rctj|}tj|}t |d|}|yt ||S)z(Prompts the user for new renewal method.zpreferred Renewal MethodN)rr'r rrr)r#preferred_renewal_methodr0rrs rPromptForRenewalMethodrsL  , ,[ 9(--h7+ -/G & ^  K 00rcd}tjj|tjddt }|S)z(Prompts the user to enter the auth code.zaPlease provide the authorization code from the domain's current registrar to transfer the domain.zAuthorization code: z& Authorization code must not be empty.)promptr?validation_callable)r rprqrPromptPasswordrd)rD auth_codes rPromptForAuthCoder&s>1'**7'' $<*,) rc |y|}t|tstj|}dj |dt |j dddz |j ddS)Nz {}.{:02d} {}unitsnanosr逖 currencyCoder3) isinstancedictr MessageToDictr.intget)rdrs rTransformMoneyTyper3sgY " B     "B   r'{Cw0Be0L,M!vvnb9 ;;rctjd|}|r|j\}}n tdd|vr1|j d}t |d|t ||dzd|fSt |d|fS)z6Parses money string as tuple (units, cents, currency).z^(\d+|\d+\.\d{2})\s*([A-Z]{3})$z3Value could not be parsed as number + currency coder4Nr5r)r\r]groups ValueErrorrr)moneyr]numberrcrs r _ParseMoneyr=s ((5u =%  IFA J KKF] KK E vfu~ F519:$6 7 :: v;1 rc|sy t|\}}}|dk(rd}t j |}|j t||dz|S#t$rtjd|dwxYw)z"Parses money string as type Money.NzYearly price 'zF' is invalid. Please specify the amount followed by the currency code.$USDr)rrr)rrr r-rr'Moneyr)r# price_stringrcentscurrencyr0s rParseYearlyPricerKs  (6E5(_H  , ,[ 9(  Jeem(  DD    \N+5 5 s A#A3cX|jd|_|jd|_||k(S)Nr)r)xys r EqualPriceras+WW_AGWW_AG a-rcptjdjt|dddd}|r|Sy)z)Asks the user to accept the yearly price.zYearly price: {} z5Do you agree to pay this yearly price for your domainTF)rsthrow_if_unattended cancel_on_norvN)rrr.r)priceacks rPromptForYearlyPriceAckris?!!!!"4U";<K  #  L rc|syd|vd|vfS)zParses registration notices. Args: notices: list of notices (lowercase-strings). Returns: Pair (public privacy ack: bool, hsts ack: bool). )FFz#public-contact-data-acknowledgementzhsts-preloadedr)noticess rParseRegisterNoticesrws    .' 9;Kw;V VVrcjtjdjt||ddd}|S)Nzp{} is a secure namespace. You may purchase {} now but it will require an SSL certificate for website connection.TF)rrrv)rrr.r)rMrs rPromptForHSTSAckrs;!!<=CV 6 "F>,  # *rcRt||j}|r>tjj dj |j |Sd}tjj|}|j||j |j }|S)a Handles waiting for the operation and printing information about it. Args: api_version: Cloud Domains API version to call. response: Response from the API call asynchronous: If true, do not wait for the operation Returns: The last information about the operation. z Started '{}'zWaiting for '{}' to complete) rr r rprqr.NamerClientFromApiVersionWaitForOperation)r#response asynchronous operation_refrDoperations_clients rrrs!hmm<-JJ%,,]-?-?-ABC / /G"))88E 11w~~m&8&8&:;=H /rc2|sytj|S)zReads the text contents from the given path. Args: path: str, The file path to read. Raises: Error: If the file cannot be read. Returns: str, The text string read from the file. N)rReadFileContents)r=s rrrs      %%rc|jr |jS|jrt|jtSy)z-Returns the batch size for listing resources.N) page_sizelimitmin_MAX_LIST_BATCH_SIZE)argss rGetListBatchSizers/ ^^ >> zz tzz/ 00 rr)NN)<__doc__ __future__rrrr\apitools.base.pyrgooglecloudsdk.api_lib.domainsrr"googlecloudsdk.command_lib.domainsr googlecloudsdk.corer r r r rgooglecloudsdk.core.consolergooglecloudsdk.core.utilrLOCATIONS_COLLECTIONrrrrr%r1rGrOrLrZr_rdrkrnrtrzrrrrrrrrrrrrrrrrrrrrrrrrs3&' %584*#*)$2*3?E D 2B /+K( 2$!% $ &B +(# @5 1 ; D,  W ,&"r