a !f4K@sdZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddl m Z dd l mZdd l mZdd lmZdd dZGdddeZGdddeZddZddZddZddZd S)z$GAX wrapper for Pubsub API requests.) CallOptions) INITIAL_PAGE)GaxError) exc_to_code) PubsubMessage) PushConfig) StatusCode)Conflict)NotFound) _to_bytesNcCs"|dur t}d|i}tfi|S)z6Helper for :meth:'_PublisherAPI.list_topics' et aliae.N page_token)rr)r optionsrS/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/pubsub/_gax.py_build_paging_options src@sLeZdZdZddZdddZdd Zd d Zd d ZddZ dddZ dS) _PublisherAPIzHelper mapping publisher-related APIs. :type gax_api: :class:`google.pubsub.v1.publisher_api.PublisherApi` :param gax_api: API object used to make GAX requests. cCs ||_dSN_gax_apiselfZgax_apirrr__init__.sz_PublisherAPI.__init__rNc CsHt|}d|f}|jj|||d}dd|D}|jp>d}||fS)aList topics for the project associated with this API. See: https://cloud.google.com/pubsub/reference/rest/v1/projects.topics/list :type project: string :param project: project ID :type page_size: int :param page_size: maximum number of topics to return, If not passed, defaults to a value set by the API. :type page_token: string :param page_token: opaque marker for the next "page" of topics. If not passed, the API will return the first page of topics. :rtype: tuple, (list, str) :returns: list of ``Topic`` resource dicts, plus a "next page token" string: if not None, indicates that more topics can be retrieved with another call (pass that value as ``page_token``). projects/%s page_sizer cSsg|]}d|jiqSnamer).0topic_pbrrr Mz-_PublisherAPI.list_topics..N)rr list_topicsnextr ) rprojectrr r path page_iterZtopicstokenrrrr!1s  z_PublisherAPI.list_topicsc CsZz|j|}Wn>tyN}z&t|jtjkr8t|WYd}~n d}~00d|jiS)aAPI call: create a topic See: https://cloud.google.com/pubsub/reference/rest/v1/projects.topics/create :type topic_path: string :param topic_path: fully-qualified path of the new topic, in format ``projects//topics/``. :rtype: dict :returns: ``Topic`` resource returned from the API. :raises: :exc:`gcloud.exceptions.Conflict` if the topic already exists Nr) rZ create_topicrrcauserFAILED_PRECONDITIONr rr topic_pathrexcrrr topic_createQsz_PublisherAPI.topic_createc CsZz|j|}Wn>tyN}z&t|jtjkr8t|WYd}~n d}~00d|jiS)aAPI call: retrieve a topic See: https://cloud.google.com/pubsub/reference/rest/v1/projects.topics/get :type topic_path: string :param topic_path: fully-qualified path of the topic, in format ``projects//topics/``. :rtype: dict :returns: ``Topic`` resource returned from the API. :raises: :exc:`gcloud.exceptions.NotFound` if the topic does not exist Nr) rZ get_topicrrr'r NOT_FOUNDr rr)rrr topic_gethsz_PublisherAPI.topic_getc CsTz|j|Wn>tyN}z&t|jtjkr8t|WYd}~n d}~00dS)a7API call: delete a topic See: https://cloud.google.com/pubsub/reference/rest/v1/projects.topics/create :type topic_path: string :param topic_path: fully-qualified path of the new topic, in format ``projects//topics/``. N)rZ delete_topicrrr'rr-r )rr*r+rrr topic_deletes z_PublisherAPI.topic_deletec Csttdd}dd|D}z|jj|||d}Wn>tyl}z&t|jtjkrVt|WYd}~n d}~00|j S)afAPI call: publish one or more messages to a topic See: https://cloud.google.com/pubsub/reference/rest/v1/projects.topics/publish :type topic_path: string :param topic_path: fully-qualified path of the topic, in format ``projects//topics/``. :type messages: list of dict :param messages: messages to be published. :rtype: list of string :returns: list of opaque IDs for published messages. :raises: :exc:`gcloud.exceptions.NotFound` if the topic does not exist F)Z is_bundlingcSsg|] }t|qSr)_message_pb_from_dict)rmessagerrrrsz/_PublisherAPI.topic_publish..)r N) rrpublishrrr'rr-r Z message_ids)rr*messagesr Z message_pbsresultr+rrr topic_publishs   z_PublisherAPI.topic_publishc Csxt|}z|jj|||d}Wn>ty\}z&t|jtjkrFt|WYd}~n d}~00| }|j pnd}||fS)aAPI call: list subscriptions bound to a topic See: https://cloud.google.com/pubsub/reference/rest/v1/projects.topics.subscriptions/list :type topic_path: string :param topic_path: fully-qualified path of the topic, in format ``projects//topics/``. :type page_size: int :param page_size: maximum number of subscriptions to return, If not passed, defaults to a value set by the API. :type page_token: string :param page_token: opaque marker for the next "page" of subscriptions. If not passed, the API will return the first page of subscriptions. :rtype: list of strings :returns: fully-qualified names of subscriptions for the supplied topic. :raises: :exc:`gcloud.exceptions.NotFound` if the topic does not exist rN) rrZlist_topic_subscriptionsrrr'rr-r r"r ) rr*rr r r%r+subsr&rrrtopic_list_subscriptionss  z&_PublisherAPI.topic_list_subscriptions)rN)rN) __name__ __module__ __qualname____doc__rr!r,r.r/r5r7rrrrr(s rc@s^eZdZdZddZdddZddd Zd d Zd d ZddZ dddZ ddZ ddZ dS)_SubscriberAPIzHelper mapping subscriber-related APIs. :type gax_api: :class:`google.pubsub.v1.publisher_api.SubscriberApi` :param gax_api: API object used to make GAX requests. cCs ||_dSrrrrrrrsz_SubscriberAPI.__init__rNc CsHt|}d|f}|jj|||d}dd|D}|jp>d}||fS)aList subscriptions for the project associated with this API. See: https://cloud.google.com/pubsub/reference/rest/v1/projects.subscriptions/list :type project: string :param project: project ID :type page_size: int :param page_size: maximum number of subscriptions to return, If not passed, defaults to a value set by the API. :type page_token: string :param page_token: opaque marker for the next "page" of subscriptions. If not passed, the API will return the first page of subscriptions. :rtype: tuple, (list, str) :returns: list of ``Subscription`` resource dicts, plus a "next page token" string: if not None, indicates that more topics can be retrieved with another call (pass that value as ``page_token``). rrcSsg|] }t|qSr)_subscription_pb_to_mapping)rsub_pbrrrrsz5_SubscriberAPI.list_subscriptions..N)rrlist_subscriptionsr"r ) rr#rr r r$r%Z subscriptionsr&rrrr?s  z!_SubscriberAPI.list_subscriptionsc Cs|durt|d}nd}|dur$d}z|j||||}Wn>tyx}z&t|jtjkrbt|WYd}~n d}~00t |S)aAPI call: create a subscription See: https://cloud.google.com/pubsub/reference/rest/v1/projects.subscriptions/create :type subscription_path: string :param subscription_path: the fully-qualified path of the new subscription, in format ``projects//subscriptions/``. :type topic_path: string :param topic_path: the fully-qualified path of the topic being subscribed, in format ``projects//topics/``. :type ack_deadline: int, or ``NoneType`` :param ack_deadline: the deadline (in seconds) by which messages pulled from the back-end must be acknowledged. :type push_endpoint: string, or ``NoneType`` :param push_endpoint: URL to which messages will be pushed by the back-end. If not set, the application must pull messages. :rtype: dict :returns: ``Subscription`` resource returned from the API. N push_endpointr) rrZcreate_subscriptionrrr'rr(r r=)rsubscription_pathr* ack_deadlinerA push_configr>r+rrrsubscription_creates z"_SubscriberAPI.subscription_createc CsXz|j|}Wn>tyN}z&t|jtjkr8t|WYd}~n d}~00t|S)aAPI call: retrieve a subscription See: https://cloud.google.com/pubsub/reference/rest/v1/projects.subscriptions/get :type subscription_path: string :param subscription_path: the fully-qualified path of the subscription, in format ``projects//subscriptions/``. :rtype: dict :returns: ``Subscription`` resource returned from the API. N) rZget_subscriptionrrr'rr-r r=)rrBr>r+rrrsubscription_get-sz_SubscriberAPI.subscription_getc CsTz|j|Wn>tyN}z&t|jtjkr8t|WYd}~n d}~00dS)aAPI call: delete a subscription See: https://cloud.google.com/pubsub/reference/rest/v1/projects.subscriptions/delete :type subscription_path: string :param subscription_path: the fully-qualified path of the subscription, in format ``projects//subscriptions/``. N)rZdelete_subscriptionrrr'rr-r )rrBr+rrrsubscription_deleteCs z"_SubscriberAPI.subscription_deletec Cs`t|d}z|j||Wn>tyZ}z&t|jtjkrDt|WYd}~n d}~00dS)aAPI call: update push config of a subscription See: https://cloud.google.com/pubsub/reference/rest/v1/projects.subscriptions/modifyPushConfig :type subscription_path: string :param subscription_path: the fully-qualified path of the new subscription, in format ``projects//subscriptions/``. :type push_endpoint: string, or ``NoneType`` :param push_endpoint: URL to which messages will be pushed by the back-end. If not set, the application must pull messages. r@N) rrZmodify_push_configrrr'rr-r )rrBrArDr+rrrsubscription_modify_push_configUs z._SubscriberAPI.subscription_modify_push_configFc Csdz|j|||}Wn>tyR}z&t|jtjkr/subscriptions/``. :type return_immediately: boolean :param return_immediately: if True, the back-end returns even if no messages are available; if False, the API call blocks until one or more messages are available. :type max_messages: int :param max_messages: the maximum number of messages to return. :rtype: list of dict :returns: the ``receivedMessages`` element of the response. NcSsg|] }t|qSr)_received_message_pb_to_mapping)rZrmpbrrrrsz4_SubscriberAPI.subscription_pull..) rpullrrr'rr-r Zreceived_messages)rrBZreturn_immediatelyZ max_messagesZ response_pbr+rrrsubscription_pullnsz _SubscriberAPI.subscription_pullc CsVz|j||Wn>tyP}z&t|jtjkr:t|WYd}~n d}~00dS)aAPI call: acknowledge retrieved messages See: https://cloud.google.com/pubsub/reference/rest/v1/projects.subscriptions/modifyPushConfig :type subscription_path: string :param subscription_path: the fully-qualified path of the new subscription, in format ``projects//subscriptions/``. :type ack_ids: list of string :param ack_ids: ack IDs of messages being acknowledged N)rZ acknowledgerrr'rr-r )rrBack_idsr+rrrsubscription_acknowledges z'_SubscriberAPI.subscription_acknowledgec CsXz|j|||Wn>tyR}z&t|jtjkr/subscriptions/``. :type ack_ids: list of string :param ack_ids: ack IDs of messages being acknowledged :type ack_deadline: int :param ack_deadline: the deadline (in seconds) by which messages pulled from the back-end must be acknowledged. N)rZmodify_ack_deadlinerrr'rr-r )rrBrMrCr+rrr subscription_modify_ack_deadlinesz/_SubscriberAPI.subscription_modify_ack_deadline)rN)NN)FrI) r8r9r:r;rr?rErFrGrHrLrNrOrrrrr<s " . "r<cCstt|d|ddS)z/Helper for :meth:`_PublisherAPI.topic_publish`.data attributes)rPrQ)rr )r1rrrr0s r0cCs2|j|j|jd}|jjdkr.d|jji|d<|S)zHelper for :meth:`list_subscriptions`, et aliae Ideally, would use a function from :mod:`protobuf.json_format`, but the right one isn't public. See: https://github.com/google/protobuf/issues/1351 )rtopicZackDeadlineSecondsZ pushEndpointZ pushConfig)rrRZack_deadline_secondsrDrA)r>mappingrrrr=s r=cCs|j|j|jdS)Helper for :meth:`pull`, et aliae Ideally, would use a function from :mod:`protobuf.json_format`, but the right one isn't public. See: https://github.com/google/protobuf/issues/1351 )Z messageIdrPrQ)Z message_idrPrQ)Z message_pbrrr_message_pb_to_mappingsrVcCs|jt|jdS)rU)ZackIdr1)Zack_idrVr1)Zreceived_message_pbrrrrJs rJ)N)r;Z google.gaxrrZgoogle.gax.errorsrZgoogle.gax.grpcrZgoogle.pubsub.v1.pubsub_pb2rrZgrpc.beta.interfacesrZgcloud.exceptionsr r Zgcloud._helpersr robjectrr<r0r=rVrJrrrrs&           .m