a(dZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd l m Zdd lmZdd lmZdd lmZddlZd ZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"d.dZ#dZ$dZ%dZ&dZ'd Z(d!Z)d"Z*d#Z+Gd$d%ejXZ-d&Z.Gd'd(ejXZ/d)Z0Gd*d+ejXZ1d,Z2d-Z3y)/z"hooks for billing budgets surface.)absolute_import)division)unicode_literalsN) extra_types)apis)messages)base) exceptions)yaml)timesc@tjdt|SNbillingbudgets)rGetMessagesModule GetApiVersionargss 6lib/googlecloudsdk/command_lib/billingbudgets/hooks.pyrr"s   0-2E FFc.tjd|Sr)rr)versions rGetMessagesModuleForVersionr&s   0' ::rct|dr6|jjtjjk(ryy)Ncalliope_commandv1v1beta1)hasattrr ReleaseTrack calliope_baseGArs rrr*s: T%&4+@+@+M+M,!!$$,%  rcft|}|dk(r |jS|jjSNr)r!googleCloudBillingBudgetsV1Budget3googleCloudBillingBudgetsV1beta1CreateBudgetRequestbudgetrreqrs rGetVersionedCreateBillingBudgetr(21 $ ' _  0 00  B B I IIrcft|}|dk(r |jS|jjSr")rr#3googleCloudBillingBudgetsV1beta1UpdateBudgetRequestr%r&s rGetVersionedUpdateBillingBudgetr,:r)rc0td}t||S)z:Convert the input to Money Type for v1beta1 Create method.rr ParseMoneymoneyrs rCreateParseToMoneyTypeV1Beta1r2B ( 3( E8 $$rc0td}t||S)z5Convert the input to Money Type for v1 Create method.rr.r0s rCreateParseToMoneyTypeV1r5H ( .( E8 $$rc0td}t||S)z:Convert the input to Money Type for v1beta1 Update method.rr.r0s rUpdateParseToMoneyTypeV1Beta1r8Nr3rc0td}t||S)z5Convert the input to Money Type for v1 Update method.rr.r0s rUpdateParseToMoneyTypeV1r:Tr6rc0td}t||S)z9Convert the input to Date Type for v1beta1 Create method.rr ParseDatedaters rCreateParseToDateTypeV1Beta1r@Z ( 3( 4 ""rc0td}t||S)z4Convert the input to Date Type for v1 Create method.rr<r>s rCreateParseToDateTypeV1rC` ( .( 4 ""rc0td}t||S)z9Convert the input to Date Type for v1beta1 Update method.rr<r>s rUpdateParseToDateTypeV1Beta1rFfrArc0td}t||S)z4Convert the input to Date Type for v1 Update method.rr<r>s rUpdateParseToDateTypeV1rHlrDrcLt|d}tjd|ddr|dd}|rtjd|dddntjd|}|drt |dnd}t |dkDrt |d}nd}|j |||S) z)Validate input and convert to Money Type.z [A-Za-z]{3}Nz\.r)unitsnanos currencyCode)CheckMoneyRegexrematchsplitintlenGoogleTypeMoney)r1r currency_code money_arrayrMrNs rr/r/rs%-XXneBCj)"#JM(5bhhueCRj!$288 <"-Q#k!n Q%  A E E  ! ! ] " < >?O$88 _ -  4 46f ++66:KK688 _ -  9 9;f;;FFJ[[F;A:O:O$D$'))7 *rcD|}|D]}||vs|j||S)N)append) old_rules rules_to_addall_threshold_rulesrules rroros0!d 9  & rc@tdj}|r|Sy)Nr)r0GoogleCloudBillingBudgetsV1beta1LastPeriodAmountuse_last_period_amountrs rLastPeriodAmountV1Beta1rs( (AA  :rc@tdj}|r|Syr")r+GoogleCloudBillingBudgetsV1LastPeriodAmountrs rLastPeriodAmountV1rs( (  7 7  :rct~|jdr%d|jjj_|SNall_updates_rule_pubsub_topic1.0)rnr$r%allUpdatesRule schemaVersionrvrr's rCreateAllUpdatesRuler3  56bgC;;BBQQ_ *rc`~|jdrd|jj_|SNnotifications_rule_pubsub_topicrrnr#notificationsRulerrs rCreateNotificationsRuler-  78LQC));;I *rct~|jdr%d|jjj_|Sr)rnr+r%rrrs rUpdateAllUpdatesRulerrrc`~|jdrd|jj_|Srrrs rUpdateNotificationsRulerrrceZdZdZy)InvalidBudgetCreditTreatmentzEError to raise when credit treatment doesn't match the credit filter.N__name__ __module__ __qualname____doc__rrrrsMrrc|jdxr|jdk(}|jdxr |j}|r |s td|s |r td|S)z:Validates credit treatment matches credit types in filter.credit_types_treatmentzinclude-specified-creditsfilter_credit_typesz^'--filter-credit-types' is required when '--credit-types-treatment=include-specified-credits'.zg'--credit-types-treatment' must be 'include-specified-credits' if '--filter-credit-types' is specified.)rnrrr) unused_refrr'budget_tracks_creditspopulated_credits_filters rValidateCreditTreatmentrs**+CDB !!%@@!--:!%!9!9#; & @ AA #; & 0 11 *rceZdZdZy)InvalidBudgetAmountInputz4Error to raise when user input does not match regex.Nrrrrrrs<rrcptjd}tj||s tdy)Nz^[0-9]*.?[0-9]+([a-zA-Z]{3})?$zoThe input is not valid for --budget-amount. It must be an int or float with an optional 3-letter currency code.)rQcompilerRr) input_stringaccepted_regexs rrPrPs9::?@. ., / " N OO 0rceZdZdZy)InvalidLabelInputz2Error to raise when user label input is not valid.Nrrrrrr s:rrc~|jdrMt|}t||}|j|gt ||j j _|S)z!Adds labels to an Update request. filter_labels)additionalProperties)rnr CreateLabels LabelsValuer, budgetFilterlabelsr)rvrr'radditional_propertys rUpdateParseLabelsrsh  o& &H&tX69A8L8L"5!69M98$   \&&!5 *rcVtj|j}t|dkDr t dt |j d}t||dkDr t d||d}|jj|tj|gS)z"Parses and validates labels input.rLzLThe input is not valid for `--filter-labels`. It must be one key/value pair.rzPThe input is not valid for `--filter-labels`. It must be one key with one value.) string_value)keyvalue) r rprrUrlistkeysrAdditionalPropertyr JsonValue)rr labels_dictrrs rrrs $,,-+ = >> [    ##S Q A BB c 1 %    0 0 k++?@ 1 BBr)z%Y-%m-%d)4r __future__rrrrQapitools.base.pyrgooglecloudsdk.api_lib.utilrrrqgooglecloudsdk.callioper rgooglecloudsdk.corer r googlecloudsdk.core.utilr rjrrrr(r,r2r5r8r:r@rCrFrHr/r=rrorrrrrrErrorrrrrPrrrrrrrs)&' (,A9*$* G;JJ% % % % # # # # <$N( V    :#3#3  "z// O ((   Br