kumify/mood/views.py

from django.views.generic import (
    TemplateView,
    ListView,
    UpdateView,
    DetailView,
    CreateView,
    DeleteView,
    View,
)
from django.contrib.auth.mixins import LoginRequiredMixin
from django.shortcuts import get_object_or_404
from django.urls import reverse_lazy
from django.http import HttpResponseRedirect, HttpResponse
from django.utils import timezone
from django.views.decorators.clickjacking import xframe_options_sameorigin
from django.utils.decorators import method_decorator
from django.db.models import Count

from .models import Status, Activity, Mood, StatusMedia, StatusActivity
from .forms import StatusForm
from .statistics import moodstats, activitystats, moodpies, activitymood, activitypies

from common.helpers import get_upload_path
from common.templatetags.images import hvhtml, bkhtml
from msgio.models import NotificationDailySchedule, Notification

from dateutil import relativedelta

from datetime import datetime

import json


class StatusListView(LoginRequiredMixin, ListView):
    template_name = "mood/status_list.html"
    model = Status

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Status List"
        context["subtitle"] = "Just a list of your mood entries."
        context["buttons"] = [
            (reverse_lazy("mood:status_create"), "New Status", "plus")
        ]
        return context

    def get_queryset(self):
        return Status.objects.filter(user=self.request.user).order_by("timestamp")


class StatusViewView(LoginRequiredMixin, DetailView):
    template_name = "mood/status_view.html"
    model = Status

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "View Status"
        context["subtitle"] = "View the details of your mood entry."
        context["buttons"] = [
            (
                reverse_lazy("mood:status_edit", kwargs={"id": self.kwargs["id"]}),
                "Edit Status",
                "pen",
            )
        ]
        return context

    def get_object(self):
        return get_object_or_404(Status, user=self.request.user, id=self.kwargs["id"])


class StatusCreateView(LoginRequiredMixin, CreateView):
    template_name = "mood/status_edit.html"
    form_class = StatusForm
    model = Status

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Create Status"
        context["subtitle"] = "How are you feeling today?"
        context["scripts"] = ["frontend/js/dropdown-to-buttons.js"]
        return context

    def form_valid(self, form):
        form.instance.user = self.request.user

        ret = super().form_valid(form)

        for activity in form.cleaned_data["activities"]:
            if activity.user == self.request.user:
                StatusActivity.objects.create(activity=activity, status=form.instance)

        for attachment in form.cleaned_data["uploads"]:
            dba = StatusMedia(status=form.instance)
            dba.file.save(get_upload_path(form.instance, attachment.name), attachment)
            dba.save()

        return ret

    def get_success_url(self):
        return reverse_lazy("mood:status_view", kwargs={"id": self.object.id})


class StatusEditView(LoginRequiredMixin, UpdateView):
    template_name = "mood/status_edit.html"
    form_class = StatusForm
    model = Status

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Update Status"
        context["subtitle"] = "Change a status you created before."
        context["scripts"] = ["frontend/js/dropdown-to-buttons.js"]
        context["buttons"] = [
            (
                reverse_lazy("mood:status_delete", kwargs={"id": self.kwargs["id"]}),
                "Delete Status",
                "trash-alt",
            )
        ]
        return context

    def get_object(self):
        return get_object_or_404(Status, user=self.request.user, id=self.kwargs["id"])

    def form_valid(self, form):
        for attachment in form.cleaned_data["uploads"]:
            dba = StatusMedia(status=form.instance)
            dba.file.save(get_upload_path(form.instance, attachment.name), attachment)
            dba.save()

        for activity in form.cleaned_data["activities"]:
            if activity.user == self.request.user:
                if activity not in form.instance.activity_set:
                    StatusActivity.objects.create(
                        activity=activity, status=form.instance
                    )

        for statusactivity in form.instance.statusactivity_set.all():
            if statusactivity.activity not in form.cleaned_data["activities"]:
                statusactivity.delete()

        return super().form_valid(form)

    def get_success_url(self):
        return reverse_lazy("mood:status_view", kwargs={"id": self.object.id})


