.dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd l mZGd d e jZd ZGddej$Zy)z'Utilities for Eventarc Enrollments API.)absolute_import)division)unicode_literals) list_pager)base)common)apis) exceptions) resourcesceZdZdZy)NoFieldsSpecifiedErrorz:Error when no fields were specified for a Patch operation.N)__name__ __module__ __qualname____doc__2lib/googlecloudsdk/api_lib/eventarc/enrollments.pyr r sBrr cxtjj|jd}|j S)Nz'eventarc.projects.locations.enrollments) collection)r REGISTRYParseRelativeNamenameSelfLink)resource enrollmentss rGetEnrollmentURIr!s8""44mm I5+    rcTeZdZdZfdZd dZdZdZdZdZ dZ d Z d Z xZ S) EnrollmentClientV1zFEnrollment Client for interaction with v1 of Eventarc Enrollments API.ctt| tjtj dt jtjtj }|j|_ |j|_ y)N enrollment) superr__init__rAPI_NAME API_VERSION_1r GetClientInstanceMESSAGES_MODULE _messagesprojects_locations_enrollments_service)selfclient __class__s rr#zEnrollmentClientV1.__init__+s\ d,--|  # #FOOV5I5I JF++DN99DMrc|jj|jj||j |}|j j |S)arCreates a new Enrollment. Args: enrollment_ref: Resource, the Enrollment to create. enrollment_message: Enrollment, the enrollment message that holds enrollment's name, cel_match, message_bus, destination, etc. dry_run: If set, the changes will not be committed, only validated Returns: A long-running operation for create. )parentr! enrollmentId validateOnly)r(1EventarcProjectsLocationsEnrollmentsCreateRequestParent RelativeNameNamer*Create)r+enrollment_refenrollment_messagedry_run create_reqs rr6zEnrollmentClientV1.Create6s_ HH!((*779)',,. I  ==   ++rc|jj|j}|jj |S)zGets the requested Enrollment. Args: enrollment_ref: Resource, the Enrollment to get. Returns: The Enrollment message. r)r(.EventarcProjectsLocationsEnrollmentsGetRequestr4r*Get)r+r7get_reqs rr>zEnrollmentClientV1.GetLsAnnKK  ( ( *LG ==  W %%rc|jj|j|}tj|j |d||dS)asList available enrollments in location. Args: location_ref: Resource, the location to list Enrollments in. limit: int or None, the total number of results to return. page_size: int, the number of entries in each batch (affects requests made, but not the yielded results). Returns: A generator of Enrollments in the location. )r/pageSizerrA)servicerequestfieldlimit batch_sizebatch_size_attribute)r(/EventarcProjectsLocationsEnrollmentsListRequestr4r YieldFromListr*)r+ location_refrE page_sizelist_reqs rListzEnrollmentClientV1.ListZsW~~MM((*YNH  # # '  rc|jj|j||}|jj |S)aUpdates the specified Enrollment. Args: enrollment_ref: Resource, the Enrollment to update. enrollment_message: Enrollment, the enrollment message that holds enrollment's name, cel_match, message_bus, destination, etc. update_mask: str, a comma-separated list of Enrollment fields to update. Returns: A long-running operation for update. )rr! updateMask)r(0EventarcProjectsLocationsEnrollmentsPatchRequestr4r*Patch)r+r7r8 update_mask patch_reqs rrQzEnrollmentClientV1.PatchrsGOO  ( ( *%PI ==  y ))rc|jj|j}|jj |S)zDeletes the specified Enrollment. Args: enrollment_ref: Resource, the Enrollment to delete. Returns: A long-running operation for delete. r<)r(1EventarcProjectsLocationsEnrollmentsDeleteRequestr4r*Delete)r+r7 delete_reqs rrVzEnrollmentClientV1.DeletesD HH,,. I  ==   ++rc|jj|j|||jnd||j|Sd|S)N)rcelMatch messageBus destinationlabels)r( Enrollmentr4)r+r7 cel_matchmessage_bus_refdestination_refr]s rBuildEnrollmentz"EnrollmentClientV1.BuildEnrollmentsv >> $ $  ( ( *  &#//1   &$002 %   %  rcg}|r|jd|r|jd|r|jd|s tddj|S)a`Builds an update mask for updating a enrollment. Args: cel_match: bool, whether to update the cel_match. destination: bool, whether to update the destination. labels: bool, whether to update the labels. Returns: The update mask as a string. Raises: NoFieldsSpecifiedError: No fields are being updated. rZr\r]z*Must specify at least one field to update.,)appendr join)r+r_r\r]rRs rBuildUpdateMaskz"EnrollmentClientV1.BuildUpdateMasksZK$' "  "#O PP 88K  rcB|jjjS)N)r(r^ LabelsValue)r+s rLabelsValueClassz#EnrollmentClientV1.LabelsValueClasss >> $ $ 0 00r)F)rrrrr#r6r>rMrQrVrbrgrj __classcell__)r-s@rrr(s3N :,, &0*&, !61rrN)r __future__rrrapitools.base.pyrgooglecloudsdk.api_lib.eventarcrrgooglecloudsdk.api_lib.utilr googlecloudsdk.corer r Errorr rEventarcClientBaserrrrrssR.&''02,*)CZ--C X100X1r