Source code for documents.forms

"""The forms defined by the documents package"""
from django import forms
from django.contrib import admin
from django.db.models import Q
from django.forms import widgets
from django.utils import timezone

from documents import models
from utils.snippets import datetime_to_lectureyear


class DocumentFileInput(widgets.ClearableFileInput):
    """
    Wrapper around Django's :class:`~django.forms.widgets.ClearableFileInput`

    It overrides the URL of the associated file when it is fetched.
    """

    template_name = "widgets/clearable_file_input.html"

    def get_context(self, name, value, attrs):
        context = super().get_context(name, value, attrs)
        if hasattr(value, "url"):
            doc = models.Document.objects.get(Q(file_en=value) | Q(file_nl=value))
            context["document_id"] = doc.pk
            if doc.file_en == value:
                context["language"] = "en"
            else:
                context["language"] = "nl"
        return context


class MinutesForm(forms.ModelForm):
    """Form that overrides the widgets for the files"""

    class Meta:
        model = models.Minutes
        fields = (
            "file_nl",
            "file_en",
            "members_only",
        )
        widgets = {
            "file_nl": DocumentFileInput,
            "file_en": DocumentFileInput,
        }


class AnnualDocumentForm(forms.ModelForm):
    """Form that provides custom functionality for annual documents"""

    class Meta:
        model = models.AnnualDocument
        exclude = ()
        widgets = {
            "year": forms.Select,
            "file_nl": DocumentFileInput,
            "file_en": DocumentFileInput,
        }

    def current_year():
        """Get the current lecture year"""
        return datetime_to_lectureyear(timezone.now())

    def year_choices():
        """Get the lecture years"""
        current = datetime_to_lectureyear(timezone.now())
        return [
            (year, "{}-{}".format(year, year + 1))
            for year in range(current + 1, 1989, -1)
        ]

    year = forms.TypedChoiceField(
        coerce=int, choices=year_choices, initial=current_year
    )


class AssociationDocumentForm(forms.ModelForm):
    """Form that overrides the widgets for the files"""

    class Meta:
        model = models.AssociationDocument
        fields = (
            "name_en",
            "name_nl",
            "file_en",
            "file_nl",
            "members_only",
        )
        widgets = {
            "file_nl": DocumentFileInput,
            "file_en": DocumentFileInput,
        }


class EventDocumentForm(forms.ModelForm):
    """Form that overrides the widgets for the files"""

    class Meta:
        model = models.EventDocument
        fields = (
            "name_en",
            "name_nl",
            "file_en",
            "file_nl",
            "members_only",
            "owner",
        )
        widgets = {
            "file_nl": DocumentFileInput,
            "file_en": DocumentFileInput,
        }


class MiscellaneousDocumentForm(forms.ModelForm):
    """Form that overrides the widgets for the files"""

    class Meta:
        model = models.MiscellaneousDocument
        fields = (
            "name_en",
            "name_nl",
            "file_en",
            "file_nl",
            "members_only",
        )
        widgets = {
            "file_nl": DocumentFileInput,
            "file_en": DocumentFileInput,
        }


class GeneralMeetingForm(forms.ModelForm):
    """Custom form for general meetings with a custom widget for documents"""

    class Meta:
        model = models.GeneralMeeting
        exclude = ()
        widgets = {
            "documents": admin.widgets.FilteredSelectMultiple(
                "documents", is_stacked=False
            )
        }