o l~i8@s.ddlmZddlmZddlmZmZddlZddlmZddl m Z ddl m Z ddlmZmZdd lmZd ZGd d d ejZGd ddejZe je jegZddeDZdeedefddZGdddeZGdddej j!Z"e#e"Z$Gdddej%Z&GdddeZ'Gddde'Z(dS))Sequence)copy) NamedTupleUnionN)models) gettext_lazy) messaging) FirebaseErrorInvalidArgumentError)FCM_DJANGO_SETTINGSic@seZdZejddddZejdeddddZej eddedd Z ej e d ddej ed d Zejed dddZGdddZddZdS)DeviceIDT) verbose_name primary_key auto_createdName) max_lengthrblanknullz Is activez/Inactive devices will not be sent notifications)rdefault help_text USER_MODEL fcmdevice)rr on_deleterelated_query_namez Creation date)r auto_now_addrc@seZdZdZdS)z Device.MetaTN)__name__ __module__ __qualname__abstractr!r!X/var/www/newdalilibackend/backend/venv/lib/python3.10/site-packages/fcm_django/models.pyMeta+sr#cCs,|jpt|dp dp|jjd|jpdS)N device_idz for z unknown user)namegetattr __class__ruserselfr!r!r"__str__.s  zDevice.__str__N)rrrr AutoFieldid CharField_r& BooleanFieldactive ForeignKeySETTINGSCASCADEr) DateTimeField date_createdr#r,r!r!r!r"r s4   r c@seZdZddZdS)_FCMDeviceManagercCs t|jSN)FCMDeviceQuerySetmodelr*r!r!r" get_queryset7s z_FCMDeviceManager.get_querysetN)rrrr<r!r!r!r"r86s r8cCsg|]}|jqSr!)code.0xr!r!r" BsrAexcreturncCs:|sdSt|}|tkr|tvS|tko|jdkp|tvS)NFzInvalid registration)typestrfcm_error_list_strr causefcm_error_list)rBexc_typer!r!r"$_validate_exception_for_deactivationEsrJc@s:eZdZUeejejfed<ee ed<ee ed<dS)FirebaseResponseDictresponseregistration_ids_sentdeactivated_registration_idsN) rrrrr BatchResponseTopicManagementResponse__annotations__listrEr!r!r!r"rKPs  rKc @seZdZedejdefddZedefddZ  dd e d e ede efd d Z dd edfdejd e d e edddef ddZde ede eejejfde efddZde efddZedefddZdd edfde ded e d e edddef ddZd S) r:messagetokencCs||_t|Sr9)rTr)rSrTr!r!r"_prepare_messageYsz"FCMDeviceQuerySet._prepare_messagerCcCsttgggdS)NrLrMrN)rKrrOr!r!r!r"!get_default_send_message_response^s z3FCMDeviceQuerySet.get_default_send_message_responseFNskip_registration_id_lookupadditional_registration_idscCs4|rt|ng}|s||jddjddd|S)a{ Uses the current filtering/QuerySet chain to get registration IDs :param skip_registration_id_lookup: skips the QuerySet lookup and solely uses the list of IDs from additional_registration_ids :param additional_registration_ids: specific registration_ids to add to the QuerySet lookup :returns a list of registration IDs Tr2registration_id)flat)rRextendfilter values_list)r+rXrYregistration_idsr!r!r"get_registration_idsfsz&FCMDeviceQuerySet.get_registration_idsDEFAULT_FIREBASE_APPappfirebase_admin.Appc  s||}|s Sg}tdt|tD]!}fdd|||tD} |tj| fd|i|jqt t || ||dS)a7 Send notification of single message for all active devices in queryset and deactivate if DELETE_INACTIVE_DEVICES setting is set to True. Bulk sends using firebase.messaging.send_each. For every 500 messages, we send a single HTTP request to Firebase (the 500 is set by the firebase-sdk). :param message: firebase.messaging.Message. If `message` includes a token/id, it will be overridden. :param skip_registration_id_lookup: skips the QuerySet lookup and solely uses the list of IDs from additional_registration_ids :param additional_registration_ids: specific registration_ids to add to the :param app: firebase_admin.App. Specify a specific app to use QuerySet lookup :param more_send_message_kwargs: Parameters for firebase.messaging.send_each() - dry_run: bool. Whether to actually send the notification to the device If there are any new parameters, you can still specify them here. :raises FirebaseError :returns FirebaseResponseDict rcsg|]}|qSr!)rU)r?rTrSr+r!r"rAs z2FCMDeviceQuerySet.send_message..rcrV) rarWrangelenMAX_MESSAGES_PER_BATCHr]r send_each responsesrKrO%deactivate_devices_with_error_results) r+rSrXrYrcmore_send_message_kwargsr`rjimessagesr!rer" send_message}s8 zFCMDeviceQuerySet.send_messager`resultscsb|sgSt|dtjrddt|D}n fdd|D}|j|djdd|||S)NrcSsg|] \}}t|jr|qSr!)rJ exception)r?itemrTr!r!r"rAs zKFCMDeviceQuerySet.deactivate_devices_with_error_results..cs g|] }t|jr|jqSr!)rJreasonindexr>r`r!r"rAs registration_id__inFrZ) isinstancer SendResponsezipr^update%_delete_inactive_devices_if_requested)r+r`rpdeactivated_idsr!rur"rks  z7FCMDeviceQuerySet.deactivate_devices_with_error_resultscCs tdr|j|ddSdS)NDELETE_INACTIVE_DEVICESrv)r4r^delete)r+r`r!r!r"r|sz7FCMDeviceQuerySet._delete_inactive_devices_if_requestedcCsttdgiggdS)NrprV)rKrrPr!r!r!r"get_default_topic_responses  z,FCMDeviceQuerySet.get_default_topic_responseshould_subscribetopicc KsR|||}|s |S|rtjntj||fd|i|}t|||||jdS)a Subscribes or Unsubscribes filtered and/or given tokens/registration_ids to given topic. :param should_subscribe: whether to have these users subscribe (True) or unsubscribe to a topic (False). :param topic: Name of the topic to subscribe to. May contain the ``/topics/`` prefix. :param skip_registration_id_lookup: skips the QuerySet lookup and solely uses the list of IDs from additional_registration_ids :param additional_registration_ids: specific registration_ids to add to the :param app: firebase_admin.App. Specify a specific app to use QuerySet lookup :param more_subscribe_kwargs: Parameters for ``firebase.messaging.subscribe_to_topic()`` If there are any new parameters, you can still specify them here. :raises FirebaseError :returns FirebaseResponseDict rcrV)rarrsubscribe_to_topicunsubscribe_from_topicrKrkerrors) r+rrrXrYrcmore_subscribe_kwargsr`rLr!r!r"handle_topic_subscriptions,z+FCMDeviceQuerySet.handle_topic_subscription)FN)rrr staticmethodrMessagerErUrKrWboolrrRrar4rorry ErrorInforkr|rrr!r!r!r"r:Xsl   5  r:c@seZdZdZdZdZdS) DeviceType)iosr)androidr)webrN)rrrIOSANDROIDWEBr!r!r!r"r src @seZdZUejeddddedddZejeded dZ eje j d d Z e Zd ed <Gd ddZedfdejdddejfddZedfdededddefddZe d deefddZeedfdejdedddejfddZdS)!AbstractFCMDevicez Device IDTzUnique device identifierr)rrrdb_indexrrzRegistration tokenMYSQL_COMPATIBILITY)runique )choicesrr:objectsc@s*eZdZdZedZejddgdgZdS)zAbstractFCMDevice.MetaT FCM devicer[r)fieldsN) rrrr r0rrIndexindexesr!r!r!r"r#s r#rbrSrcrdrCc Ksh|js tddS|j|_ztdtj|fd|i|idWSty3}z||j|d}~ww)a Send single message. The message's token should be blank (and will be overridden if not). Responds with message ID string. :param message: firebase.messaging.Message. If `message` includes a token/id, it will be overridden. :param app: firebase_admin.App. Specify a specific app to use :param more_send_message_kwargs: Parameters for firebase.messaging.send_each() - dry_run: bool. Whether to actually send the notification to the device If there are any new parameters, you can still specify them here. :raises FirebaseError :returns messaging.SendResponse or FirebaseError if the device was deactivated due to an error. Nr&rc)r2rryr[rTsendr $deactivate_devices_with_error_result)r+rSrcrler!r!r"ro&s zAbstractFCMDevice.send_messagerrcKsH|jg}|r tjntj||fd|i|}t||t|j||jdS)an Subscribes or Unsubscribes based on instance's registration_id :param should_subscribe: whether to have these users subscribe (True) or unsubscribe to a topic (False). :param topic: Name of the topic to subscribe to. May contain the ``/topics/`` prefix. :param app: firebase_admin.App. Specify a specific app to use :param more_subscribe_kwargs: Parameters for ``firebase.messaging.subscribe_to_topic()`` If there are any new parameters, you can still specify them here. :raises FirebaseError :returns FirebaseResponseDict rcrV) r[rrrrKrDrrkr)r+rrrcr_r_idsrLr!r!r"rJs$ z+AbstractFCMDevice.handle_topic_subscriptionNcCs|j|gtd|i|gS)Nr&)rrkrry)clsr[ firebase_excr&r!r!r"rnsz6AbstractFCMDevice.deactivate_devices_with_error_result topic_namecKs(||_tdtj|fd|i|idS)Nr&rc)rrryr)rSrrcrlr!r!r"send_topic_messagevs z$AbstractFCMDevice.send_topic_messager9)rrrrr/r0r$ TextFieldr4r[rrrDFCMDeviceManagerrrQr#rrryrorrErKr classmethodrRrrrr!r!r!r"rsd   ( $ rc@seZdZGdddZdS) FCMDevicec@sFeZdZedZedZedsejddgdgZ dZ e ddZ d S) zFCMDevice.Metarz FCM devicesrr[r)r fcm_djangorN)rrrr0rverbose_name_pluralr4rrr app_labelswapperswappable_setting swappabler!r!r!r"r#sr#N)rrrr#r!r!r!r"rsr))collections.abcrrtypingrrr django.dbrdjango.utils.translationrr0firebase_adminrfirebase_admin.exceptionsr r fcm_django.settingsr r4rhModelr Managerr8UnregisteredErrorSenderIdMismatchErrorrHrFrrJrKqueryQuerySetr: from_querysetr TextChoicesrrrr!r!r!r"s2     $  /v