65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
|
from .apns import send_notification
|
||
|
|
from music.model.user import User
|
||
|
|
from music.model.playlist import Playlist
|
||
|
|
from music.model.tag import Tag
|
||
|
|
import logging
|
||
|
|
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
def notify_user_playlist_update(user: User, playlist: Playlist):
|
||
|
|
|
||
|
|
if user.notify_playlist_updates:
|
||
|
|
notify_user(user=user, alert={
|
||
|
|
"title": "Playlist Updated",
|
||
|
|
"body": f"{playlist.name} was just updated"
|
||
|
|
}, priority=1)
|
||
|
|
|
||
|
|
|
||
|
|
def notify_user_tag_update(user: User, tag: Tag):
|
||
|
|
|
||
|
|
if user.notify_tag_updates:
|
||
|
|
notify_user(user=user, alert={
|
||
|
|
"title": "Tag Updated",
|
||
|
|
"body": f"{tag.name} was just updated"
|
||
|
|
}, priority=1)
|
||
|
|
|
||
|
|
|
||
|
|
def notify_admin_new_user(user: User, new_username: str):
|
||
|
|
|
||
|
|
if user.notify_admins:
|
||
|
|
notify_user(user=user, alert={
|
||
|
|
"title": "New User",
|
||
|
|
"body": f"{new_username} just created an account"
|
||
|
|
})
|
||
|
|
|
||
|
|
|
||
|
|
def notify_user(user: User,
|
||
|
|
alert: dict | str = None,
|
||
|
|
badge: int = None,
|
||
|
|
expiry: int = None,
|
||
|
|
priority: int = None):
|
||
|
|
|
||
|
|
if user.apns_tokens:
|
||
|
|
logger.debug(f"notifying {user.username}")
|
||
|
|
|
||
|
|
stale_tokens = []
|
||
|
|
for device in user.apns_tokens:
|
||
|
|
resp = send_notification(device_token=device,
|
||
|
|
alert=alert,
|
||
|
|
badge=badge,
|
||
|
|
expiry=expiry,
|
||
|
|
priority=priority)
|
||
|
|
|
||
|
|
if 400 <= resp.status_code < 600:
|
||
|
|
logger.warning(f'http error when notifying {user.username}: {resp.content}')
|
||
|
|
|
||
|
|
if resp.status_code == 410:
|
||
|
|
logger.warning(f'http error 410 when notifying {user.username}: removing stale token')
|
||
|
|
stale_tokens += [device]
|
||
|
|
|
||
|
|
if len(stale_tokens) > 0:
|
||
|
|
user.apns_tokens = [i for i in user.apns_tokens if i not in stale_tokens]
|
||
|
|
user.update()
|