from django.views.generic import FormView, TemplateView

from procurement.forms import ComponentSearchForm, SupplierSearchForm
from procurement.models import Supplier, Component, Representative


class ComponentSearchView(FormView):
    template_name = 'procurement/source_components.html'
    form_class = ComponentSearchForm

    component = None
    supplier_results = None

    def get_context_data(self):
        context = super().get_context_data()

        try:
            suppliers_last_updated = Supplier.objects.latest('updated').time_since_update
        except Supplier.DoesNotExist:
            suppliers_last_updated = ''

        try:
            representatives_last_updated = Representative.objects.latest('updated').time_since_update
        except Representative.DoesNotExist:
            representatives_last_updated = ''

        try:
            components_last_updated = Component.objects.latest('updated').time_since_update
        except Component.DoesNotExist:
            components_last_updated = ''

        context.update({
            'page_name': 'Component Search',
            'component': self.component,
            'supplier_results': self.supplier_results,
            'supplier_count': Supplier.objects.all().count(),
            'suppliers_last_updated': suppliers_last_updated,
            'component_count': Component.objects.all().count(),
            'components_last_updated': components_last_updated,
            'representative_count': Representative.objects.all().count(),
            'representatives_last_updated': representatives_last_updated,
        })
        return context

    def form_valid(self, form):
        self.component = form.cleaned_data['component']
        if self.component:
            self.supplier_results = self.component.suppliers.filter(is_authorized=True)

        return super(ComponentSearchView, self).get(self.request)

class SupplierSearchView(FormView):
    template_name = "procurement/view_suppliers.html"
    form_class = SupplierSearchForm

    supplier = None
    component_form = None

    def form_valid(self, form):
        self.supplier = form.cleaned_data['supplier']
        if self.supplier:
            self.component_form = ComponentSearchForm()
            self.component_form.fields["component"].queryset = self.supplier.components
        return super(SupplierSearchView, self).get(self.request)

    def get_context_data(self):
        context = super().get_context_data()
        context["supplier"] = self.supplier
        context["component_form"] = self.component_form
        context["page_name"] = "View Supplier Details"
        return context

class DocumentationView(TemplateView):
    template_name = 'procurement/documentation.html'

    def get_context_data(self):
        context = super().get_context_data()
        context.update({
            'page_name': 'Documentation',
        })
        return context