class StatusDeleteView(LoginRequiredMixin, DeleteView):
    template_name = "mood/status_delete.html"
    model = Status

    def get_object(self):
        return get_object_or_404(Status, user=self.request.user, id=self.kwargs["id"])

    def get_success_url(self):
        return reverse_lazy("mood:status_list")


class ActivityListView(LoginRequiredMixin, ListView):
    template_name = "mood/activity_list.html"
    model = Activity

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Activities"
        context["subtitle"] = "The activities you have defined."
        context["buttons"] = [
            (reverse_lazy("mood:activity_create"), "Create Activity", "pen")
        ]
        return context

    def get_queryset(self):
        return Activity.objects.filter(user=self.request.user)


class ActivityEditView(LoginRequiredMixin, UpdateView):
    template_name = "mood/activity_edit.html"
    model = Activity
    fields = ["name", "icon", "color", "hidden"]

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Edit Activity"
        context["subtitle"] = "Make changes to the activity."
        context["scripts"] = [
            "colorfield/jscolor/jscolor.js",
            "colorfield/colorfield.js",
            "frontend/js/fontawesome-iconpicker.min.js",
            "frontend/js/iconpicker-loader.js",
        ]
        context["styles"] = ["frontend/css/fontawesome-iconpicker.min.css"]
        context["buttons"] = [
            (
                reverse_lazy("mood:activity_delete", kwargs={"id": self.kwargs["id"]}),
                "Delete Activity",
                "trash-alt",
            )
        ]
        return context

    def get_object(self):
        return get_object_or_404(Activity, user=self.request.user, id=self.kwargs["id"])

    def get_success_url(self):
        return reverse_lazy("mood:activity_list")


class ActivityCreateView(LoginRequiredMixin, CreateView):
    template_name = "mood/activity_edit.html"
    model = Activity
    fields = ["name", "icon", "color"]

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Create Activity"
        context["subtitle"] = "Add a new activity."
        context["scripts"] = [
            "colorfield/jscolor/jscolor.js",
            "colorfield/colorfield.js",
            "frontend/js/fontawesome-iconpicker.min.js",
            "frontend/js/iconpicker-loader.js",
        ]
        context["styles"] = ["frontend/css/fontawesome-iconpicker.min.css"]
        return context

    def form_valid(self, form):
        obj = form.save(commit=False)
        obj.user = self.request.user
        return super().form_valid(form)

    def get_success_url(self):
        return reverse_lazy("mood:activity_list")


class ActivityDeleteView(LoginRequiredMixin, DeleteView):
    template_name = "mood/activity_delete.html"
    model = Activity

    def get_object(self):
        return get_object_or_404(Activity, user=self.request.user, id=self.kwargs["id"])

    def get_success_url(self):
        return reverse_lazy("mood:activity_list")


class MoodListView(LoginRequiredMixin, ListView):
    template_name = "mood/mood_list.html"
    model = Mood

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Moods"
        context["subtitle"] = "The different moods you have defined."
        return context

    def get_queryset(self):
        return Mood.objects.filter(user=self.request.user)


class MoodEditView(LoginRequiredMixin, UpdateView):
    template_name = "mood/mood_edit.html"
    model = Mood
    fields = ["name", "icon", "color", "value"]

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Edit Mood"
        context["subtitle"] = "Make changes to the mood."
        context["scripts"] = [
            "colorfield/jscolor/jscolor.js",
            "colorfield/colorfield.js",
            "frontend/js/fontawesome-iconpicker.min.js",
            "frontend/js/iconpicker-loader.js",
        ]
        context["styles"] = ["frontend/css/fontawesome-iconpicker.min.css"]
        return context

    def get_object(self):
        return get_object_or_404(Mood, user=self.request.user, id=self.kwargs["id"])

    def get_success_url(self):
        return reverse_lazy("mood:mood_list")


