a |fG@sdZddlZddlmZddlmZddlZddlZddlmZddlmZddlm Z ddlm Z dd lm Z d Z gd Z e jZe jZe jZe jZe jZe jZe jZe jZe jZe jZe jZejZejZe jZe jZe jZe jZe jZe j Z e j!Z!e j"Z"e j#Z#d d Z$d#ddZ%d$ddZ&d%ddZ'd&ddZ(d'ddZ)GdddZ*GdddZ+GdddZ,Gdd d Z-Gd!d"d"Z.dS)(z Firebase Cloud Messaging module.N)http)_auth) _http_client)_messaging_encoder)_messaging_utils) _gapic_utils)_utilsZ _messaging) AndroidConfigAndroidFCMOptionsAndroidNotification APNSConfigAPNSFCMOptions APNSPayloadApsApsAlert BatchResponse CriticalSound ErrorInfo FCMOptions LightSettingsMessageMulticastMessage NotificationQuotaExceededErrorSenderIdMismatchError SendResponseThirdPartyAuthErrorTopicManagementResponseUnregisteredError WebpushConfigWebpushFCMOptionsWebpushNotificationWebpushNotificationActionsendsend_allsend_multicastsubscribe_to_topicunsubscribe_from_topiccCst|ttSN)rZget_app_service_MESSAGING_ATTRIBUTE_MessagingService)appr,RD:\Projects\storyit_web\backend\venv\Lib\site-packages\firebase_admin/messaging.py_get_messaging_service_sr.FcCst|||S)aSends the given message via Firebase Cloud Messaging (FCM). If the ``dry_run`` mode is enabled, the message will not be actually delivered to the recipients. Instead FCM performs all the usual validations, and emulates the send operation. Args: message: An instance of ``messaging.Message``. dry_run: A boolean indicating whether to run the operation in dry run mode (optional). app: An App instance (optional). Returns: string: A message ID string that uniquely identifies the sent message. Raises: FirebaseError: If an error occurs while sending the message to the FCM service. ValueError: If the input arguments are invalid. )r.r#)messagedry_runr+r,r,r-r#bsr#cCst|||S)aSends the given list of messages via Firebase Cloud Messaging as a single batch. If the ``dry_run`` mode is enabled, the message will not be actually delivered to the recipients. Instead FCM performs all the usual validations, and emulates the send operation. Args: messages: A list of ``messaging.Message`` instances. dry_run: A boolean indicating whether to run the operation in dry run mode (optional). app: An App instance (optional). Returns: BatchResponse: A ``messaging.BatchResponse`` instance. Raises: FirebaseError: If an error occurs while sending the message to the FCM service. ValueError: If the input arguments are invalid. )r.r$)messagesr0r+r,r,r-r$vsr$cs6ttstdfddjD}t|||S)aSends the given mutlicast message to all tokens via Firebase Cloud Messaging (FCM). If the ``dry_run`` mode is enabled, the message will not be actually delivered to the recipients. Instead FCM performs all the usual validations, and emulates the send operation. Args: multicast_message: An instance of ``messaging.MulticastMessage``. dry_run: A boolean indicating whether to run the operation in dry run mode (optional). app: An App instance (optional). Returns: BatchResponse: A ``messaging.BatchResponse`` instance. Raises: FirebaseError: If an error occurs while sending the message to the FCM service. ValueError: If the input arguments are invalid. z@Message must be an instance of messaging.MulticastMessage class.c s.g|]&}tjjjjjj|dqS))data notificationandroidwebpushapns fcm_optionstoken)rr2r3r4r5r6r7).0r8multicast_messager,r- sz"send_multicast..) isinstancer ValueErrortokensr.r$)r;r0r+r1r,r:r-r%s    r%cCst|||dS)a4Subscribes a list of registration tokens to an FCM topic. Args: tokens: A non-empty list of device registration tokens. List may not have more than 1000 elements. topic: Name of the topic to subscribe to. May contain the ``/topics/`` prefix. app: An App instance (optional). Returns: TopicManagementResponse: A ``TopicManagementResponse`` instance. Raises: FirebaseError: If an error occurs while communicating with instance ID service. ValueError: If the input arguments are invalid. ziid/v1:batchAddr.make_topic_management_requestr?topicr+r,r,r-r&sr&cCst|||dS)a<Unsubscribes a list of registration tokens from an FCM topic. Args: tokens: A non-empty list of device registration tokens. List may not have more than 1000 elements. topic: Name of the topic to unsubscribe from. May contain the ``/topics/`` prefix. app: An App instance (optional). Returns: TopicManagementResponse: A ``TopicManagementResponse`` instance. Raises: FirebaseError: If an error occurs while communicating with instance ID service. ValueError: If the input arguments are invalid. ziid/v1:batchRemover@rBr,r,r-r'sr'c@s0eZdZdZddZeddZeddZdS) rzBAn error encountered when performing a topic management operation.cCs||_||_dSr()_index_reason)selfindexreasonr,r,r-__init__szErrorInfo.__init__cCs|jS)zBIndex of the registration token to which this error is related to.)rDrFr,r,r-rGszErrorInfo.indexcCs|jS)z*String describing the nature of the error.)rErJr,r,r-rHszErrorInfo.reasonN)__name__ __module__ __qualname____doc__rIpropertyrGrHr,r,r,r-rs  rc@s<eZdZdZddZeddZeddZedd Zd S) rz8The response received from a topic management operation.cCst|trd|vr td|d|_d|_g|_t|dD]D\}}d|vrt|jd7_|jt ||dq>|jd7_q>dS)Nresultsz*Unexpected topic management response: {0}.rerror) r=dictr>format_success_count_failure_count_errors enumerateappendr)rFresprGresultr,r,r-rIsz TopicManagementResponse.__init__cCs|jS)zCNumber of tokens that were successfully subscribed or unsubscribed.rUrJr,r,r- success_countsz%TopicManagementResponse.success_countcCs|jS)zLNumber of tokens that could not be subscribed or unsubscribed due to errors.)rVrJr,r,r- failure_countsz%TopicManagementResponse.failure_countcCs|jS)z;A list of ``messaging.ErrorInfo`` objects (possibly empty).)rWrJr,r,r-errorsszTopicManagementResponse.errorsN) rKrLrMrNrIrOr]r^r_r,r,r,r-rs   rc@s<eZdZdZddZeddZeddZedd Zd S) rz:The response received from a batch request to the FCM API.cCs||_tdd|D|_dS)NcSsg|]}|jr|qSr,)success)r9rZr,r,r-r<z*BatchResponse.__init__..) _responseslenrU)rF responsesr,r,r-rIszBatchResponse.__init__cCs|jS)z>A list of ``messaging.SendResponse`` objects (possibly empty).)rbrJr,r,r-rd szBatchResponse.responsescCs|jSr(r\rJr,r,r-r]szBatchResponse.success_countcCst|j|jSr()rcrdr]rJr,r,r-r^szBatchResponse.failure_countN) rKrLrMrNrIrOrdr]r^r,r,r,r-rs  rc@s<eZdZdZddZeddZeddZedd Zd S) rzHThe response received from an individual batched request to the FCM API.cCs"||_d|_|r|dd|_dS)Nname) _exception _message_idget)rFrZ exceptionr,r,r-rIszSendResponse.__init__cCs|jS)z9A message ID string that uniquely identifies the message.)rgrJr,r,r- message_id szSendResponse.message_idcCs|jduo|j S)z3A boolean indicating if the request was successful.N)rgrfrJr,r,r-r`%szSendResponse.successcCs|jS)zTA ``FirebaseError`` if an error occurs while sending the message to the FCM service.)rfrJr,r,r-ri*szSendResponse.exceptionN) rKrLrMrNrIrOrjr`rir,r,r,r-rs  rc@seZdZdZdZdZdZddiZe Z e e e e edZdd Zed d Zd$d dZd%ddZddZddZddZddZddZddZeddZedd Zed!d"Zd#S)&r*zKService class that implements Firebase Cloud Messaging (FCM) functionality.z8https://fcm.googleapis.com/v1/projects/{0}/messages:sendz https://fcm.googleapis.com/batchzhttps://iid.googleapis.comZaccess_token_authtrue)ZAPNS_AUTH_ERRORZQUOTA_EXCEEDEDZSENDER_ID_MISMATCHZTHIRD_PARTY_AUTH_ERRORZ UNREGISTEREDcCsn|j}|stdtj||_ddtjd|_|j dt j }|j |_t j|j|d|_tj|_dS)NzProject ID is required to access Cloud Messaging service. Either set the projectId option, or use service account credentials. Alternatively, set the GOOGLE_CLOUD_PROJECT environment variable.2zfire-admin-python/{0})zX-GOOG-API-FORMAT-VERSIONzX-FIREBASE-CLIENTZ httpTimeout) credentialtimeout) project_idr>r*FCM_URLrT_fcm_urlfirebase_admin __version__ _fcm_headersoptionsrhrZDEFAULT_TIMEOUT_SECONDSrmget_credential _credentialZJsonHttpClient_clientrZauthorized_http_build_transport)rFr+rornr,r,r-rIAs  z_MessagingService.__init__cCst|tstd|j|S)Nz7Message must be an instance of messaging.Message class.)r=rr> JSON_ENCODERdefault)clsr/r,r,r-encode_messageRs z _MessagingService.encode_messageFc Csh|||}z|jjd|j|j|d}Wn2tjjyZ}z||WYd}~nd}~00|dSdS)z2Sends the given message to FCM via the FCM v1 API.post)urlheadersjsonNre) _message_datarxbodyrqrtrequests exceptionsRequestException_handle_fcm_error)rFr/r0r2rZrQr,r,r-r#Xs   z_MessagingService.sendc st|tstdt|dkr&tdgfdd}tj|tjd}j }|D]<}t  ||}tj |jjd|jd}||qXz |Wn.ty} z| WYd } ~ nd } ~ 00tSd S) z2Sends the given messages to FCM via the batch API.z7messages must be a list of messaging.Message instances.iz1messages must not contain more than 500 elements.cs*d}|r|}t||}|dSr()_handle_batch_errorrrY)_responserQriZ send_responserdrFr,r-batch_callbackps   z2_MessagingService.send_all..batch_callback)callbackZ batch_uriPOST)rZpostprocurimethodrrN)r=listr>rcrZBatchHttpRequestr* FCM_BATCH_URLryrwrdumpsrZ HttpRequest _postprocrqrtaddexecute Exceptionrr) rFr1r0rbatch transportr/rreqrQr,rr-r$gs4      z_MessagingService.send_allc Cst|tr|g}t|tr|s&tddd|D}|r@tdt|trN|sVtd|dsjd|}||d}d tj|}z|jj d ||tj d }Wn2t j j y}z||WYd }~nd }~00t|Sd S) z;Invokes the IID service for topic management functionality.z7Tokens must be a string or a non-empty list of strings.cSsg|]}t|tr|s|qSr,)r=str)r9tr,r,r-r<razC_MessagingService.make_topic_management_request..z!Tokens must be non-empty strings.z!Topic must be a non-empty string.z/topics/z /topics/{0})toZregistration_tokensz{0}/{1}r~)rrrN)r=rrr> startswithrTr*IID_URLrxr IID_HEADERSrrr_handle_iid_errorr) rFr?rCZ operationZ invalid_strr2rrZrQr,r,r-rAs2     z/_MessagingService.make_topic_management_requestcCsdt|i}|rd|d<|S)Nr/TZ validate_only)r*r})rFr/r0r2r,r,r-rsz_MessagingService._message_datacCst|S)z'Handle response from batch API request.)rloadsdecode)rFrrr,r,r-rsz_MessagingService._postproccCst|tjS)z)Handles errors received from the FCM API.)rZ#handle_platform_error_from_requestsr*_build_fcm_error_requestsrFrQr,r,r-rsz#_MessagingService._handle_fcm_errorcCs|jdurt|i}z|j}t|tr2|}WntyFYn0|d}d}|rfd|}nd|jj |jj }t||S)z1Handles errors received from the Instance ID API.NrQz(Error while calling the IID service: {0}z4Unexpected HTTP response with status: {0}; body: {1}) rrZhandle_requests_errorrr=rSr>rhrT status_codecontentr)rFrQr2Z parsed_bodycodemsgr,r,r-rs"       z#_MessagingService._handle_iid_errorcCst|tjS)zMHandles errors received from the googleapiclient while making batch requests.)rZ*handle_platform_error_from_googleapiclientr* _build_fcm_error_googleapiclientrr,r,r-rsz%_MessagingService._handle_batch_errorcCs"||}|r||||jdSdSzfParses an error response from the FCM API and creates a FCM-specific exception if appropriate.)cause http_responseN)_build_fcm_errorr)r|rQr/ error_dictexc_typer,r,r-rs z+_MessagingService._build_fcm_error_requestscCs ||}|r||||dSdSr)r)r|rQr/rrrr,r,r-rs z2_MessagingService._build_fcm_error_googleapiclientcCsF|sdSd}|dgD] }|ddkr|d}q:qtj|S)Ndetailsz@typez3type.googleapis.com/google.firebase.fcm.v1.FcmErrorZ errorCode)rhr*FCM_ERROR_TYPES)r|rZfcm_codedetailr,r,r-rs z"_MessagingService._build_fcm_errorN)F)F)rKrLrMrNrprrrrZMessageEncoderrzrrrrrrI classmethodr}r#r$rArrrrrrrrr,r,r,r-r*0s:   &  r*)FN)FN)FN)N)N)/rNrZgoogleapiclientrrrrrrrrrrr)__all__r r r r r rrrrrrrrrrr r!r"rrrrr.r#r$r%r&r'rrrrr*r,r,r,r-sZ       $