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

from archilance.mixins import BaseMixin
from .models import User


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


class ContractorListView(ListView):
    model = User
    template_name = 'contractor_list.html'
    context_object_name = 'contractor'


class ContractorProfileDetailView(DetailView):
    model = User
    template_name = 'contractor_profile.html'
    context_object_name = 'contractor'


class ContractorOfficeDetailView(DetailView):
    model = User
    template_name = 'contractor_office.html'
    context_object_name = 'contractor'


class CustomerProfileOpenProjectsView(BaseMixin, DetailView):
    model = User
    template_name = 'customer_profile_open_projects.html'
    context_object_name = 'customer'
    
    def get_context_data(self, **kwargs):
        c = super().get_context_data(**kwargs)
        c['projects'] = self.object.projects.filter(state='active')
        return c


class CustomerProfileTrashedProjectsView(BaseMixin, DetailView):
    model = User
    template_name = 'customer_profile_trashed_projects.html'
    context_object_name = 'customer'
    
    def get_context_data(self, **kwargs):
        c = super().get_context_data(**kwargs)
        c['projects'] = self.object.projects.filter(state='trashed')
        return c


class CustomerProfileCurrentProjectsView(BaseMixin, DetailView):
    model = User
    template_name = 'customer_profile_current_projects.html'
    context_object_name = 'customer'


class CustomerProfileReviewsView(BaseMixin, DetailView):
    model = User
    template_name = 'customer_profile_reviews.html'
    context_object_name = 'customer'