diff options
author | 2017-09-28 10:03:40 -0700 | |
---|---|---|
committer | 2017-09-28 10:14:55 -0700 | |
commit | bd886d918ef2adbabd16c61fdd2e47984e21dfd7 (patch) | |
tree | d41683dffa58fd698df450d148fab3cc2521b0c5 /django/engagementmanager/notifications.py | |
parent | 474554adad912f3edb7ddc3ad14406abb369fb3c (diff) |
initial seed code commit VVP-5
Change-Id: I6560c87ef48a6d0d1fe8197c7c6439c7e6ad653f
Signed-off-by: Paul McGoldrick <paul.mcgoldrick@att.com>
Diffstat (limited to 'django/engagementmanager/notifications.py')
-rwxr-xr-x | django/engagementmanager/notifications.py | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/django/engagementmanager/notifications.py b/django/engagementmanager/notifications.py new file mode 100755 index 0000000..59f2e1f --- /dev/null +++ b/django/engagementmanager/notifications.py @@ -0,0 +1,71 @@ +# +# ============LICENSE_START========================================== +# org.onap.vvp/engagementmgr +# =================================================================== +# Copyright © 2017 AT&T Intellectual Property. All rights reserved. +# =================================================================== +# +# Unless otherwise specified, all software contained herein is licensed +# under the Apache License, Version 2.0 (the “License”); +# you may not use this software except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# +# +# Unless otherwise specified, all documentation contained herein is licensed +# under the Creative Commons License, Attribution 4.0 Intl. (the “License”); +# you may not use this documentation except in compliance with the License. +# You may obtain a copy of the License at +# +# https://creativecommons.org/licenses/by/4.0/ +# +# Unless required by applicable law or agreed to in writing, documentation +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# ============LICENSE_END============================================ +# +# ECOMP is a trademark and service mark of AT&T Intellectual Property. +from engagementmanager.models import Notification, IceUserProfile +from engagementmanager.serializers import ThinNotificationModelSerializer +from engagementmanager.service.logging_service import LoggingServiceFactory + +logger = LoggingServiceFactory.get_logger() + + +def num_of_notifications_for_user(user_uuid): + user = IceUserProfile.objects.get(uuid=user_uuid) + amountOfnotificationForUser = Notification.objects.filter(user=user, is_read=False).filter().count() + return amountOfnotificationForUser + + +def pull_recent_notifications(user_uuid, offset, limit): + ModifiedNotificationsSet = [] + logger.debug("Pulling unread notifications from DB") + user = IceUserProfile.objects.get(uuid=user_uuid) + notificationsList = Notification.objects.filter(user=user).order_by('is_read', '-activity__create_time')[int(offset):int(offset) + int(limit)] + num_of_objects = Notification.objects.filter(user=user).count() + for notif in notificationsList: + serializedNotification = ThinNotificationModelSerializer(notif).data + ModifiedNotificationsSet.append(serializedNotification) + notif.is_read = True + notif.save() + return ModifiedNotificationsSet, num_of_objects + + +def reset_num_of_notifications_for_user(user_uuid): + user = IceUserProfile.objects.get(uuid=user_uuid) + notificationForUser = Notification.objects.filter(user=user, is_read=False) + for notif in notificationForUser: + notif.is_read = True + notif.save() |