9LdZddlmZdZdZdZdZdeded efd Z Gd d Z y )z+Client for interacting with Management Hub.)apiszedition_config,filterfolders organizationsprojects resource_type resource_idreturnc|d|dS)z$Returns the full id of the resource./z/locations/global/managementHubrrs 8lib/googlecloudsdk/api_lib/storage/management_hub_api.py_get_full_id_stringrs ?!K=(G HHceZdZdZfdZ ddZdZ ddZ ddZdZ dZ d Z d Z d Z d Z dd Z ddZ ddZxZS)ManagementHubApizClient for Management Hub API.ctt| tjdd|_tj dd|_y)Nstoragev2)superr__init__ core_apisGetClientInstanceclientGetMessagesModulemessages)self __class__s rrzManagementHubApi.__init__"s9 D*,--i>DK// 4@DMrc|jj}|jjjj|_||_||t |}||S)z3Disables the management hub for the given resource. managementHub updateMaskname)r ManagementHubEditionConfigValueValuesEnumDISABLED editionConfigr#_FULL_UPDATE_MASK)rfull_idrequest_message_class patch_methodmanagement_hubrequests r_disable_management_hubz(ManagementHubApi._disable_management_hub'sa]]002N ##@@II "N#$$ G   rc>g}|9|j|Dcgc]}|jj| c}|9|j|Dcgc]}|jj| c}|jj|Scc}wcc}w)aReturns the cloud storage buckets filter for the given args. Args: bucket_ids: List of bucket ids to be included in the filter. bucket_id_regexes: List of bucket id regexes to be included in the filter. Returns: The cloud storage buckets filter message. )bucketId) bucketIdRegex)cloudStorageBuckets)extendrCloudStorageBucket&ManagementHubFilterCloudStorageBuckets)r bucket_idsbucket_id_regexesbuckets_filter bucket_idbucket_id_regexs r!_get_cloud_storage_buckets_filterz2ManagementHubApi._get_cloud_storage_buckets_filter<sN%%i -- * *I * >% $!2!2o -- * * * I!2 == ? ?* @  s #B#Bc |jj|_d} |r4|jjjj |_| dzS|jjjj|_|0|jj||j_ | dz } |0|jj||j_ | dz } ||&|j|||j_ | dz } ||&|j|||j_ | dz } | S)aUpdates the management hub filter and returns update_mask. Args: management_hub: The management hub object to be updated. inherit_from_parent: Whether to inherit config from the parent. include_locations: List of locations to be included in the filter. exclude_locations: List of locations to be excluded in the filter. include_bucket_ids: List of bucket ids to be included in the filter. exclude_bucket_ids: List of bucket ids 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_locationsz&,filter.included_cloud_storage_bucketsz&,filter.excluded_cloud_storage_buckets)rManagementHubFilterfilterr$r%INHERITr'STANDARD(ManagementHubFilterCloudStorageLocationsincludedCloudStorageLocationsexcludedCloudStorageLocationsr;includedCloudStorageBucketsexcludedCloudStorageBuckets) rr,inherit_from_parentinclude_locationsexclude_locationsinclude_bucket_idsexclude_bucket_idsinclude_bucket_id_regexesexclude_bucket_id_regexes update_masks r_set_management_hub_filterz+ManagementHubApi._set_management_hub_filterXsl:!MM==?N"K -- % % B B J J" "" ##@@II $ -- @ @) A 9 ??k$ -- @ @) A 9 ??k%)B)N  0 0 "; 7 ==k%)B)N  0 0 "; 7 ==k rc |jj} || _|j| |||||||} | | | | |S)z2Updates the management hub for the given resource.r )rr$r#rP) rr)rHrIrJrKrLrMrNr*r+r,rOs r_update_management_hubz'ManagementHubApi._update_management_hubsd]]002N!N11!! K ([w  rctt|}|jj|}|jj j |S)zGets the management hub for the given sub folder. Args: folder_id: Id of the GCP hierarchy folder. Returns: The management hub object for the given sub folder. r r#)r_FOLDER_RESOURCE_TYPEr.StorageFoldersLocationsGetManagementHubRequestrfolders_locationsGetManagementHub)r folder_id full_namer-s rget_sub_folder_management_hubz.ManagementHubApi.get_sub_folder_management_hubsO$+ImmJJ KG ;; ( ( 9 9' BBrctt|}|jj|}|jj j |S)zGets the management hub for the given project. Args: project_name: Name of the GCP project. Returns: The management hub object for the given project. r rT)r_PROJECT_RESOURCE_TYPEr/StorageProjectsLocationsGetManagementHubRequestrprojects_locationsrX)r project_namerZr-s rget_project_management_hubz+ManagementHubApi.get_project_management_hubsO$,,ImmKK LG ;; ) ) : :7 CCrctt|}|jj|}|jj j |S)zGets the management hub for the given organization. Args: organization_id: Id of the GCP organization. Returns: The management hub object for the given organization. r rT)r_ORGANIZATION_RESOURCE_TYPEr4StorageOrganizationsLocationsGetManagementHubRequestrorganizations_locationsrX)rorganization_idrZr-s rget_organization_management_hubz0ManagementHubApi.get_organization_management_hubsR$1I JJ K  ;; . . ? ? HHrctt|}|j||jj|j j jS)zDisables the management hub for the given project. Args: project_name: Name of the GCP project. Returns: The management hub object for the given project. r r)r*r+)rr]r.r2StorageProjectsLocationsUpdateManagementHubRequestrr_UpdateManagementHub)rr`rZs rdisable_project_management_hubz/ManagementHubApi.disable_project_management_hubsP$,,I  ' '"mm^^[[33GG ( rctt|}|j||jj|j j jS)zDisables the management hub for the given organization. Args: organization_id: Id of the GCP organization. Returns: The management hub object for the given organization. r ri)rrcr.r7StorageOrganizationsLocationsUpdateManagementHubRequestrrerk)rrfrZs r#disable_organization_management_hubz4ManagementHubApi.disable_organization_management_hubsP$1I  ' '"mmcc[[88LL ( rctt|}|j||jj|j j jS)zDisables the management hub for the given sub folder. Args: sub_folder_id: Id of the GCP hierarchy folder. Returns: The management hub object for the given sub folder. r ri)rrUr.r1StorageFoldersLocationsUpdateManagementHubRequestrrWrk)r sub_folder_idrZs r!disable_sub_folder_management_hubz2ManagementHubApi.disable_sub_folder_management_hub%sP$+I  ' '"mm]][[22FF ( rc td|} |j| ||||||||jj|jj j S)aUpdates the management hub for the given GCP project. Args: project: Name of the GCP project. inherit_from_parent: Whether to inherit config from the parent. include_locations: List of locations to be included in the filter. exclude_locations: List of locations to be excluded in the filter. include_bucket_ids: List of bucket ids to be included in the filter. exclude_bucket_ids: List of bucket ids 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 management hub object. rr )rrRrrjrr_rk) rprojectrHrIrJrKrLrMrNrZs rupdate_project_management_hubz.ManagementHubApi.update_project_management_hub8s`<$ gI  & &!! HH &&::  rc td|} |j| ||||||||jj|jj j S)aUpdates the management hub for the given GCP sub folder. Args: sub_folder: The GCP sub folder name. inherit_from_parent: Whether to inherit config from the parent. include_locations: List of locations to be included in the filter. exclude_locations: List of locations to be excluded in the filter. include_bucket_ids: List of bucket ids to be included in the filter. exclude_bucket_ids: List of bucket ids 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 management hub object. rr )rrRrrqrrWrk) r sub_folderrHrIrJrKrLrMrNrZs r update_sub_folder_management_hubz1ManagementHubApi.update_sub_folder_management_hubgs`:$ZI  & &!! GG %%99  rc td|} |j| ||||||||jj|jj j S)aUpdates the management hub for the given GCP organization. Args: organization: The GCP organization name. inherit_from_parent: Whether to inherit config from the parent. include_locations: List of locations to be included in the filter. exclude_locations: List of locations to be excluded in the filter. include_bucket_ids: List of bucket ids to be included in the filter. exclude_bucket_ids: List of bucket ids 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 management hub object. rr )rrRrrnrrerk) r organizationrHrIrJrKrLrMrNrZs r"update_organization_management_hubz3ManagementHubApi.update_organization_management_hubs`:$%<I  & &!! MM ++??  r)NNN)FNNNNNN) NNNNNNNNNN)NNNNNNN)__name__ __module__ __qualname____doc__rr.r;rPrRr[rargrlrorsrvryr| __classcell__)rs@rrrs&A  !*>  $ $K^ $ $  DC$D$I(&&, $ $-d $ $,b $ $,rrN) rgooglecloudsdk.api_lib.utilrrr(rUrcr]strrrr rrrsQ29,!-#IsIII bbr