o 54g? @sdZddlZddlmZddlmZddlmZddlm Z ddl m Z dd l mZdd lmZmZmZdd d Zddddd ddgid ddf ddZ dddZdddZ dddZdS)z Apple Push Notification Service Documentation is available on the iOS Developer Library: https://developer.apple.com/library/content/documentation/NetworkingInternet/Conceptual/RemoteNotificationsPG/APNSOverview.html N)client) credentials)errors)payload)models) get_manager) APNSErrorAPNSUnsupportedPriorityAPNSServerErrorcCs||dur(t|st|}t|}nt|\}}}|p't|||}tj|t |t |d}| |S)N) use_sandboxuse_alternative_port) rhas_auth_token_credsget_apns_certificateapns2_credentialsCertificateCredentialsget_apns_auth_credsTokenCredentials apns2_client APNsClientget_apns_use_sandboxget_apns_use_alternative_portconnect)credsapplication_idcertkeyPathkeyIdteamIdrr]/var/www/html/notificationserver/venv/lib/python3.10/site-packages/push_notifications/apns.py_apns_create_sockets     r!Fc CsX|s|s| rtj|r |ni|| |d}n|}t|r||}tj||||| | | || d S)N)bodybody_localized_keybody_localized_argsaction_localized_key) alertbadgesoundcategoryurl_argscustom thread_idcontent_availablemutable_content) apns2_payload PayloadAlertcallablePayload)tokenr&rr'r(r)r-action_loc_keyloc_keyloc_argsextrar.r,r* apns2_alertrrr _apns_prepare)s  r9c  st||d}i}dd|d<|dsttd|d<dd}|r?z tt||d<Wn ty>td|wdd|d<|rbfdd|D} |j | t j |d fi|St |fi} |j || t j |d fi|dS) N)rr expirationi'priorityzUnsupported priority %d collapse_idcs(g|]}tj|t|fidqS))r3r)r Notificationr9).0ridr&kwargsrr Ss  z_apns_send..)r)r!popinttimerNotificationPrioritystr ValueErrorr send_notification_batchrget_apns_topicr9send_notification) registration_idr&batchrrrArnotification_kwargsr;datarr@r _apns_send=s:       rPc Ksrzt||f||d|WdStjy8}zt|tjr-tjjj|d}d|_ | t |j j dd}~ww)as Sends an APNS notification to a single registration_id. This will send the notification as form data. If sending multiple notifications, it is more efficient to use apns_send_bulk_message() Note that if set alert should always be a string. If it is not set, it won"t be included in the notification. You will need to pass None to this for silent notifications. )rr)rLF)statusN)rP apns2_errors APNsException isinstance Unregisteredr APNSDeviceobjectsgetactivesaver __class____name__)rLr&rrrAapns2_exceptiondevicerrr apns_send_messageds"   r_cKsHt||fd||d|}dd|D}tjjj|djdd|S)a! Sends an APNS notification to one or more registration_ids. The registration_ids argument needs to be a list. Note that if set alert should always be a string. If it is not set, it won"t be included in the notification. You will need to pass None to this for silent notifications. T)rMrrcSsg|] \}}|dkr|qS)rUr)r>r3resultrrr rBsz*apns_send_bulk_message..)registration_id__inF)rY)rPitemsrrVrWfilterupdate)registration_idsr&rrrAresultsinactive_tokensrrr apns_send_bulk_message~s rh)NN)FNN)__doc__rEapns2rrrrrrRrr/rconfr exceptionsr r r r!r9rPr_rhrrrr s&         '