class MoodCreateView(LoginRequiredMixin, CreateView):
    template_name = "mood/mood_edit.html"
    model = Mood
    fields = ["name", "icon", "color", "value"]

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Create Activity"
        context["subtitle"] = "Add a new activity."
        context["scripts"] = [
            "colorfield/jscolor/jscolor.js",
            "colorfield/colorfield.js",
            "frontend/js/fontawesome-iconpicker.min.js",
            "frontend/js/iconpicker-loader.js",
        ]
        context["styles"] = ["frontend/css/fontawesome-iconpicker.min.css"]
        return context

    def form_valid(self, form):
        obj = form.save(commit=False)
        obj.user = self.request.user
        return super().form_valid(form)

    def get_success_url(self):
        return reverse_lazy("mood:activity_list")


class NotificationListView(LoginRequiredMixin, ListView):
    template_name = "mood/notification_list.html"
    model = NotificationDailySchedule
    fields = ["time"]

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Notifications"
        context["subtitle"] = "The daily reminders you have set up."
        context["buttons"] = [
            (reverse_lazy("mood:notification_create"), "New Notification", "plus")
        ]
        return context

    def get_queryset(self):
        return NotificationDailySchedule.objects.filter(
            notification__recipient=self.request.user, notification__app="mood"
        )


class NotificationCreateView(LoginRequiredMixin, CreateView):
    template_name = "mood/notification_edit.html"
    model = NotificationDailySchedule
    fields = ["time"]

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Create Notification"
        context["subtitle"] = "Add a new daily notification."
        return context

    def form_valid(self, form):
        notification = Notification.objects.create(
            content="Hi, it's time for a new Kumify entry! Go to %KUMIFYURL% to document your mood!",
            recipient=self.request.user,
            app="mood",
        )
        obj = form.save(commit=False)
        obj.notification = notification
        return super().form_valid(form)

    def get_success_url(self):
        return reverse_lazy("mood:notification_list")


class NotificationEditView(LoginRequiredMixin, UpdateView):
    template_name = "mood/notification_edit.html"
    model = NotificationDailySchedule
    fields = ["time"]

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = "Edit Notification"
        context["subtitle"] = "Change the time of a daily notification."
        context["buttons"] = [
            (
                reverse_lazy("mood:notification_delete", args=[self.kwargs["id"]]),
                "Delete Notification",
            )
        ]
        return context

    def get_success_url(self):
        return reverse_lazy("mood:notification_list")

    def get_object(self):
        return get_object_or_404(
            NotificationDailySchedule,
            notification__recipient=self.request.user,
            id=self.kwargs["id"],
        )


class NotificationDeleteView(LoginRequiredMixin, DeleteView):
    template_name = "mood/notification_delete.html"
    model = NotificationDailySchedule

    def get_object(self):
        return get_object_or_404(
            NotificationDailySchedule,
            notification__recipient=self.request.user,
            id=self.kwargs["id"],
        )

    def delete(self, request, *args, **kwargs):
        self.object = self.get_object()
        success_url = self.get_success_url()
        self.object.notification.delete()
        return HttpResponseRedirect(success_url)

    def get_success_url(self):
        return reverse_lazy("mood:notification_list")


class MoodStatisticsView(LoginRequiredMixin, TemplateView):
    template_name = "mood/statistics.html"

    def get_context_data(self, **kwargs):
        startdate = self.request.GET.get("start")
        enddate = self.request.GET.get("end")

        if enddate:
            maxdate = datetime.strptime(enddate, "%Y-%m-%d")
        else:
            maxdate = timezone.now()

        if startdate:
            mindate = datetime.strptime(startdate, "%Y-%m-%d")
        else:
            mindate = maxdate - relativedelta.relativedelta(weeks=1) # noqa: F841
            # TODO: Do something with this...?

        context = super().get_context_data(**kwargs)
        context["title"] = "Statistics"
        context["activities"] = activitystats(self.request.user)
        return context


