4LdZddlmZdZdZdZdZdeded efd Z Gd d Z y )z5Client for interacting with Storage Intelligence API.)apiszedition_config,filterfolders organizationsprojects resource_type resource_idreturnc|d|dS)z$Returns the full id of the resource./z$/locations/global/intelligenceConfigrrs >lib/googlecloudsdk/api_lib/storage/storage_intelligence_api.py_get_full_id_stringrs ?!K=(L MMceZdZdZfdZ ddZ ddZ ddZdZdZ dZ d Z d Z d Z dd Z dd Z ddZxZS)StorageIntelligenceApiz$Client for Storage Intelligence API.ctt| tjdd|_tj dd|_y)Nstoragev2)superr__init__ core_apisGetClientInstanceclientGetMessagesModulemessages)self __class__s rrzStorageIntelligenceApi.__init__"s9 $02--i>DK// 4@DMrc|jj}|jjjj|_||_||t |}||S)z-Disables Intelligence for the given resource.intelligenceConfig updateMaskname)rIntelligenceConfigEditionConfigValueValuesEnumDISABLED editionConfigr#_FULL_UPDATE_MASK)rfull_idrequest_message_class patch_methodintelligence_configrequests r_disable_intelligencez,StorageIntelligenceApi._disable_intelligence'sc--::< ((EENN% '#.$ G   rc|jj|_d}|r4|jjjj |_|dzS|r*|jjjjn)|jjjj|_|0|jj||j_ |dz }|0|jj||j_ |dz }|0|jj||j_ |dz }|0|jj||j_|dz }|S) aUpdates the Intelligence filter and returns update_mask. Args: intelligence_config: The Intelligence Config object to be updated. inherit_from_parent: Whether to inherit config from the parent. trial_edition: Whether to enable Storage Intelligence for TRIAL edition. include_locations: List of locations to be included in the filter. exclude_locations: List of locations to be excluded in the filter. include_bucket_id_regexes: List of bucket id regexes to be included in the filter. exclude_bucket_id_regexes: List of bucket id regexes to be excluded in the filter. Returns: The update mask to be used for the request. edition_configz,filter) locationsz(,filter.included_cloud_storage_locationsz(,filter.excluded_cloud_storage_locations)bucketIdRegexesz&,filter.included_cloud_storage_bucketsz&,filter.excluded_cloud_storage_buckets)rFilterfilterr$r%INHERITr'TRIALSTANDARDCloudStorageLocationsincludedCloudStorageLocationsexcludedCloudStorageLocationsCloudStorageBucketsincludedCloudStorageBucketsexcludedCloudStorageBuckets) rr,inherit_from_parent trial_editioninclude_locationsexclude_locationsinclude_bucket_id_regexesexclude_bucket_id_regexes update_masks r_set_intelligence_filterz/StorageIntelligenceApi._set_intelligence_filter<s6"&!5!5!7"K -- * * G G O O' ""  ((EEKK ]] - - J J S S%$ -- - -8I - J  >??k$ -- - -8I - J  >??k!, -- + +7 ,   < ==k , -- + +7 ,   < ==k rc |jj} || _|j| ||||||} | || | |S)z7Updates the Intelligence Config for the given resource.r )rr$r#rE) rr)r>r?r@rArBrCr*r+r,rDs r_update_intelligence_configz2StorageIntelligenceApi._update_intelligence_configse--::<&//!!K 2"  rctt|}|jj|}|jj j |S)zGets the Intelligence Config for the given sub folder. Args: folder_id: Id of the GCP hierarchy folder. Returns: The Intelligence Config object. r r#)r_FOLDER_RESOURCE_TYPEr3StorageFoldersLocationsGetIntelligenceConfigRequestrfolders_locationsGetIntelligenceConfig)r folder_id full_namer-s r"get_sub_folder_intelligence_configz9StorageIntelligenceApi.get_sub_folder_intelligence_configsO$+ImmOO PG ;; ( ( > >w GGrctt|}|jj|}|jj j |S)zGets the Intelligence Config for the given project. Args: project_name: Name of the GCP project. Returns: The Intelligence Config object. r rI)r_PROJECT_RESOURCE_TYPEr4StorageProjectsLocationsGetIntelligenceConfigRequestrprojects_locationsrM)r project_namerOr-s rget_project_intelligence_configz6StorageIntelligenceApi.get_project_intelligence_configsR$,,I JJ K  ;; ) ) ? ? HHrctt|}|jj|}|jj j |S)zGets the Intelligence Config for the given organization. Args: organization_id: Id of the GCP organization. Returns: The Intelligence Config object. r rI)r_ORGANIZATION_RESOURCE_TYPEr9StorageOrganizationsLocationsGetIntelligenceConfigRequestrorganizations_locationsrM)rorganization_idrOr-s r$get_organization_intelligence_configz;StorageIntelligenceApi.get_organization_intelligence_configsR$1I OO P  ;; . . D DW MMrctt|}|j||jj|j j jS)zDisables the Storage Intelligence for the given project. Args: project_name: Name of the GCP project. Returns: The Intelligence Config object. r r)r*r+)rrRr.r7StorageProjectsLocationsUpdateIntelligenceConfigRequestrrTUpdateIntelligenceConfig)rrUrOs r#disable_project_intelligence_configz:StorageIntelligenceApi.disable_project_intelligence_configsP$,,I  % %"mmcc[[33LL & rctt|}|j||jj|j j jS)zDisables the Storage Intelligence for the given organization. Args: organization_id: Id of the GCP organization. Returns: The Intelligence Config object. r r^)rrXr.rr?r@rArBrCrOs r"update_project_intelligence_configz9StorageIntelligenceApi.update_project_intelligence_configs]6$ gI  + +!! MM &&??  rc td|}|j||||||||jj|jj j S)aUpdates the Intelligence Config for the given GCP sub folder. Args: sub_folder: The GCP sub folder name. inherit_from_parent: Whether to inherit config from the parent. trial_edition: Whether to enable Storage Intelligence for TRIAL edition. include_locations: List of locations to be included in the filter. exclude_locations: List of locations to be excluded in the filter. include_bucket_id_regexes: List of bucket id regexes to be included in the filter. exclude_bucket_id_regexes: List of bucket id regexes to be excluded in the filter. Returns: The Intelligence Config object. rr )rrGrrfrrLr`) r sub_folderr>r?r@rArBrCrOs r%update_sub_folder_intelligence_configz>  rc td|}|j||||||||jj|jj j S)aUpdates the Intelligence Config for the given GCP organization. Args: organization: The GCP organization name. inherit_from_parent: Whether to inherit config from the parent. trial_edition: Whether to enable Storage Intelligence for TRIAL edition. include_locations: List of locations to be included in the filter. exclude_locations: List of locations to be excluded in the filter. include_bucket_id_regexes: List of bucket id regexes to be included in the filter. exclude_bucket_id_regexes: List of bucket id regexes to be excluded in the filter. Returns: The Intelligence Config object. rr )rrGrrcrrZr`) r organizationr>r?r@rArBrCrOs r'update_organization_intelligence_configz>StorageIntelligenceApi.update_organization_intelligence_configos]4$%<I  + +!! RR ++DD  r)NNN)FFNNNN) NFFNNNNNN)__name__ __module__ __qualname____doc__rr.rErGrPrVr\rardrhrkrnrq __classcell__)rs@rrrs,A  !0  $ $GV $ $  DH$I(N(&&,  $ $)\  $ $(Z  $ $(rrN) rugooglecloudsdk.api_lib.utilrrr(rJrXrRstrrrr rrrysQ<9,!-#NsNNN xxr