o ˷eG@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)_utils _messaging) AndroidConfigAndroidFCMOptionsAndroidNotification APNSConfigAPNSFCMOptions APNSPayloadApsApsAlert BatchResponse CriticalSound ErrorInfo FCMOptions LightSettingsMessageMulticastMessage NotificationQuotaExceededErrorSenderIdMismatchError SendResponseThirdPartyAuthErrorTopicManagementResponseUnregisteredError WebpushConfigWebpushFCMOptionsWebpushNotificationWebpushNotificationActionsendsend_allsend_multicastsubscribe_to_topicunsubscribe_from_topiccCst|ttSN)rget_app_service_MESSAGING_ATTRIBUTE_MessagingService)appr.O/var/www/ideatree/venv/lib/python3.10/site-packages/firebase_admin/messaging.py_get_messaging_service_sr0FcCt|||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. )r0r$)messagedry_runr-r.r.r/r$br$cCr1)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. )r0r%)messagesr3r-r.r.r/r%vr4r%cs6tts tdfddjD}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)rr6r7r8r9r:r;).0r<multicast_messager.r/ s z"send_multicast..) isinstancer ValueErrortokensr0r%)r?r3r-r5r.r>r/r&s   r&cCt|||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:batchAddr0make_topic_management_requestrCtopicr-r.r.r/r'r'cCrD)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:batchRemoverErGr.r.r/r(rIr(c@s0eZdZdZddZeddZeddZdS) rzBAn error encountered when performing a topic management operation.cCs||_||_dSr))_index_reason)selfindexreasonr.r.r/__init__s zErrorInfo.__init__cC|jS)zBIndex of the registration token to which this error is related to.)rJrLr.r.r/rMzErrorInfo.indexcCrP)z*String describing the nature of the error.)rKrQr.r.r/rNrRzErrorInfo.reasonN)__name__ __module__ __qualname____doc__rOpropertyrMrNr.r.r.r/rs rc@<eZdZdZddZeddZeddZedd Zd S) rz8The response received from a topic management operation.cCst|tr d|vrtd|d|_d|_g|_t|dD]"\}}d|vr:|jd7_|jt ||dq|jd7_qdS)Nresultsz*Unexpected topic management response: {0}.rerror) rAdictrBformat_success_count_failure_count_errors enumerateappendr)rLresprMresultr.r.r/rOsz TopicManagementResponse.__init__cCrP)zCNumber of tokens that were successfully subscribed or unsubscribed.r^rQr.r.r/ success_countrRz%TopicManagementResponse.success_countcCrP)zLNumber of tokens that could not be subscribed or unsubscribed due to errors.)r_rQr.r.r/ failure_countrRz%TopicManagementResponse.failure_countcCrP)z;A list of ``messaging.ErrorInfo`` objects (possibly empty).)r`rQr.r.r/errorsrRzTopicManagementResponse.errorsN) rSrTrUrVrOrWrfrgrhr.r.r.r/rs  rc@rX) rz:The response received from a batch request to the FCM API.cCs||_tdd|D|_dS)NcSsg|]}|jr|qSr.)success)r=rcr.r.r/r@sz*BatchResponse.__init__..) _responseslenr^)rL responsesr.r.r/rOszBatchResponse.__init__cCrP)z>A list of ``messaging.SendResponse`` objects (possibly empty).)rjrQr.r.r/rl rRzBatchResponse.responsescCrPr)rerQr.r.r/rfszBatchResponse.success_countcCst|j|jSr))rkrlrfrQr.r.r/rgszBatchResponse.failure_countN) rSrTrUrVrOrWrlrfrgr.r.r.r/rs  rc@rX) rzHThe response received from an individual batched request to the FCM API.cCs&||_d|_|r|dd|_dSdS)Nname) _exception _message_idget)rLrc exceptionr.r.r/rOs zSendResponse.__init__cCrP)z9A message ID string that uniquely identifies the message.)rorQr.r.r/ message_id rRzSendResponse.message_idcCs|jduo|j S)z3A boolean indicating if the request was successful.N)rornrQr.r.r/ri%szSendResponse.successcCrP)zTA ``FirebaseError`` if an error occurs while sending the message to the FCM service.)rnrQr.r.r/rq*rRzSendResponse.exceptionN) rSrTrUrVrOrWrrrirqr.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.comaccess_token_authtrue)APNS_AUTH_ERRORQUOTA_EXCEEDEDSENDER_ID_MISMATCHTHIRD_PARTY_AUTH_ERROR UNREGISTEREDcCsn|j}|s tdtj||_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-CLIENT httpTimeout) credentialtimeout) project_idrBr,FCM_URLr]_fcm_urlfirebase_admin __version__ _fcm_headersoptionsrprDEFAULT_TIMEOUT_SECONDSr|get_credential _credentialJsonHttpClient_clientrauthorized_http_build_transport)rLr-r~r}r.r.r/rOAs   z_MessagingService.__init__cCst|ts td|j|S)Nz7Message must be an instance of messaging.Message class.)rArrB JSON_ENCODERdefault)clsr2r.r.r/encode_messageRs  z _MessagingService.encode_messageFc CsV|||}z|jjd|j|j|d}W|dStjjy*}z||d}~ww)z2Sends the given message to FCM via the FCM v1 API.post)urlheadersjsonNrm) _message_datarbodyrrrequests exceptionsRequestException_handle_fcm_error)rLr2r3r6rcrZr.r.r/r$Xs   z_MessagingService.sendc st|ts tdt|dkrtdgfdd}tj|tjd}j }|D]}t  ||}tj |jjd|jd}||q,z |WtStye} z| d } ~ ww) 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_errorrrb)_responserZrq send_responserlrLr.r/batch_callbackps   z2_MessagingService.send_all..batch_callback)callback batch_uriPOST)rpostprocurimethodrrN)rAlistrBrkrBatchHttpRequestr, FCM_BATCH_URLrrrdumpsr HttpRequest _postprocrraddexecute Exceptionrr) rLr5r3rbatch transportr2rreqrZr.rr/r%gs8      z_MessagingService.send_allc Cst|tr|g}t|tr|stddd|D}|r tdt|tr'|s+td|ds5d|}||d}d tj|}z|jj d ||tj d }Wt|St j j yd}z||d }~ww) 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.)rAstr)r=tr.r.r/r@szC_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})toregistration_tokensz{0}/{1}r)rrrN)rArrrB startswithr]r,IID_URLrr IID_HEADERSrrr_handle_iid_errorr) rLrCrH operation invalid_strr6rrcrZr.r.r/rFs6     z/_MessagingService.make_topic_management_requestcCsdt|i}|r d|d<|S)Nr2T validate_only)r,r)rLr2r3r6r.r.r/rsz_MessagingService._message_datacCst|S)z'Handle response from batch API request.)rloadsdecode)rLrrr.r.r/rsz_MessagingService._postproccCt|tjS)z)Handles errors received from the FCM API.)r#handle_platform_error_from_requestsr,_build_fcm_error_requestsrLrZr.r.r/rz#_MessagingService._handle_fcm_errorcCs|jdur t|i}z|j}t|tr|}Wn ty#Ynw|d}d}|r3d|}n d|jj |jj }t||S)z1Handles errors received from the Instance ID API.NrZz(Error while calling the IID service: {0}z4Unexpected HTTP response with status: {0}; body: {1}) rrhandle_requests_errorrrAr\rBrpr] status_codecontentr)rLrZr6 parsed_bodycodemsgr.r.r/rs&        z#_MessagingService._handle_iid_errorcCr)zMHandles errors received from the googleapiclient while making batch requests.)r*handle_platform_error_from_googleapiclientr, _build_fcm_error_googleapiclientrr.r.r/rrz%_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)rrZr2 error_dictexc_typer.r.r/rs z+_MessagingService._build_fcm_error_requestscCs ||}|r||||dSdSr)r)rrZr2rrrr.r.r/rs z2_MessagingService._build_fcm_error_googleapiclientcCsF|sdSd}|dgD]}|ddkr|d}nq tj|S)Ndetailsz@typez3type.googleapis.com/google.firebase.fcm.v1.FcmError errorCode)rpr,FCM_ERROR_TYPES)rrfcm_codedetailr.r.r/rs  z"_MessagingService._build_fcm_errorN)F)rSrTrUrVrrrrrMessageEncoderrrrrrrrO classmethodrr$r%rFrrrrrrrrr.r.r.r/r,0s<   &  r,)FNr))/rVrgoogleapiclientrrrrrrrrrr+__all__r r r r rrrrrrrrrrr r!r"r#rrrrr0r$r%r&r'r(rrrrr,r.r.r.r/s\       $