diff options
| -rw-r--r-- | notifications.py | 39 |
1 files changed, 18 insertions, 21 deletions
diff --git a/notifications.py b/notifications.py index f5c7e77..bb235fa 100644 --- a/notifications.py +++ b/notifications.py @@ -42,24 +42,21 @@ def broadcast_to_topic(topic: str, title: str, body: str, image): print(f"No tokens found for topic '{topic}'") return - for token in tokens: - message = messaging.Message( - notification=messaging.Notification( - title=title, - body=body, - image=image - ), - token=token - ) - try: - response = messaging.send(message) - print(f"Sent to {token}: {response}") - except Exception as e: - print(f"Error sending to {token}: {e}") - -if __name__ == "__main__": - topic = "sdvx" - title = "New sdvx Update!" - body = "Check out the latest DDR content now!" - - broadcast_to_topic(topic, title, body) + multicast_message = messaging.MulticastMessage( + notification=messaging.Notification( + title=title, + body=body, + image=image + ), + tokens=tokens + ) + try: + response = messaging.send_each_for_multicast(multicast_message) + print(f"Successfully sent {response.success_count} messages out of {len(tokens)}") + if response.failure_count > 0: + print(f"Failed to send {response.failure_count} messages") + for idx, error in enumerate(response.responses): + if not error.success: + print(f"Error for token {tokens[idx]}: {error.exception}") + except Exception as e: + print(f"Error sending multicast message: {e}") |