diff --git a/apns2/client.py b/apns2/client.py index fbd99da..8f8632f 100644 --- a/apns2/client.py +++ b/apns2/client.py @@ -21,10 +21,12 @@ class APNsClient(object): ssl_context.load_cert_chain(cert_file) self.__connection = HTTP20Connection(server, port, ssl_context=ssl_context) - def send_notification(self, token_hex, notification, topic=None): + def send_notification(self, token_hex, notification, prioriy=NotificationPriority.Immediate, topic=None): json_payload = dumps(notification.dict(), ensure_ascii=False, separators=(',',':')).encode('utf-8') - headers = {} + headers = { + 'apns-priority': prioriy.value + } if topic: headers['apns-topic'] = topic diff --git a/apns2/client.pyi b/apns2/client.pyi index 13c706a..c5e4c5d 100644 --- a/apns2/client.pyi +++ b/apns2/client.pyi @@ -1,5 +1,6 @@ from typing import Optional +from apns2.client import NotificationPriority from apns2.payload import Payload @@ -10,6 +11,7 @@ class APNsClient(object): use_alternate_port: bool = False) -> None: ... def send_notification(self, - token_hex: str, - notification: Payload, - topic: Optional[str] = None) -> None: ... + token_hex: str, + notification: Payload, + prioriy: NotificationPriority = NotificationPriority.Immediate, + topic: Optional[str] = None) -> None: ...