fruit_views

from django.urls import reverse_lazy
from django.views import generic

from fruitipediaApp.web.forms import CreateFruitForm, CreateCategoryForm, FruitEditForm, FruitDeleteForm
from fruitipediaApp.web.models import Fruit, Category


class FruitTemplateView(generic.TemplateView):
    template_name = 'common/index.html'


class DashBoardView(generic.ListView):
    model = Fruit
    template_name = 'common/dashboard.html'

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        fruits = Fruit.objects.all()
        context['fruits'] = fruits
        return context


class CreateFruitView(generic.CreateView):
    model = Fruit
    form_class = CreateFruitForm
    template_name = 'fruits/create-fruit.html'
    success_url = reverse_lazy('dashboard')


class CreateCategoryView(generic.CreateView):
    model = Category
    form_class = CreateCategoryForm
    template_name = 'categories/create-category.html'
    success_url = reverse_lazy('dashboard')


class FruitDetailsView(generic.DetailView):
    template_name = 'fruits/details-fruit.html'
    model = Fruit


class FruitEditView(generic.UpdateView):
    model = Fruit
    form_class = FruitEditForm
    success_url = reverse_lazy('dashboard')
    template_name = 'fruits/edit-fruit.html'


class FruitDeleteView(generic.DeleteView):
    template_name = 'fruits/delete-fruit.html'
    form_class = FruitDeleteForm
    model = Fruit
    success_url = reverse_lazy('dashboard')

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['form'] = FruitDeleteForm(instance=self.object)
        return context