push: Move remove_push_device_token to library.

This commit is contained in:
Tim Abbott
2017-03-05 19:11:13 -08:00
parent c0ad9c02fd
commit 657dfcbddc
2 changed files with 15 additions and 13 deletions

View File

@@ -8,6 +8,7 @@ from zerver.models import PushDeviceToken, Message, Recipient, UserProfile, \
receives_online_notifications
from zerver.models import get_user_profile_by_id
from zerver.lib.avatar import avatar_url
from zerver.lib.request import JsonableError
from zerver.lib.timestamp import datetime_to_timestamp, timestamp_to_datetime
from zerver.decorator import statsd_increment
from zerver.lib.utils import generate_random_token
@@ -355,3 +356,11 @@ def add_push_device_token(user_profile, token_str, kind, ios_app_id=None):
if not created:
token.last_updated = timezone.now()
token.save(update_fields=['last_updated'])
def remove_push_device_token(request, user_profile, token_str, kind):
# type: (HttpRequest, UserProfile, str, int) -> None
try:
token = PushDeviceToken.objects.get(token=token_str, kind=kind)
token.delete()
except PushDeviceToken.DoesNotExist:
raise JsonableError(_("Token does not exist"))