from django.shortcuts import render
from django.contrib.auth.models import Group
from django.views.generic import ListView, DetailView
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