from django.shortcuts import render
from django.contrib.auth.models import Group
from django.views.generic import ListView, DetailView, View, UpdateView
from .models import User


class UserListView(ListView):
    model = User
    template_name = 'users/users_list.html'
    context_object_name = 'users'


class UserInfoListView(ListView):
    model = User
    template_name = 'users/users_info_list.html'

    def get_context_data(self, **kwargs):
        context = super(UserInfoListView, self).get_context_data(**kwargs)
        context['groups'] = Group.objects.all()
        context['performers'] = User.perform_objects.all()
        context['customers'] = User.customers_objects.all()
        return context

class UserDetailView(DetailView):
    model = User



class UserView(View):
    template_name = 'users/contractor.html'

    def get(self, request, pk):
        user = User.objects.get(pk=pk)
        user_group = user.groups.all().first()
        if 'Заказчики' in user_group.name:
            self.template_name = 'users/customer.html'

        return render(request, self.template_name, {'user': user})
        # import code; code.interact(local=dict(globals(), **locals()))


class UserUpdateView(UpdateView):
    model = User

# class ContractorDetailView(DetailView):
#     model = User
#     template_name = 'users/contractor.html'
#
#     def get_queryset(self):
#         qs = super().get_queryset()
#         return User.contractor_objects.get(self.pk)


class CustomerDetailView(DetailView):
    model = User