AdZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z d Z Gd d e jZ Gd d eZdZy)zBUtilities for the API to configure cross-project networking (XPN).)absolute_import)division)unicode_literals) list_pager) base_classes) exceptions)utilsv1c eZdZy) XpnApiErrorN)__name__ __module__ __qualname__-lib/googlecloudsdk/api_lib/compute/xpn_api.pyr r srr cZeZdZdZdZdZddZdZdZdZ d Z d Z d Z d Z d ZdZy) XpnClientzA client for interacting with the cross-project networking (XPN) API. The XPN API is a subset of the Google Compute Engine API. cV||_|j|_|j|_yN)compute_clientapitools_clientclientmessages)selfrs r__init__zXpnClient.__init__(s$(D 00DK"++DMrc>|jj|g|S)N)requestserrors_to_collect)r MakeRequests)rrequesterrorss r _MakeRequestzXpnClient._MakeRequest-s(    + +  , ""rNcg}t|j||}|r2|xsd}dj|}tj|t ||dS)Nzcomplete all requestszCould not {0}:r)listr#formatr RaiseExceptionr )r request_tuple operation_msgr"resultsmsgs r_MakeRequestSynczXpnClient._MakeRequestSync2sX F4$$]F;'>m  # #M 2c 6;4 1:rc|jjd|jj|f}dj |}|j ||y)z4Enable the project with the given ID as an XPN host. EnableXpnHostprojectzenable [{project}] as XPN hostN)rprojectsr#ComputeProjectsEnableXpnHostRequestr&r,rr0r(r+s r EnableHostzXpnClient.EnableHost=sV  99'9JLM + 1 1' 1 BC--rc|jjd|jj|f}dj |}|j ||y)z5Disable the project with the given ID as an XPN host.DisableXpnHostr/zdisable [{project}] as XPN hostN)rr1r$ComputeProjectsDisableXpnHostRequestr&r,r3s r DisableHostzXpnClient.DisableHostFsV  ::7:KMM , 2 27 2 CC--rc|jjd|jj|f}dj |}|j ||S)z'Get the XPN host for the given project. GetXpnHostr/z$get XPN host for project [{project}])rr1r ComputeProjectsGetXpnHostRequestr&r,r3s rGetHostProjectzXpnClient.GetHostProjectOsY  66w6GIM 1 7 7 7 HC   44rc|jj|}tj|jj |ddddS)Nr/GetXpnResources maxResults resourcesmethodbatch_size_attribute batch_sizefield)r%ComputeProjectsGetXpnResourcesRequestr YieldFromListrr1)rr0r!s rListEnabledResourceszXpnClient.ListEnabledResourcesXsNmmAABG  # #  )  rc|jj||jj|}tj|j j |ddddS)ayList the projects in an organization that are enabled as XPN hosts. Args: project: str, project ID to make the request with. organization_id: str, the ID of the organization to list XPN hosts for. If None, the organization is inferred from the project. Returns: Generator for `Project`s corresponding to XPN hosts in the organization. ) organization)r0projectsListXpnHostsRequest ListXpnHostsr?r@itemsrB)r"ComputeProjectsListXpnHostsRequestProjectsListXpnHostsRequestrrHrr1)rr0organization_idr!s rListOrganizationHostProjectsz&XpnClient.ListOrganizationHostProjectscslmm>>$(MM$M$M(%N%*?+G  # # )  rc&|jj|jj||}|jj||}|jj d|f}dj ||}|j||y)a,Associate the given resource with the given XPN host project. Args: host_project: str, ID of the XPN host project associated_resource: ID of the resource to associate with host_project xpn_resource_type: XpnResourceId.TypeValueValuesEnum, the type of the resource idtype xpnResource)r0 projectsEnableXpnResourceRequestEnableXpnResourcezAenable resource [{0}] as an associated resource for project [{1}]N)r ProjectsEnableXpnResourceRequest XpnResourceId'ComputeProjectsEnableXpnResourceRequestrr1r&r,)r host_projectassociated_resourcexpn_resource_typeprojects_enable_requestr!r(r+s r_EnableXpnAssociatedResourcez&XpnClient._EnableXpnAssociatedResourcezs#mmLLMM//""0$M mmCC)@DBG[[))+>HM  &':L I--rc|jjj}|j|||jy)zAssociate the given project with the given XPN host project. Args: host_project: str, ID of the XPN host project associated_project: ID of the project to associate r`N)rr\TypeValueValuesEnumrbPROJECTrr^associated_project xpn_typess rEnableXpnAssociatedProjectz$XpnClient.EnableXpnAssociatedProjects; ++??I%%(IrsPI&''750*""ZOZOz"r