from django.views.generic import TemplateView
from django.utils.translation import ugettext_lazy as _

from cart.models import Client
from core.models import Certificate


class IndexView(TemplateView):
    template_name = 'index/index.html'
    title = _('Русские программы')

    def get_clients(self):
        return Client.active.only('name', 'image').all()

    def get_certs(self):
        return Certificate.active.only('name', 'image').all()

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['title'] = self.title
        context['clients'] = self.get_clients()
        context['certs'] = self.get_certs()
        return context