77 lines
2.5 KiB
Python
77 lines
2.5 KiB
Python
MAX_PAYLOAD_SIZE = 4096
|
|
|
|
|
|
class PayloadAlert(object):
|
|
def __init__(self, title=None, title_localized_key=None, title_localized_args=None,
|
|
body=None, body_localized_key=None, body_localized_args=None,
|
|
action_localized_key=None,
|
|
launch_image=None
|
|
):
|
|
super().__init__()
|
|
self.title = title
|
|
self.title_localized_key = title_localized_key
|
|
self.title_localized_args = title_localized_args
|
|
self.body = body
|
|
self.body_localized_key = body_localized_key
|
|
self.body_localized_args = body_localized_args
|
|
self.action_localized_key = action_localized_key
|
|
self.launch_image = launch_image
|
|
|
|
def dict(self):
|
|
result = {}
|
|
|
|
if self.title:
|
|
result['title'] = self.title
|
|
if self.title_localized_key:
|
|
result['title-loc-key'] = self.title_localized_key
|
|
if self.title_localized_args:
|
|
result['title-loc-args'] = self.title_localized_args
|
|
|
|
if self.body:
|
|
result['body'] = self.body
|
|
if self.body_localized_key:
|
|
result['loc-key'] = self.body_localized_key
|
|
if self.body_localized_args:
|
|
result['loc-args'] = self.body_localized_args
|
|
|
|
if self.action_localized_key:
|
|
result['action-loc-key'] = self.action_localized_key
|
|
|
|
if self.launch_image:
|
|
result['launch-image'] = self.launch_image
|
|
|
|
return result
|
|
|
|
|
|
class Payload(object):
|
|
def __init__(self, alert=None, badge=None, sound=None, content_available=None, category=None, custom=None):
|
|
super(Payload, self).__init__()
|
|
self.alert = alert
|
|
self.badge = badge
|
|
self.sound = sound
|
|
self.content_available = content_available
|
|
self.category = category
|
|
self.custom = custom
|
|
|
|
def dict(self):
|
|
result = {
|
|
'aps': {}
|
|
}
|
|
if self.alert:
|
|
if self.alert is PayloadAlert:
|
|
result['aps']['alert'] = self.alert.dict()
|
|
else:
|
|
result['aps']['alert'] = self.alert
|
|
if self.badge:
|
|
result['aps']['badge'] = self.badge
|
|
if self.sound:
|
|
result['aps']['sound'] = self.sound
|
|
if self.content_available:
|
|
result['aps']['content-available'] = 1
|
|
if self.category:
|
|
result['aps']['category'] = self.category
|
|
if self.custom:
|
|
result.update(self.custom)
|
|
|
|
return result
|