U k7gR)@sddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl Z ddl mZmZmZmZmZmZGdddeZdS)N) HTTPAdapter)Retry)service_account)AuthenticationErrorInvalidDataErrorFCMErrorFCMSenderIdMismatchErrorFCMServerErrorFCMNotRegisteredErrorc @sjeZdZdZdeedddZeddZddd Zd d Z d d Z ddZ ddZ ddZ dddZdS)BaseAPIz&https://fcm.googleapis.com/v1/projectsN)service_account_file project_idc Cs||_||_|jd|jd|_d|_||_t|_||_|sN|sNt d|rt |t rd|ksld|kr||_|j j ||dkrzdd lm}|Wntk rYnX||_dS) a Override existing init function to give ability to use v1 endpoints of Firebase Cloud Messaging API Attributes: service_account_file (str): path to service account JSON file project_id (str): project ID of Google account credentials (Credentials): Google oauth2 credentials instance, such as ADC proxy_dict (dict): proxy settings dictionary, use proxy (keys: `http`, `https`) env (dict): environment settings dictionary, for example "app_engine" json_encoder (BaseJSONEncoder): JSON encoder adapter (BaseAdapter): adapter instance /z/messages:sendNzLPlease provide a service account file path or credentials in the constructorhttphttpsZ app_enginer) appengine)r r FCM_END_POINTZFCM_REQ_PROXIEScustom_adapter threadinglocal thread_local credentialsr isinstancedictrequests_sessionproxiesupdateZrequests_toolbelt.adaptersrZ monkeypatchModuleNotFoundError json_encoder) selfr r rZ proxy_dictenvradapterrr"1/tmp/pip-unpacked-wheel-vxwmpmkg/pyfcm/baseapi.py__init__s8   zBaseAPI.__init__cCst|jdddkrttdddgtjtdgBd}|jp>t|d}t|j_ |jj d||jj d |d |j_ t }|jj |kr|jj j ||d |j_ |jj S) NriiPOST)backoff_factorstatus_forcelistZallowed_methods) max_retrieszhttp://zhttps://ri)getattrrrZDEFAULT_ALLOWED_METHODS frozensetrrrequestsSessionrmountZ token_expirytimeheadersrrequest_headers)rretriesr!Zcurrent_timestampr"r"r#rRs    zBaseAPI.requests_sessioncCsX|jj|j||d}d|jkrTt|jddkrTt|jd}t||||S|S)N)datatimeoutz Retry-Afterr)rpostrr0intr/sleep send_request)rpayloadr4responseZ sleep_timer"r"r#r8fs  zBaseAPI.send_requestcsJddl}ddlm}fdd|D}||j||d}|S)Nrr%) fetch_taskscsg|]}jf|qSr") parse_payload).0paramsrr"r# xsz.BaseAPI.send_async_request..)Z end_pointr0payloadsr4)asyncioZ async_fcmr;Znew_event_loopZrun_until_completerr1)rZ params_listr4rBr;rA responsesr"r?r#send_async_requestss  zBaseAPI.send_async_requestc CsrzB|jrtjj|jdgd}n|j}tjjj }| ||j WSt k rl}z t |W5d}~XYnXdS)z Generates access token from credentials. If token expires then new access token is generated. Returns: str: Access token z2https://www.googleapis.com/auth/firebase.messaging)ZscopesN)r rZ CredentialsZfrom_service_account_filergoogleauth transportr,RequestZrefreshtoken Exceptionr)rrrequester"r"r#_get_access_tokens zBaseAPI._get_access_tokencCsdd|dS)z Generates request headers including Content-Type and Authorization of Bearer token Returns: dict: request headers zapplication/jsonzBearer )z Content-Type Authorization)rMr?r"r"r#r1s zBaseAPI.request_headerscCstj|dd|jdddS)z Standardized json.dumps function with separators and sorted keys set Args: data (dict or list): data to be dumped Returns: string: json ),:TF) separators sort_keyscls ensure_asciiutf8)jsondumpsrencode)rr3r"r"r# json_dumpss zBaseAPI.json_dumpscCs|jdkr:d|jkr0t|jddkr0tdq|Snd|jdkrNtdnP|jdkrdt|jn:|jdkrxtd n&|jd krt d ntd |jd dS)a Parses the json response sent back by the server and tries to get out the important return variables Returns: dict: name (str) - The identifier of the message sent, in the format of projects/*/messages/{message_id} Raises: FCMServerError: FCM is temporary not available AuthenticationError: error authenticating the sender account InvalidDataError: data passed to FCM was incorrecly structured FCMSenderIdMismatchError: the authenticated sender is different from the sender registered to the token FCMNotRegisteredError: device token is missing, not registered, or invalid zcontent-lengthrz2FCM server connection error, the response is emptyiz4There was an error authenticating the sender accountiizWThe authenticated sender ID is different from the sender ID for the registration token.izToken not registeredz)FCM server error: Unexpected status code z.. The server might be temporarily unavailable.N) status_coder0r6r rVrrtextrr )rr:r"r"r#parse_responses0         zBaseAPI.parse_responseFc Cs,t} |r|| d<|r|| d<|r*|| d<|rJt|trB|| d<ntd|rjt|trb|| d<ntd| rt| tr| | d<ntd | rt| tr| | d <ntd | rt| tr| | d <ntd i| d<|r|| dd<|r|| dd<|r|| dd<|r|s|s| d=|| | dS)z :rtype: json rIZtopic conditionr3z,Provided data_payload is in the wrong formatZandroidz.Provided android_config is in the wrong formatZwebpushz.Provided webpush_config is in the wrong formatZapnsz+Provided apns_config is in the wrong format fcm_optionsz+Provided fcm_options is in the wrong formatZ notificationtitlebodyimage)messageZ validate_only)rrrrY)rZ fcm_tokenZnotification_titleZnotification_bodyZnotification_imageZ data_payloadZ topic_nameZtopic_conditionZandroid_configZ apns_configZwebpush_configr_dry_runZ fcm_payloadr"r"r#r<sL             zBaseAPI.parse_payload)NNNNN)NN) NNNNNNNNNNNF)__name__ __module__ __qualname__rstrr$propertyrr8rDrMr1rYr]r<r"r"r"r#r s> 4   -r )rVosr/rr,Zrequests.adaptersrurllib3rZ google.oauth2rZgoogle.auth.transport.requestsrEZ pyfcm.errorsrrrrr r objectr r"r"r"r#s