
    p                         S r SSKJr  SSKJr  SSKJr  SSKrSSKJr  SSKJ	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S rg)z,Utilities for Data Catalog entries commands.    )absolute_import)division)unicode_literalsN)
list_pager)GetClientInstance)GetMessagesModule)logc                 v    [         R                  R                  SR                  UR                  5      5        U $ )Nz+Items were removed from address group [{}].r	   statusPrintformataddress_groupresponseargss     7lib/googlecloudsdk/command_lib/network_security/util.pyLogRemoveItemsSuccessr      s/    **3::4;M;MN 
/    c                 v    [         R                  R                  SR                  UR                  5      5        U $ )Nz'Items were added to address group [{}].r   r   s     r   LogAddItemsSuccessr   #   s/    **/66t7I7IJ 
/r   c                      g)zSet default location to global.global r   r   r   SetGlobalLocationr   *   s    	r   c                     UR                   n[        R                  R                  U5      U:X  aA  [        R                  R	                  UR
                  5      nU< SU< 3UR                  l        U$ )N/)sourceospathbasenamedirnameaddressGroupcloneAddressGroupItemsRequestsourceAddressGroup)_argrequestsource_namelocations        r   FormatSourceAddressGroupr+   /   sX    

+WWk"k1www334H@G))< 
.r   c                     [         R                  R                  SR                  UR                  UR
                  5      5        U $ )Nz2Items were cloned to address group [{}] from [{}].)r	   r   r   r   r   r   r   s     r   LogCloneItemsSuccessr-   :   s8    **:AA


dkk
 
/r   c                 r    [        U 5      R                  n[        U 5      nUR                  n[	        X$U5      $ N)r    projects_locations_addressGroupsr   BNetworksecurityProjectsLocationsAddressGroupsListReferencesRequestListAddressGroupReferencesrelease_trackr   servicemessagesrequest_types        r   !ListProjectAddressGroupReferencesr8   C   s9    m,MM'}-(QQ  
$G4	@@r   c                 r    [        U 5      R                  n[        U 5      nUR                  n[	        X$U5      $ r/   )r   %organizations_locations_addressGroupsr   GNetworksecurityOrganizationsLocationsAddressGroupsListReferencesRequestr2   r3   s        r   &ListOrganizationAddressGroupReferencesr<   L   sA    )) 
 }-(VV  
$G4	@@r   c                     UR                   R                  R                  5       nU" UR                  5       S9n[        R
                  " U UUR                  UR                  SSSSSS9	$ )N)r#   ListReferencesaddressGroupReferences	pageTokennextPageTokenpageSize)limit
batch_sizemethodfieldcurrent_token_attributenext_token_attributebatch_size_attribute)CONCEPTSr   ParseRelativeNamer   YieldFromListrC   	page_size)r5   r7   r   r   r(   s        r   r2   r2   W   sc    ----335-m&@&@&BC'		!	!JJ$)*%

 
r   )__doc__
__future__r   r   r   r   apitools.base.pyr   'googlecloudsdk.api_lib.network_securityr   r   googlecloudsdk.corer	   r   r   r   r+   r-   r8   r<   r2   r   r   r   <module>rT      sL    3 &  ' 	 ' E E #
AAr   