dZddlmZddlmZddlmZddlmZddlmZddl mZddl m Z dZ d Z hd Zdd Z dd Z ddZdZdZdZdZdZy )z1Service Consumer Management API helper functions.)absolute_import)division)unicode_literals) exceptions) list_pager)apiszservices/%s/%sz%s/producerOverrides/%s>folders/ projects/organizations/Nct|t}|j}|jt||fz}t j |j||d|dS)aList service quota metrics for a consumer. Args: service: The service to list metrics for. consumer: The consumer to list metrics for, e.g. "projects/123". page_size: The page size to list. limit: The max number of metrics to return. Raises: exceptions.PermissionDeniedException: when listing metrics fails. apitools_exceptions.HttpError: Another miscellaneous error with the service. Returns: The list of quota metrics )parentpageSizemetrics)limitbatch_size_attribute batch_sizefield)_ValidateConsumer_GetClientInstanceMESSAGES_MODULE@ServiceconsumermanagementServicesConsumerQuotaMetricsListRequest_SERVICE_CONSUMER_RESOURCEr YieldFromListservices_consumerQuotaMetrics)serviceconsumer page_sizerclientmessagesrequests *lib/googlecloudsdk/api_lib/services/scm.pyListQuotaMetricsr"sm H  &  # #(  U U '7H*= = V ?'  ! ! ** %   c t|t}|j}t||} |j t ||fz|j |j|j|||| g|} |jj| S#tjtjf$r.} tj| tj Yd} ~ yd} ~ wwxYw)aUpdate a quota override. Args: service: The service to update a quota override for. consumer: The consumer to update a quota override for, e.g. "projects/123". metric: The quota metric name. unit: The unit of quota metric. dimensions: The dimensions of the override in dictionary format. It can be None. value: The override integer value. force: Force override update even if the change results in a substantial decrease in available quota. Raises: exceptions.UpdateQuotaOverridePermissionDeniedException: when updating an override fails. apitools_exceptions.HttpError: Another miscellaneous error with the service. Returns: The quota override operation. )metricunit overrideValue dimensions) overrides) inlineSourceforce)r %v1Beta1ImportProducerOverridesRequestN)rrr_GetDimensionsSServiceconsumermanagementServicesConsumerQuotaMetricsImportProducerOverridesRequestr%V1Beta1ImportProducerOverridesRequestV1Beta1OverrideInlineSourceV1Beta1QuotaOverriderImportProducerOverridesapitools_exceptionsHttpForbiddenErrorHttpNotFoundErrorr ReraiseError,UpdateQuotaOverridePermissionDeniedException) rrr%r&r(valuer+rrdimensions_messager es r!UpdateQuotaOverrideCallr;>s8H  &  # #(%h ;  h h '7H*= =,4,,;;//#$)!3 05<-  i 'D  / / G G PP  0 0  / / 1D : B BDDDsB#C,>$C''C,c|t|t}|j}t||||}t||fz} |j | |} |j j| S#tjtjf$r.} tj| tjYd} ~ yd} ~ wwxYw)auDelete a quota override. Args: service: The service to delete a quota aoverride for. consumer: The consumer to delete a quota override for, e.g. "projects/123". metric: The quota metric name. unit: The unit of quota metric. override_id: The override ID. force: Force override deletion even if the change results in a substantial decrease in available quota. Raises: exceptions.DeleteQuotaOverridePermissionDeniedException: when deleting an override fails. apitools_exceptions.HttpError: Another miscellaneous error with the service. Returns: The quota override operation. )namer+N)rrr_GetMetricResourceName_LIMIT_OVERRIDE_RESOURCEYServiceconsumermanagementServicesConsumerQuotaMetricsLimitsProducerOverridesDeleteRequest6services_consumerQuotaMetrics_limits_producerOverridesDeleter3r4r5rr6,DeleteQuotaOverridePermissionDeniedException) rrr%r& override_idr+rrr r=r r:s r!DeleteQuotaOverrideCallrEus2H  &  # #( !'8VT B& !V[$9 9$  n n   o 'D  H H O O   0 0  / / 1D : B BDDDsA**#B; $B66B;c |y|jj}|t|jDcgc]}|j |||c}Scc}w)N)keyr8)additionalProperties)r1DimensionsValuesortedkeysAdditionalProperty)rr(dtks r!r-r-sk $$44" *//+,,a  AZ]  ;,  sAct||}|D]B}|j|k(s|jD]!}|j|k(s|jccSDt j d|d|d)aGet the metric resource name from metric name and unit. Args: service: The service to manage an override for. consumer: The consumer to manage an override for, e.g. "projects/123". metric: The quota metric name. unit: The unit of quota metric. Raises: exceptions.Error: when the limit with given metric and unit is not found. Returns: The quota override operation. zlimit not found with name "z " and unit "z".)r"r%consumerQuotaLimitsr&r=rError)rrr%r&rmqs r!r>r>si Wh /' axx6$$! 66T>-%   $( ))r#c2t}|j}|j|} |jj |S#t j t jf$r.}tj|tjYd}~yd}~wwxYw)a)Make API call to get an operation. Args: name: The name of the operation. Raises: exceptions.OperationErrorException: when the getting operation API fails. apitools_exceptions.HttpError: Another miscellaneous error with the service. Returns: The result of the operation )r=N) rr-ServiceconsumermanagementOperationsGetRequest operationsGetr3r4r5rr6OperationErrorException)r=rrr r:s r! GetOperationrYs  &  # #(  B B B M'C     ))  0 0  / / 1CAzAABBCsA#B($BBcntD]}|j|sytjd|z)Nzinvalid consumer format "%s".)_VALID_CONSUMER_PREFIX startswithrrQ)rprefixs r!rrs6&f6" ' 88CDDr#c.tjddS)Nserviceconsumermanagementv1beta1)rGetClientInstancer#r!rrs   ;Y GGr#)NN)F)__doc__ __future__rrrapitools.base.pyrr3rgooglecloudsdk.api_lib.servicesgooglecloudsdk.api_lib.utilrrr?r[r"r;rEr-r>rYrrrbr#r!rhsk8&'>'6,-4DJ#( 4Dx#( )DX  )2C.EHr#