
                         F    S r SSKJr  SSKJr  SrSrS rS rS r	S	 r
S
 rg)z,Utility functions for the KMS Inventory CLI.    )
list_pager)apiskmsinventoryv1c                  @    [         R                  " [        [        5      $ N)r   GetClientInstanceDEFAULT_API_NAMEDEFAULT_API_VERSION     4lib/googlecloudsdk/api_lib/kmsinventory/inventory.pyr	   r	          			 02E	FFr   c                  @    [         R                  " [        [        5      $ r   )r   GetMessagesModuler
   r   r   r   r   r   r       r   r   c           	          [        5       n[        5       R                  SU -   S9n[        R                  " UR
                  UUR                  SUR                  SS9$ )Nz	projects/)parentpageSize
cryptoKeys)limitbatch_size_attribute
batch_sizefield)r	   r   )KmsinventoryProjectsCryptoKeysListRequestr   YieldFromListprojects_cryptoKeysr   	page_size)projectargsclientrequests       r   ListKeysr"   $   s`    &II7" J $' 
	!	!  JJ%
 r   c                 z    [        5       n[        5       R                  U S9nUR                  R	                  U5      $ )N)name)r	   r   RKmsinventoryProjectsLocationsKeyRingsCryptoKeysGetProtectedResourcesSummaryRequest&projects_locations_keyRings_cryptoKeysGetProtectedResourcesSummary)r$   r    r!   s      r   r'   r'   2   sI    & VV W  
 
	6	6	S	S
 r   c           
          [        5       n[        5       R                  XUS9n[        R                  " UR
                  USUR                  SUR                  SS9$ )N)scope	cryptoKeyresourceTypesSearchr   protectedResources)methodr   r   r   r   )r	   r   8KmsinventoryOrganizationsProtectedResourcesSearchRequestr   r    organizations_protectedResourcesr   r   )r)   key_nameresource_typesr   r    r!   s         r   SearchProtectedResourcesr3   ;   s`    &XX^ Y ' 
	!	!--JJ% 
 r   N)__doc__apitools.base.pyr   googlecloudsdk.api_lib.utilr   r
   r   r	   r   r"   r'   r3   r   r   r   <module>r7      s9     3 ( ,!  GGr   