
                         p    S 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.    N)
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*   7   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   !ListProjectAddressGroupReferencesr5   @   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   GNetworksecurityOrganizationsLocationsAddressGroupsListReferencesRequestr/   r0   s        r   &ListOrganizationAddressGroupReferencesr9   I   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   YieldFromListr@   	page_size)r2   r4   r   r   r%   s        r   r/   r/   T   sc    ----335-m&@&@&BC'		!	!JJ$)*%

 
r   )__doc__r   apitools.base.pyr   'googlecloudsdk.api_lib.network_securityr   r   googlecloudsdk.corer   r   r   r   r(   r*   r5   r9   r/   r   r   r   <module>rP      sC    3 	 ' E E #
AAr   