
    i	                     j    d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dZ	dZ
d	 Zd
 Zd Zd Zd Zy)z,Utility functions for the KMS Inventory CLI.    )absolute_import)division)unicode_literals)
list_pager)apiskmsinventoryv1c                  >    t        j                  t        t              S N)r   GetClientInstanceDEFAULT_API_NAMEDEFAULT_API_VERSION     4lib/googlecloudsdk/api_lib/kmsinventory/inventory.pyr   r          			 02E	FFr   c                  >    t        j                  t        t              S r   )r   GetMessagesModuler   r   r   r   r   r   r   #   r   r   c                     t               }t               j                  d| z         }t        j                  |j
                  ||j                  d|j                  d      S )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                     t               }t               j                  |       }|j                  j	                  |      S )N)name)r   r   RKmsinventoryProjectsLocationsKeyRingsCryptoKeysGetProtectedResourcesSummaryRequest&projects_locations_keyRings_cryptoKeysGetProtectedResourcesSummary)r'   r#   r$   s      r   r*   r*   5   sI    & VV W  
 
	6	6	S	S
 r   c           	          t               }t               j                  | ||      }t        j                  |j
                  |d|j                  d|j                  d      S )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   SearchProtectedResourcesr6   >   sb    &XXX^ Y ' 
	!	!--JJ% 
 r   N)__doc__
__future__r   r   r   apitools.base.pyr   googlecloudsdk.api_lib.utilr   r   r   r   r   r%   r*   r6   r   r   r   <module>r;      sB     3 '  ' ' ,!  GGr   