acronomy/acros/views.py

from datetime import date

from django.contrib.auth.mixins import LoginRequiredMixin
from django.contrib.messages.views import SuccessMessageMixin
from django.http import HttpResponse
from django.views import generic
from rest_framework import viewsets, filters

from acros.forms import EditForm, AddForm
from acros.models import Acronym, Tag, AcroOfTheDay
from acros.serializers import AcronymSerializer, AcronymListSerializer, TagSerializer
from acros.utils.assets import get_css

handler404 = 'acros.views.PageNotFoundView'


class IndexView(generic.TemplateView):
    template_name = "acros/index.html"

    def get_context_data(self, **kwargs):
        data = super().get_context_data(**kwargs)
        aotd = AcroOfTheDay.objects.filter(date=date.today()).select_related('acronym')
        data['acronyms_of_the_day'] = aotd
        data['num_acronyms'] = Acronym.objects.all().count()
        return data


class PageNotFoundView(generic.TemplateView):
    template_name = "404.html"


class OverView(generic.ListView):
    template_name = "acros/overview.html"
    model = Acronym
    context_object_name = 'acros'
    ordering = "name"


class DetailView(generic.DetailView):
    template_name = 'acros/detail.html'
    context_object_name = 'acro'
    model = Acronym


class EditView(LoginRequiredMixin, SuccessMessageMixin, generic.UpdateView):
    template_name = 'acros/edit.html'
    context_object_name = 'acro'
    model = Acronym
    # fields = ['name', 'full_name', "description_md", "tags"]
    form_class = EditForm
    success_message = 'Acronym "%(name)s" was edited successfully'


class AddView(LoginRequiredMixin, SuccessMessageMixin, generic.CreateView):
    template_name = "acros/add.html"
    form_class = AddForm
    model = Acronym
    success_message = 'Acronym "%(name)s" was created successfully'


class TagListView(generic.ListView):
    template_name = "acros/taglist.html"
    model = Tag
    context_object_name = 'tags'
    ordering = "name"
    queryset = Tag.objects.exclude(acronyms=None)


class TagAcroView(generic.ListView):
    template_name = "acros/tagacro.html"
    context_object_name = 'acros'
    ordering = "name"
    allow_empty = False

    def get_queryset(self):
        return Acronym.objects.filter(tags__slug__exact=self.kwargs['slug'])


#### API Views ####

class AcronymViewSet(viewsets.ReadOnlyModelViewSet):
    """
    API endpoint for viewing Acronyms
    """
    queryset = Acronym.objects.all().order_by('name')
    serializer_class = AcronymSerializer

    serializer_classes = {
        'list': AcronymListSerializer,
        'retrieve': AcronymSerializer,
    }
    default_serializer_class = AcronymListSerializer

    filter_backends = [filters.SearchFilter]
    search_fields = ['name']

    def get_serializer_class(self):
        return self.serializer_classes.get(self.action, self.default_serializer_class)


class TagViewSet(viewsets.ReadOnlyModelViewSet):
    queryset = Tag.objects.all().order_by("name")
    serializer_class = TagSerializer


##### DEBUG views #####


def debug_css(request):
    css, source_map = get_css(debug=True)
    return HttpResponse(css, content_type="text/css")


def debug_css_sourcemap(request):
    css, source_map = get_css(debug=True)
    return HttpResponse(source_map, content_type="application/json")