PdZddlmZddlmZddlmZddlmZddlm Z ddlmZddl m Z dZ d Z d Zd ZGd d eZGddeZGddeZGddeZGddeZdZdZy)z1Helpers for interacting with the Procurement API.)absolute_import)division)unicode_literals) exceptions)apis) properties cloudcommerceconsumerprocurementv1alpha1c>tjttSN)rGetMessagesModule&COMMERCE_PROCUREMENT_CONSUMER_API_NAME)COMMERCE_PROCUREMENT_CONSUMER_API_VERSION7lib/googlecloudsdk/api_lib/commerce_procurement/apis.pyr r    F I KKrc>tjttSr )rGetClientInstancerrrrrrr#rrceZdZdZej ZejZe dZ e dZ e dZ y)Accountsz@The Accounts set of Commerce Procurement Consumer API functions.c*tjSr )rbillingAccounts_accountsrrr GetServicezAccounts.GetService0s   7 77rctj|} tjj |S#t j $r}tj|d}~wwxYw)zCalls the Procurement Consumer Accounts.Get method. Args: account_name: Name of an account. Returns: (Account) nameN) r ACloudcommerceconsumerprocurementBillingAccountsAccountsGetRequestrrGetapitools_exceptions HttpErrorr HttpException) account_namerequesterrors rrz Accounts.Get4sk GG H ,  " & &w //  ( (,  $ $U ++,"?A,A''A,ctj|||} tjj |S#t j $r}tj|d}~wwxYw)a2Calls the Procurement Consumer Accounts.List method. Args: billing_account_name: Name of a billing account. page_size: Max size of records to be retrieved in page. page_token: Token to specify page in list. Returns: List of Accounts and next page token if applicable. parentpageSize pageTokenN) r BCloudcommerceconsumerprocurementBillingAccountsAccountsListRequestrrListr r!rr")billing_account_name page_size page_tokenr$r%s rr-z Accounts.ListFsp HH#i:IO ,  " ' ' 00  ( (,  $ $U ++,s"AA.A))A.N) __name__ __module__ __qualname____doc__r r GET_REQUESTr, LIST_REQUEST staticmethodrrr-rrrrr(sgH!EE"FF88,,",,rrceZdZdZej ZejZe dZ e dZ e dZ y) EntitlementszDThe Entitlements set of Commerce Procurement Consumer API functions.c*tjSr )rprojects_entitlementsrrrrzEntitlements.GetServicecs   4 44rctj|} tjj |S#t j $r}tj|d}~wwxYw)zCalls the Procurement Consumer Entitlements.Get method. Args: entitlement_name: Name of an entitlement. Returns: (Entitlement) rN) r >CloudcommerceconsumerprocurementProjectsEntitlementsGetRequestr9rrr r!rr")entitlement_namer$r%s rrzEntitlements.Getgsk DD E ,  $ $ & * *7 33  ( (,  $ $U ++,r&cLdtjjjj z}t j |||} tjj|S#tj$r}tj|d}~wwxYw)aCalls the Procurement Consumer Entitlements.List method. Args: page_size: Max size of records to be retrieved in page. page_token: Token to specify page in list. Returns: List of Entitlements and next page token if applicable. projects/%sr(N)rVALUEScoreproject GetOrFailr ?CloudcommerceconsumerprocurementProjectsEntitlementsListRequestr9rr-r r!rr")r/r0 project_namer$r%s rr-zEntitlements.Listys!:#4#4#9#9#A#A#K#K#MMLEEi:FG ,  $ $ & + +G 44  ( (,  $ $U ++,s"A66B# BB#N) r1r2r3r4r r=r5rEr6r7rrr-rrrr9r9[sgL!BB"CC55,,",,rr9ceZdZdZej ZejZe dZ e dZ e dZ y) FreeTrialszCThe Free Trials set of Commerce Procurement Consumer API functions.c*tjSr )rprojects_freeTrialsrrrrzFreeTrials.GetServices   2 22rcdtjjjj z}d|z}t j ||}t j||} tjj|S#tj$r}tj|d}~wwxYw)zCalls the Procurement Consumer FreeTrials.Create method. Args: provider_id: Id of the provider. product_external_name: Name of the product. Returns: (Operation) r@ providers/%s)providerproductExternalName)r)7googleCloudCommerceConsumerProcurementV1alpha1FreeTrialN)rrArBrCrDr 7GoogleCloudCommerceConsumerProcurementV1alpha1FreeTrial?CloudcommerceconsumerprocurementProjectsFreeTrialsCreateRequestrHrCreater r!rr") provider_idproduct_external_namerF provider_name free_trialr$r%s rrRzFreeTrials.Creates!:#4#4#9#9#A#A#K#K#MML"[0M"==4I>K EE@JFL ,  " " $ + +G 44  ( (,  $ $U ++,s2"BC(B==CcNdtjjjj z}t j ||||} tjj|S#tj$r}tj|d}~wwxYw)aHCalls the Procurement Consumer FreeTrials.List method. Args: page_size: Max size of records to be retrieved in page. page_token: Token to specify page in list. filter_rule: The filter that can be used to limit the the result. Returns: List of Free Trials and next page token if applicable. r@r)r*r+filterN)rrArBrCrDr =CloudcommerceconsumerprocurementProjectsFreeTrialsListRequestrHrr-r r!rr")r/r0 filter_rulerFr$r%s rr-zFreeTrials.Lists!:#4#4#9#9#A#A#K#K#MMLCC D ,  " " $ ) )' 22  ( (,  $ $U ++,s"A77B$ BB$N) r1r2r3r4r rQCREATE_REQUESTrZr6r7rrRr-rrrrHrHsgK$CC"AA33,,0,,rrHceZdZdZej ZejZejZ ejZ ejZedZedZedZedZedZedZy) Ordersz>The Orders set of Commerce Procurement Consumer API functions.c*tjSr )rbillingAccounts_ordersrrrrzOrders.GetServices   5 55rctj|}tj||} tj j |S#t j$r}tj|d}~wwxYw)zCalls the Procurement Consumer Orders.Cancel method. Args: order_name: Name of an order. etag: Weak etag for validation purpose. Returns: (Operation) )etag)r@googleCloudCommerceConsumerProcurementV1alpha1CancelOrderRequestN) r @GoogleCloudCommerceConsumerProcurementV1alpha1CancelOrderRequestBCloudcommerceconsumerprocurementBillingAccountsOrdersCancelRequestr^rCancelr r!rr") order_namerbcancel_detail_requestr$r%s rrfz Orders.Cancels.FF G HH I^I ,    ' ' 00  ( (,  $ $U ++,s"AB-BBctj|} tjj |S#t j $r}tj|d}~wwxYw)zCalls the Procurement Consumer Orders.Get method. Args: order_name: Name of an order. Returns: (Order) rN) r ?CloudcommerceconsumerprocurementBillingAccountsOrdersGetRequestr^rrr r!rr")rgr$r%s rrz Orders.Getsk EE F ,    $ $W --  ( (,  $ $U ++,r&ctj||||} tjj |S#t j $r}tj|d}~wwxYw)avCalls the Procurement Consumer Orders.List method. Args: billing_account_name: Name of a billing account. page_size: Max size of records to be retrieved in page. page_token: Token to specify page in list. filter_rule: The filter that can be used to limit the the result. Returns: List of orders and next page token if applicable. rXN) r @CloudcommerceconsumerprocurementBillingAccountsOrdersListRequestr^rr-r r!rr")r.r/r0r[r$r%s rr-z Orders.Listst FF# G ,    % %g ..  ( (,  $ $U ++,s"AA/A**A/c &|rg}|D]}d}d}g} |jD]-\} } | dk(s| dk(r| dk(r|d}| dk(r|d}(| jdd} t| dkrtj | } n| d d k(rtj | d } n{| d d k(r'tj t | d } nL| d dk(r'tj t | d} ntdj| d | jtj| | 0|r|rtj||| }nd}|jtj|dt|d|tj|tj|}n=tj|tj!t#||}tj%||} t&j)j+|S#t,j.$r}t1j2|d}~wwxYw)aCalls the Procurement Consumer Orders.Cancel method. Args: order_name: Name of an order. etag: Weak etag for validation purpose. product_requests: Modification details if order modification is based on product plans. quote_change_type: Change type if order modification is based on quote. new_quote_external_name: External name of new quote. Returns: (Operation) z line-item-idzline-item-change-typeproduct-external-nameflavor-external-name: stringValuerstrint int64Valuedouble doubleValuezUnrecognized value type {}rvaluerNflavorExternalName parametersN) lineItemId changeTypenewLineItemInfo) modifications)rbmodifyProductsOrderRequest)rnewQuoteExternalName)rbmodifyQuoteOrderRequest)r@googleCloudCommerceConsumerProcurementV1alpha1ModifyOrderRequest)itemssplitlenr rs8&'>,2*)K&,6)K K 0,v0,f0,60,f;,;,|J,VJ,Z,,:Q6Mr