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 itertools import natsort import pydash as _; _.map = _.map_; _.filter = _.filter_ from .forms import UserEditForm, ContractorFilterForm, ContractorFinancicalInfoForm from .mixins import CheckForUserMixin from .models import User, Team, 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 = teams = None 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') party_types = form.cleaned_data.get('party_types') get_contractors = get_teams = None if not party_types: get_contractors = get_teams = True elif party_types == 'contractors': get_contractors = True elif party_types == 'teams': get_teams = True if get_contractors: contractors = User.contractor_objects.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) # contr_count = contractors.count() # display_msg = 'Найдено %s исполнителей' % contr_count if contr_count > 0 else 'Ничего не найдено' if get_teams: teams = Team.objects.all() # team_count = teams.count() # display_msg = 'Найдено %s команд' % team_count if team_count > 0 else 'Ничего не найдено' display_msg = 'Результаты поиска' else: display_msg = 'Пожалуйста, введите корректные данные' if form.errors: messages.info(request, ( '

Произошла ошибка (form)

' '
{form}
' ).format(form=pformat(form.errors))) coll = tuple(itertools.chain(contractors, teams)) coll = natsort.natsorted(coll, key=lambda obj: getattr(obj, 'username', None) or getattr(obj, 'name')) paginator = Paginator(coll, settings.PAGE_SIZE) page = request.GET.get('page') try: coll = paginator.page(page) except PageNotAnInteger: coll = paginator.page(1) except EmptyPage: coll = paginator.page(paginator.num_pages) context.update({ 'form': form, 'coll': coll, 'is_paginated': True, 'page_obj': coll, '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()))