proekton.com/users/views.py

from django.conf import settings
from django.contrib import messages
from django.contrib.auth.models import Group
from django.core.mail import send_mail
from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger
from django.core.urlresolvers import reverse
from django.http import HttpResponse
from django.shortcuts import render, get_object_or_404, redirect
from django.views.generic import ListView, DetailView, View, UpdateView, CreateView
from pprint import pprint, pformat
import pydash as _; _.map = _.map_; _.filter = _.filter_

from .forms import UserEditForm, ContractorFilterForm, ContractorFinancicalInfoForm
from .mixins import CheckForUserMixin
from .models import User, ContractorFinancialInfo
from archilance.mixins import BaseMixin
from common.utils import get_or_none
from projects.forms import PortfolioForm
from specializations.models import Specialization
from work_sell.forms import WorkSellForm


def send_mail_test(request):
    send_mail('Subject here', 'Here is the message.Mukhtar hello ', '', ['muhtarzubanchi05@gmail.com'], fail_silently=False)
    return HttpResponse("Mail send")


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


class ContractorFilterView(BaseMixin, View):
    template_name = 'contractor_filter.html'
    form_class = ContractorFilterForm
    
    def get(self, request, *args, **kwargs):
        form = self.form_class(request.GET, request=request)
        context = self.get_context_data(**_.merge({}, request.GET, kwargs))
        
        contractors = User.contractor_objects
        
        if form.is_valid():
            cro = form.cleaned_data.get('cro')
            specialization = form.cleaned_data.get('specialization')
            location = form.cleaned_data.get('location')
            work_type = form.cleaned_data.get('work_type')
            build_classif = form.cleaned_data.get('building_classification')
            constr_type = form.cleaned_data.get('construction_type')
            
            contractors = contractors.filter(cro=cro)
            
            if specialization:
                contractors = contractors.filter(
                    # specialization__lft__gte=specialization.lft,
                    # specialization__rght__lte=specialization.rght,
                    
                    contractor_specializations=specialization, # TODO: Honor the hierarchical data structure
                )
            
            if location:
                contractors = contractors.filter(
                    location__lft__gte=location.lft,
                    location__rght__lte=location.rght,
                )
            
            if work_type:
                contractors = contractors.filter(orders__project__work_type=work_type).distinct() # TODO: OK?
            
            if build_classif:
                contractors = contractors.filter(orders__project__realty__building_classification=build_classif)
            
            if constr_type:
                contractors = contractors.filter(orders__project__realty__construction_type=constr_type)
            
            # import code; code.interact(local=dict(globals(), **locals()))
            
            # order_by = form.cleaned_data.get('order_by')
            # last_order_by = form.cleaned_data.get('last_order_by')
            # reverse_order = form.cleaned_data.get('reverse_order')
            # 
            # if order_by:
            #     if order_by == last_order_by:
            #         reverse_order = not reverse_order
            #     else:
            #         reverse_order = False
            #     
            #     projects = projects.order_by('-%s' % order_by if reverse_order else order_by)
            #     last_order_by = order_by
            # elif last_order_by:
            #     projects = projects.order_by('-%s' % last_order_by if reverse_order else last_order_by)
            # 
            # context.update({
            #     'last_order_by': last_order_by,
            #     'reverse_order': reverse_order,
            # })
            
            contr_count = contractors.count()
            display_msg = 'Найдено %s исполнителей' % contr_count if contr_count > 0 else 'Ничего не найдено'
        else:
            display_msg = 'Пожалуйста, введите корректные данные'
            
            if form.errors:
                messages.info(request, (
                    '<p>Произошла ошибка (form)</p>'
                    '<pre>{form}</pre>'
                ).format(form=pformat(form.errors)))
        
        
        paginator = Paginator(contractors.all(), settings.PAGE_SIZE)
        page = request.GET.get('page')
        
        try:
            contractors = paginator.page(page)
        except PageNotAnInteger:
            contractors = paginator.page(1)
        except EmptyPage:
            contractors = paginator.page(paginator.num_pages)
        
        context.update({
            'form': form,
            'contractors': contractors,
            'is_paginated': True,
            'page_obj': contractors,
            'display_msg': display_msg,
        })
        
        return render(request, self.template_name, context)


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

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['worksell_form'] = self.worksell_form_class
        context['portfolio_form'] = self.portfolio_form_class
        context['resume_diploms'] = self.object.contractor_resume.resume_files.filter(type='diplom')
        context['resume_cro'] = self.object.contractor_resume.resume_files.filter(type='cro')
        return context


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'


class CustomerProfileEditView(CheckForUserMixin, View):
    form_class = UserEditForm
    template_name = 'customer_profile_edit.html'

    def get(self, request, *args, **kwargs):
        instance = get_object_or_404(User, pk=request.user.pk)
        form = self.form_class(instance=instance)
        nodes = Specialization.objects.all()
        return render(request, self.template_name, {'form': form, 'nodes': nodes})

    def post(self, request, *args, **kwargs):
        instance = request.user
        form = self.form_class(request.POST, request.FILES, instance=instance)
        if form.is_valid():
            # import code; code.interact(local=dict(globals(), **locals()))
            form.save()
            messages.info(request, 'Отправили post запрос')
            return redirect(reverse('users:customer-profile-open-projects', kwargs={'pk': request.user.pk}))
        return render(request, self.template_name, {'form': form})


class ContractorProfileEditView(CheckForUserMixin, View):
    form_class = UserEditForm
    template_name = 'contractor_profile_edit.html'

    def get(self, request, *args, **kwargs):
        instance = get_object_or_404(User, pk=request.user.pk)
        form = self.form_class(instance=instance)
        root = get_or_none(Specialization, name='_root')
        nodes = root.get_descendants('lft')

        return render(request, self.template_name, {'form': form, 'nodes': nodes})

    def post(self, request, *args, **kwargs):
        instance = request.user
        form = self.form_class(request.POST, request.FILES, instance=instance)
        if form.is_valid():
            # import code; code.interact(local=dict(globals(), **locals()))
            instance = form.save(commit=False)
            instance.save()
            messages.info(request, 'Данные успешно отредактированы')
            return redirect(reverse('users:contractor-edit', kwargs={'pk': request.user.pk}))

        return render(request, self.template_name, {'form': form})


class ContractorFinancialInfoEdit(CheckForUserMixin, View):
    template_name = 'contractor_financical_info_edit.html'
    form_class = ContractorFinancicalInfoForm

    def get(self, request, *args, **kwargs):
        instance = request.user.contractor_financial_info
        # import code; code.interact(local=dict(globals(), **locals()))
        form = self.form_class(instance=instance)
        return render(request, self.template_name, {'form': form})

    def post(self, request, *args, **kwargs):
        instance = request.user.contractor_financial_info
        form = self.form_class(request.POST, request.FILES, instance=instance)
        if form.is_valid():
            financical_instance = form.save()
            request.user.contractor_financial_info = financical_instance
            request.user.save()
            messages.info(request, 'Данные успешно отредактированы')
            return redirect(reverse('users:contractor-financical', kwargs={'pk': request.user.pk}))

        return render(request, self.template_name, {'form': form})


# import code; code.interact(local=dict(globals(), **locals()))