digest_app/module_notification/views.py

import itertools
import logging
from datetime import datetime, timedelta

from django.contrib import messages
from django.contrib.auth.mixins import LoginRequiredMixin
from django.db.models import Q
from django.http import HttpRequest
from django.http.response import HttpResponse as HttpResponse
from django.shortcuts import get_object_or_404, redirect, render
from django.urls import reverse_lazy
from django.views import generic
from django_datatables_view.base_datatable_view import BaseDatatableView

from module_iam import iam_constant, permission
from module_iam.iam_constant import PRINCIPAL_TYPE_USER
from module_iam.models import IAmPrincipal
from module_project import constants, date_utils
from module_project.mixins import ActionMixin
from module_project.service import OneSignalService
from module_project.utils import JsonResponseUtil

from .forms import PushNotificationForm
from .models import PushNotification


# Create your views here.
class NotificationView(permission.ResourcePermissionRequiredMixin, LoginRequiredMixin, generic.TemplateView):
    page_name = iam_constant.RESOURCE_MANAGE_NOTIFICATION
    resource = iam_constant.RESOURCE_MANAGE_NOTIFICATION
    template_name = "module_notification/notification.html"

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["page_name"] = self.page_name
        return context


class NotificationListJsonView(BaseDatatableView):
    model = PushNotification
    columns = ["id", "title", "message", "active", "timestamp"]
    order_columns = ["id", "title", "message", "active", "timestamp"]

    FILTER_ICONTAINS = "icontains"

    def get_filter_method(self):
        """Returns preferred filter method"""
        return self.FILTER_ICONTAINS

    def get_initial_queryset(self):
        deleted_flag = self.request.GET.get("deleted_flag", None)
        return self.model.objects.filter(deleted=deleted_flag)

    def ordering(self, qs):
        order = self.request.GET.get('order[0][dir]', None)
        if order:
            column_index = int(self.request.GET.get('order[0][column]', None)) - 1
            order_column = self.order_columns[column_index]

            if order == "asc":
                qs = qs.order_by(order_column)
            elif order == "desc":
                qs = qs.order_by("-" + order_column)

        return qs


class NotificationCreateOrUpdateView(permission.ResourcePermissionRequiredMixin, LoginRequiredMixin, generic.View):
    # Set the page_name and resource
    page_name = iam_constant.RESOURCE_MANAGE_NOTIFICATION
    resource = iam_constant.RESOURCE_MANAGE_NOTIFICATION

    # Initialize the action as ACTION_CREATE (can change based on logic)
    action = iam_constant.ACTION_CREATE  # Default action

    template_name = "module_notification/add_notification.html"
    model = PushNotification
    form_class = PushNotificationForm
    success_url = reverse_lazy("module_notification:notification")
    error_message = "An error occurred while saving the data."

    # Determine the success message dynamically based on whether it's an update or create
    def get_success_message(self):
        self.success_message = (
            constants.RECORD_CREATED if not self.object else constants.RECORD_UPDATED
        )
        return self.success_message

    # Get the object (if exists) based on URL parameter 'pk
    def get_object(self):
        pk = self.kwargs.get("pk")
        return get_object_or_404(self.model, pk=pk) if pk else None

    # Add page_name and operation to the context
    def get_context_data(self, **kwargs):
        context = {
            "page_name": self.page_name,
            "operation": "Add" if not self.object else "Edit",
        }
        context.update(kwargs)  # Include any additional context data passed to the view
        return context

    def get(self, request, *args, **kwargs):
        self.object = self.get_object()

        # If an object is found, change action to ACTION_UPDATE
        if self.object is not None:
            self.action = iam_constant.ACTION_UPDATE

        form = self.form_class(instance=self.object)
        context = self.get_context_data(form=form)
        return render(request, self.template_name, context=context)

    def post(self, request, *args, **kwargs):
        print("Request data: ", request.POST)
        self.object = self.get_object()

        # If an object is found, change action to ACTION_UPDATE
        if self.object is not None:
            self.action = iam_constant.ACTION_UPDATE

        form = self.form_class(request.POST, instance=self.object)
        if not form.is_valid():
            print(form.errors)
            context = self.get_context_data(form=form)
            return render(request, self.template_name, context=context)

        form.save()
        messages.success(self.request, self.get_success_message())
        return redirect(self.success_url)


class NotificationActionView(ActionMixin):
    model = PushNotification

class NotificationArchiveView(permission.ResourcePermissionRequiredMixin, LoginRequiredMixin, generic.TemplateView):
    page_name = iam_constant.RESOURCE_MANAGE_NOTIFICATION
    resource = iam_constant.RESOURCE_MANAGE_NOTIFICATION
    action = None
    template_name = "module_notification/notification_archive.html"
    model = PushNotification

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["page_name"] = self.page_name
        return context

class NotificationSendView(generic.View):
    model = PushNotification

    def get_image_url(self, obj, field_name, request):
        image_field = getattr(obj, field_name)
        if image_field:
            return request.build_absolute_uri(image_field.url)
        return ""

    def post(self, request, *args, **kwargs):
        id = request.POST.get("id")
        obj = self.model.objects.filter(pk=int(id)).first()

        if not obj:
            return JsonResponseUtil.error(
                message="No notification with such ID exists."
            )
        
        if not obj.active:
            return JsonResponseUtil.error(
                message="The notification cannot be sent because it is inactive. Please activate the notification before attempting to send it."
            )

        # Get the current date and subtract 15 days
        fifteen_days_ago = datetime.now() - timedelta(days=3)

        # Filter the IAmPrincipal objects based on the last_login field being greater than or equal to fifteen_days_ago
        player_ids = list(
            IAmPrincipal.objects.filter(
                principal_type__name=PRINCIPAL_TYPE_USER,
                last_login__gte=fifteen_days_ago,
            ).values_list("player_id", flat=True)
        )
        # removing none from list
        player_ids = list(itertools.filterfalse(lambda x: x is None, player_ids))

        print(f"player id is {player_ids}")

        try:
            notification = OneSignalService()
            response = notification.send_notification(
                headings=obj.title,
                contents=obj.message,
                include_player_ids=player_ids,
            )
        except Exception as e:
            error_response = {
                "status": 400,
                "message": constants.INTERNAL_SERVER_ERROR,
                "errors": str(e),
            }
            return JsonResponseUtil.error(**error_response)

        return JsonResponseUtil.success(message="success")