dZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z Gd d eZGd d eZGd deZy)zNCommands for interacting with the Cloud NetApp Files Quota Rules API resource.) list_pager) constants)util)waiter)base)log) resourcesceZdZdZej j fdZedZ edZ dZ dZ ddZ dd Zd Zd Z dd Zd Zy)QuotaRulesClientzMWrapper for working with Quota Rules in the Cloud NetApp Files API Client. cB||_|jtjjk(rt |_y|jtjj k(rt|_ytdjtj|)Nz [{}] is not a valid API version.) release_trackr ReleaseTrackBETABetaQuotaRulesAdapter_adapterGAQuotaRulesAdapter ValueErrorformatnetapp_api_util VERSION_MAP)selfr s ?lib/googlecloudsdk/api_lib/netapp/volumes/quota_rules/client.py__init__zQuotaRulesClient.__init__s{&D T..333+-dm  t0033 3')dm 9@@  % %m 46 77c.|jjSN)rclientrs rrzQuotaRulesClient.client)s ==  rc.|jjSr)rmessagesrs rr!zQuotaRulesClient.messages-s == ! !!rctjtj|jj|dj |j S)aWaits on the long-running operation until the done field is True. Args: operation_ref: the operation reference. Raises: waiter.OperationError: if the operation contains an error. Returns: the 'response' field of the Operation. zWaiting for [{0}] to finish)rWaitForCloudOperationPollerNoResourcesrprojects_locations_operationsrName)r operation_refs rWaitForOperationz!QuotaRulesClient.WaitForOperation1sL >>.. KK 5 5 78E%,,]-?-?-AB DDrc@|jj||j|}|jjj |}|r|St jj|jtj}|j|S)z(Create a Cloud NetApp Volume Quota Rule.)parent quotaRuleId quotaRule collection) r!5NetappProjectsLocationsVolumesQuotaRulesCreateRequestr&r%projects_locations_volumes_quotaRulesCreater REGISTRYParseRelativeNamenamerOPERATIONS_COLLECTIONr()rquota_rule_ref volume_refasync_configrequest create_opr's rCreateQuotaRulez QuotaRulesClient.CreateQuotaRuleBs KK&++- L   AAHHI &&889#B#B9M   //rNc|jj}||_||_||_||_||_||_|S)zFParses the command line arguments for Create Quota Rule into a config.)r! QuotaRuler4typetarget diskLimitMib descriptionlabels)rr4quota_rule_typer@disk_limit_mibrBrC quota_rules rParseQuotaRuleConfigz%QuotaRulesClient.ParseQuotaRuleConfigUsJ((*JJO%JOJ,J(JJ rcf|jj|}|jjj |}|j D]&}t jdj|(tj|jj|tj|dS)aoList Cloud NetApp Volume Quota Rules. Args: volume_ref: The parent Volume to list Cloud Netapp Volume QuotaRules. limit: The number of Cloud Netapp Volume QuotaRules to limit the results to. This limit is passed to the server and the server does the limiting. Returns: Generator that yields the Cloud Netapp Volume QuotaRules. )r*zLocation {} may be unreachable.pageSize)fieldlimitbatch_size_attribute) r!3NetappProjectsLocationsVolumesQuotaRulesListRequestrr0List unreachablerwarningrr YieldFromListrQUOTA_RULE_RESOURCE)rr7rKr:responselocations rListQuotaRuleszQuotaRulesClient.ListQuotaRuleshsmmOOPG{{@@EEgNH(( kk3::8DE)  # # 99++'  rc|jj|j}|jjj |S)z%Get a Cloud NetApp Volume Quota Rule.r4)r!2NetappProjectsLocationsVolumesQuotaRulesGetRequest RelativeNamerr0Get)rr6r:s r GetQuotaRulezQuotaRulesClient.GetQuotaRulesEmmNN  ( ( *OG ;; < < @ @ IIrc<|jj|j}|jjj |}|r|St jj|jtj}|j|S)z(Delete a Cloud NetApp Volume Quota Rule.rWr-) r!5NetappProjectsLocationsVolumesQuotaRulesDeleteRequestrYrr0Deleter r2r3r4rr5r()rr6r8r: delete_opr's rDeleteQuotaRulez QuotaRulesClient.DeleteQuotaRules KK,,. L   AAHHI &&889#B#B9M   //rc<|||_|||_|||_|S)aGParses updates into a quota rule config. Args: quota_rule_config: The quota rule config to update. disk_limit_mib: int, a new disk limit, if any. description: str, a new description, if any. labels: LabelsValue message, the new labels value, if any. Returns: The quota rule message. )rArBrC)rquota_rule_configrErBrCs rParseUpdatedQuotaRuleConfigz,QuotaRulesClient.ParseUpdatedQuotaRuleConfigs7$!'5$&1# !' rc@|jj|j||}|jjj |}|r|St jj|jtj}|j|S)aeUpdates a Cloud NetApp Volume Quota Rule. Args: quota_rule_ref: the reference to the Quota Rule. quota_rule: Quota rule config, the updated quota rule. update_mask: str, a comma-separated list of updated fields. async_: bool, if False, wait for the operation to complete. Returns: an Operation or Volume message. )r4 updateMaskr,r-) r!4NetappProjectsLocationsVolumesQuotaRulesPatchRequestrYrr0Patchr r2r3r4rr5r()rr6rF update_maskr8r: update_opr's rUpdateQuotaRulez QuotaRulesClient.UpdateQuotaRules JJ,,."  K   AAGGPI &&889#B#B9DM   //r)NNNNNNr)NNN)__name__ __module__ __qualname____doc__rrrrpropertyrr!r(r<rGrUr[r`rcrjrrr r s$(#4#4#9#97     " "D"0*   &4J0(  40rr ceZdZdZdZy)rz;Adapter for the Cloud NetApp Files API Quota Rule resource.ctjj|_t j |j|_t j|j|_yN)r ) rrrr rGetClientInstancerGetMessagesModuler!rs rrzQuotaRulesAdapter.__init__sN**--D!33((DK$55((DMrN)rkrlrmrnrrprrrrs Crrc"eZdZdZfdZxZS)rz@Adapter for the Beta Cloud NetApp Files API Quota Rule resource.ctt| tjj |_tj|j |_ tj|j |_ yrs) superrrrrrr rrtrrur!)r __class__s rrzBetaQuotaRulesAdapter.__init__s] /1**//D!33((DK$55((DMr)rkrlrmrnr __classcell__)rys@rrrsHrrN)rnapitools.base.pyrgooglecloudsdk.api_lib.netapprrrgooglecloudsdk.api_lib.utilrgooglecloudsdk.calliopergooglecloudsdk.corerr objectr rrrprrrsH U'3A.(#)q0vq0h   - r