paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-14 02:48:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Annotate push_notifications.py.

Browse Source
This commit is contained in:
Umair Khan
2016-08-08 13:50:10 +05:00
committed by Tim Abbott
parent 9c13e9a072
commit acd76eb604
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
zerver/lib/push_notifications.py
View File

@@ -2,7 +2,7 @@ from __future__ import absolute_import
import random import random
from six import text_type from six import text_type
from typing import Any, Dict, Optional from typing import Any, Dict, Optional, SupportsInt
from zerver.models import PushDeviceToken, UserProfile from zerver.models import PushDeviceToken, UserProfile
from zerver.models import get_user_profile_by_id from zerver.models import get_user_profile_by_id
@@ -47,11 +47,13 @@ dbx_connection = None
assert isinstance(settings.APNS_SANDBOX, bool) assert isinstance(settings.APNS_SANDBOX, bool)
def get_apns_key(identifer): def get_apns_key(identifer):
# type: (SupportsInt) -> str
return 'apns:' + str(identifer) return 'apns:' + str(identifer)
class APNsMessage(object): class APNsMessage(object):
def __init__(self, user, tokens, alert=None, badge=None, sound=None, def __init__(self, user, tokens, alert=None, badge=None, sound=None,
category=None, **kwargs): category=None, **kwargs):
# type: (UserProfile, List[text_type], text_type, int, text_type, text_type, **Any) -> None
self.frame = Frame() self.frame = Frame()
self.tokens = tokens self.tokens = tokens
expiry = int(time.time() + 24 * 3600) expiry = int(time.time() + 24 * 3600)
@@ -67,9 +69,11 @@ class APNsMessage(object):
self.frame.add_item(token, payload, identifier, expiry, priority) self.frame.add_item(token, payload, identifier, expiry, priority)
def get_frame(self): def get_frame(self):
# type: () -> Frame
return self.frame return self.frame
def response_listener(error_response): def response_listener(error_response):
# type: (Dict[str, SupportsInt]) -> None
identifier = error_response['identifier'] identifier = error_response['identifier']
key = get_apns_key(identifier) key = get_apns_key(identifier)
if not redis_client.exists(key): if not redis_client.exists(key):
@@ -95,6 +99,7 @@ def response_listener(error_response):
pass pass
def get_connection(cert_file, key_file): def get_connection(cert_file, key_file):
# type: (str, str) -> APNs
connection = APNs(use_sandbox=settings.APNS_SANDBOX, connection = APNs(use_sandbox=settings.APNS_SANDBOX,
cert_file=cert_file, cert_file=cert_file,
key_file=key_file, key_file=key_file,