dZddlmZddlmZddlmZddlmZddlmZddl m Z ddZ dd Z Gd d e jZGd deZGddeZy )z*Utilities for Cloud Pub/Sub Snapshots API.)absolute_import)division)unicode_literals) list_pager)apis) exceptionsc2tjdd|S)Npubsubv1no_http)rGetClientInstancer s .lib/googlecloudsdk/api_lib/pubsub/snapshots.pyrrs  $ @@Nc6|xs t}|jSN)rMESSAGES_MODULE)clients rGetMessagesModuler s  (&(&   rceZdZdZy)NoFieldsSpecifiedErrorz:Error when no fields were specified for a Patch operation.N)__name__ __module__ __qualname____doc__rrrr%sBrrceZdZdZdZy)_SnapshotUpdateSettingz-Data container class for updating a snapshot.c ||_||_yr) field_namevalue)selfr r!s r__init__z_SnapshotUpdateSetting.__init__,s DODJrN)rrrrr#rrrrr)s 5rrc<eZdZdZd dZd dZdZdZd dZd dZ y) SnapshotsClientz6Client for snapshots service in the Cloud Pub/Sub API.Nc|xs t|_|xs t||_|jj|_yr)rrrmessagesprojects_snapshots_service)r"rr's rr#zSnapshotsClient.__init__4s5/-/DK9 1& 9DMKK22DMrc|jj|jj|j|||j}|jj |S)zCreates a Snapshot.) subscriptionlabelstags)createSnapshotRequestname)r'$PubsubProjectsSnapshotsCreateRequestCreateSnapshotRequest RelativeNamer)Create)r" snapshot_refsubscription_refr,r- create_reqs rr3zSnapshotsClient.Create9skCC"mmAA)668B & & ( D*J ==   ++rc|jj|j}|jj |S)zGets a Snapshot. Args: snapshot_ref (Resource): Resource reference to the Snapshot to get. Returns: Snapshot: The snapshot. snapshot)r'!PubsubProjectsSnapshotsGetRequestr2r)Get)r"r4get_reqs rr;zSnapshotsClient.GetCs>mm==**,>.G ==  W %%rc|jj|j}|jj |S)zDeletes a Snapshot.r8)r'$PubsubProjectsSnapshotsDeleteRequestr2r)Delete)r"r4 delete_reqs rr?zSnapshotsClient.DeleteOs=CC**,D.J ==   ++rc|jj|j|}tj|j ||ddS)aELists Snapshots for a given project. Args: project_ref (Resource): Resource reference to Project to list Snapshots from. page_size (int): the number of entries in each batch (affects requests made, but not the yielded results). Returns: A generator of Snapshots in the Project. )projectpageSize snapshotsrC) batch_sizefieldbatch_size_attribute)r'"PubsubProjectsSnapshotsListRequestr2r YieldFromListr))r" project_ref page_sizelist_reqs rListzSnapshotsClient.ListUsR}}??((*@H  # # xI  <445*  "#O PP AA"mmAAxx ,B. & & ( B*I ==  y ))r)NN)dr) rrrrr#r3r;r?rMrXrrrr%r%1s#>3 , &, <&*rr%)Fr)r __future__rrrapitools.base.pyrgooglecloudsdk.api_lib.utilrgooglecloudsdk.corerrrErrorrobjectrr%rrrrdsU 1&'',*A CZ--CVS*fS*r