mirror of
https://github.com/samuelclay/NewsBlur.git
synced 2025-04-13 09:42:01 +00:00
100 lines
No EOL
3.6 KiB
Python
100 lines
No EOL
3.6 KiB
Python
import redis
|
|
from django.conf import settings
|
|
from django.contrib.admin.views.decorators import staff_member_required
|
|
from utils import json_functions as json
|
|
from utils.user_functions import get_user, ajax_login_required
|
|
from apps.notifications.models import MUserFeedNotification, MUserNotificationTokens
|
|
from apps.rss_feeds.models import Feed
|
|
from utils.view_functions import required_params
|
|
from utils import log as logging
|
|
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def notifications_by_feed(request):
|
|
user = get_user(request)
|
|
notifications_by_feed = MUserFeedNotification.feeds_for_user(user.pk)
|
|
|
|
return notifications_by_feed
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def set_notifications_for_feed(request):
|
|
user = get_user(request)
|
|
feed_id = request.POST['feed_id']
|
|
notification_types = request.POST.getlist('notification_types') or request.POST.getlist('notification_types[]')
|
|
notification_filter = request.POST.get('notification_filter')
|
|
|
|
try:
|
|
notification = MUserFeedNotification.objects.get(user_id=user.pk, feed_id=feed_id)
|
|
except MUserFeedNotification.DoesNotExist:
|
|
params = {
|
|
"user_id": user.pk,
|
|
"feed_id": feed_id,
|
|
}
|
|
notification = MUserFeedNotification.objects.create(**params)
|
|
|
|
web_was_off = not notification.is_web
|
|
notification.is_focus = bool(notification_filter == "focus")
|
|
notification.is_email = bool('email' in notification_types)
|
|
notification.is_ios = bool('ios' in notification_types)
|
|
notification.is_android = bool('android' in notification_types)
|
|
notification.is_web = bool('web' in notification_types)
|
|
notification.save()
|
|
|
|
if (not notification.is_email and
|
|
not notification.is_ios and
|
|
not notification.is_android and
|
|
not notification.is_web):
|
|
notification.delete()
|
|
|
|
r = redis.Redis(connection_pool=settings.REDIS_PUBSUB_POOL)
|
|
if web_was_off and notification.is_web:
|
|
r.publish(user.username, 'notification:setup:%s' % feed_id)
|
|
|
|
notifications_by_feed = MUserFeedNotification.feeds_for_user(user.pk)
|
|
|
|
return {"notifications_by_feed": notifications_by_feed}
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def set_apns_token(request):
|
|
user = get_user(request)
|
|
tokens = MUserNotificationTokens.get_tokens_for_user(user.pk)
|
|
apns_token = request.REQUEST['apns_token']
|
|
|
|
logging.user(user, "~FCUpdating APNS push token")
|
|
if apns_token not in tokens.ios_tokens:
|
|
tokens.ios_tokens.append(apns_token)
|
|
tokens.save()
|
|
return {'message': 'Token saved.'}
|
|
|
|
return {'message': 'Token already saved.'}
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def set_android_token(request):
|
|
user = get_user(request)
|
|
tokens = MUserNotificationTokens.get_tokens_for_user(user.pk)
|
|
token = request.REQUEST['token']
|
|
|
|
logging.user(user, "~FCUpdating Android push token")
|
|
if token not in tokens.android_tokens:
|
|
tokens.android_tokens.append(token)
|
|
tokens.save()
|
|
return {'message': 'Token saved.'}
|
|
|
|
return {'message': 'Token already saved.'}
|
|
|
|
@required_params(feed_id=int)
|
|
@staff_member_required
|
|
@json.json_view
|
|
def force_push(request):
|
|
user = get_user(request)
|
|
feed_id = request.REQUEST['feed_id']
|
|
count = int(request.REQUEST.get('count', 1))
|
|
|
|
logging.user(user, "~BM~FWForce pushing %s stories: ~SB%s" % (count, Feed.get_by_id(feed_id)))
|
|
sent_count, user_count = MUserFeedNotification.push_feed_notifications(feed_id, new_stories=count, force=True)
|
|
|
|
return {"message": "Pushed %s notifications to %s users" % (sent_count, user_count)} |