class MoodCSVView(LoginRequiredMixin, View):
    def get(self, request, *args, **kwargs):
        res = HttpResponse(content_type="text/csv")
        res["content-disposition"] = 'filename="mood.csv"'

        startdate = request.GET.get("start")
        enddate = request.GET.get("end")

        maxdate = None
        mindate = None

        if enddate:
            maxdate = datetime.strptime(enddate, "%Y-%m-%d")

            if not startdate:
                mindate = maxdate - relativedelta.relativedelta(weeks=1)

        if startdate:
            mindate = datetime.strptime(startdate, "%Y-%m-%d")

            if not enddate:
                maxdate = mindate + relativedelta.relativedelta(weeks=1)

        if not maxdate:
            maxdate = timezone.now()
            mindate = maxdate - relativedelta.relativedelta(weeks=1)

        output = "date,value"

        for status in Status.objects.filter(
            user=request.user, timestamp__gte=mindate, timestamp__lte=maxdate
        ):
            if status.mood:
                date = status.timestamp.strftime("%Y-%m-%d %H:%M")
                output += f"\n{date},{status.mood.value}"

        res.write(output)
        return res


class MoodPlotView(LoginRequiredMixin, View):
    @method_decorator(xframe_options_sameorigin)
    def dispatch(self, *args, **kwargs):
        return super().dispatch(*args, **kwargs)

    def get(self, request, *args, **kwargs):
        res = HttpResponse(content_type="text/html")

        res.write(hvhtml(moodstats(request.user)))
        return res


class MoodPiesView(LoginRequiredMixin, View):
    @method_decorator(xframe_options_sameorigin)
    def dispatch(self, *args, **kwargs):
        return super().dispatch(*args, **kwargs)

    def get(self, request, *args, **kwargs):
        res = HttpResponse(content_type="text/html")

        res.write(bkhtml(moodpies(request.user)))
        return res


class ActivityStatisticsView(LoginRequiredMixin, TemplateView):
    template_name = "mood/statistics_activity.html"

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        activity = get_object_or_404(Activity, user=self.request.user, id=kwargs["id"])
        context["title"] = "Activity Statistics for %s" % activity.name
        return context


class ActivityPlotView(LoginRequiredMixin, View):
    @method_decorator(xframe_options_sameorigin)
    def dispatch(self, *args, **kwargs):
        return super().dispatch(*args, **kwargs)

    def get(self, request, *args, **kwargs):
        res = HttpResponse(content_type="text/html")

        res.write(
            hvhtml(
                activitymood(
                    get_object_or_404(Activity, user=request.user, id=kwargs["id"])
                )
            )
        )
        return res


class ActivityPiesView(LoginRequiredMixin, View):
    @method_decorator(xframe_options_sameorigin)
    def dispatch(self, *args, **kwargs):
        return super().dispatch(*args, **kwargs)

    def get(self, request, *args, **kwargs):
        res = HttpResponse(content_type="text/html")

        res.write(
            bkhtml(
                activitypies(
                    get_object_or_404(Activity, user=request.user, id=kwargs["id"])
                )
            )
        )
        return res


class MoodCountHeatmapJSONView(LoginRequiredMixin, View):
    """Returns a JSON object with the mood entries for a given time period.

    This is used in conjunction with the Cal-Heatmap library to display a
    heatmap of mood entries.
    """

    def get(self, request, *args, **kwargs):
        res = HttpResponse(content_type="application/json")

        start = request.GET.get("start")
        end = request.GET.get("end")

        if end:
            maxdate = datetime.strptime(end, "%Y-%m-%d")
        else:
            maxdate = timezone.now()

        if start:
            mindate = datetime.strptime(start, "%Y-%m-%d")
        else:
            mindate = maxdate - relativedelta.relativedelta(years=1)

        data = (
            Status.objects.filter(
                user=request.user, timestamp__gte=mindate, timestamp__lte=maxdate
        
            )
            .annotate(value=Count("id"))
        )

        # TODO: Should eventually change this so that it returns a *color* as a value and the count as a tooltip
        data = [{"date": d.timestamp.strftime("%Y-%m-%d"), "value": d.value} for d in data]

        res.write(json.dumps(data))

        return res