# -*- coding: utf-8 -*-
#  Copyright (c) 2019-2021 Ivan LUCAS.
#  Noethysweb, application de gestion multi-activités.
#  Distribué sous licence GNU GPL.

import datetime
from django import forms
from django.db.models import Q
from crispy_forms.helper import FormHelper
from crispy_forms.layout import Layout, Hidden
from crispy_forms.bootstrap import Field, PrependedText
from core.models import ComptaOperationBudgetaire, ComptaAnalytique, ComptaCategorie
from core.widgets import DatePickerWidget
from core.utils import utils_parametres, utils_preferences
from core.utils.utils_commandes import Commandes


class Formulaire(forms.ModelForm):
    class Meta:
        model = ComptaOperationBudgetaire
        fields = "__all__"
        widgets = {
            "date_budget": DatePickerWidget(),
        }
        help_texts = {
            "date_budget": "Cette ventilation apparaîtra sur le budget dont la période contient cette date.",
        }

    def __init__(self, *args, **kwargs):
        request = kwargs.pop("request", None)
        idcompte = kwargs.pop("categorie")
        type_operation = kwargs.pop("type", None)
        mode = kwargs.pop("mode")

        super(Formulaire, self).__init__(*args, **kwargs)
        self.helper = FormHelper()
        self.helper.form_id = "operations_budgetaires_form"
        self.helper.form_method = 'post'

        self.helper.form_class = 'form-horizontal'
        self.helper.label_class = 'col-md-2'
        self.helper.field_class = 'col-md-10'

        # Date budget
        if not self.instance.pk:
            self.fields["date_budget"].initial = datetime.date.today()

        # Analytique
        self.fields["analytique"].queryset = ComptaAnalytique.objects.filter(Q(structure__in=request.user.structures.all()) | Q(structure__isnull=True)).order_by("nom")
        if not self.instance.pk:
            analytique_defaut = utils_parametres.Get(utilisateur=request.user, nom="analytique", categorie="ventilation_tresorerie", valeur=None)
            if analytique_defaut:
                self.fields["analytique"].initial = analytique_defaut

        # Catégorie
        categories = ComptaCategorie.objects.filter(Q(type=type_operation) & (Q(structure__in=request.user.structures.all()) | Q(structure__isnull=True))).order_by("nom")
        self.fields["categorie"].choices = [(None, "---------")] + [(categorie.pk, categorie.nom) for categorie in categories]

        self.helper.layout = Layout(
            Commandes(annuler_url="{{ view.get_success_url }}"),
            Hidden("compte", value=idcompte),
            Hidden("type", value=type_operation),
            Field("date_budget"),
            Field("analytique"),
            Field("categorie"),
            Field("libelle"),
            PrependedText("montant", utils_preferences.Get_symbole_monnaie()),
        )

    def clean(self):
        if not self.cleaned_data["montant"]:
            self.add_error("montant", "Vous devez saisir un montant.")
            return
        return self.cleaned_data