tweaks notifications for favs

This commit is contained in:
Mouse Reeve
2021-01-10 19:29:39 -08:00
parent 5b7d9d1973
commit 31aef46564
2 changed files with 109 additions and 6 deletions

View File

@ -332,12 +332,13 @@ def handle_favorite(user, status):
fav_activity = favorite.to_activity()
broadcast(
user, fav_activity, privacy='direct', direct_recipients=[status.user])
create_notification(
status.user,
'FAVORITE',
related_user=user,
related_status=status
)
if status.user.local:
create_notification(
status.user,
'FAVORITE',
related_user=user,
related_status=status
)
def handle_unfavorite(user, status):
@ -355,6 +356,13 @@ def handle_unfavorite(user, status):
favorite.delete()
broadcast(user, fav_activity, direct_recipients=[status.user])
# check for notification
if status.user.local:
models.Notification.objects.filter(
user=status.user, related_user=user,
status=status, notification_type='FAVORITE'
).first().delete()
def handle_boost(user, status):
''' a user wishes to boost a status '''