# dashboard/forms.py

from __future__ import annotations

from typing import TYPE_CHECKING

from django import forms

if TYPE_CHECKING:
    pass

from .models import Dashboard, DataSource


class DataSourceUploadForm(forms.ModelForm):
    """Form for uploading CSV files"""

    class Meta:
        model = DataSource
        fields = ["name", "description", "file"]

    def __init__(self, *args, **kwargs):
        self.company = kwargs.pop("company", None)
        super().__init__(*args, **kwargs)

    def save(self, commit=True):
        instance = super().save(commit=False)
        if self.company:
            instance.company = self.company
        if commit:
            instance.save()
        return instance


class DashboardForm(forms.ModelForm):
    """Form for creating and editing dashboards"""

    class Meta:
        model = Dashboard
        fields = ["name", "description", "data_sources"]

    def __init__(self, *args, **kwargs):
        self.company = kwargs.pop("company", None)
        super().__init__(*args, **kwargs)

        if self.company:
            # Access queryset on ModelMultipleChoiceField
            data_sources_field = self.fields["data_sources"]  # type: ignore[assignment]
            data_sources_field.queryset = DataSource.objects.filter(company=self.company)  # type: ignore[attr-defined]

    def save(self, commit=True):
        instance = super().save(commit=False)
        if self.company:
            instance.company = self.company
        if commit:
            instance.save()
            self.save_m2m()
        return instance