from django.contrib import messages from django.contrib.auth.mixins import PermissionRequiredMixin, LoginRequiredMixin from django.core.urlresolvers import reverse, reverse_lazy from django.http import HttpResponseForbidden, HttpResponseRedirect, HttpResponse from django.shortcuts import render, get_object_or_404, redirect from django.views.generic import ListView, DetailView, CreateView, View, TemplateView, FormView from django.views.generic.base import ContextMixin from pprint import pprint, pformat import pydash as _ from .models import Project, Portfolio, Candidate, Answer, Realty, Order from archilance.mixins import BaseMixin from users.models import User from .forms import ( AnswerForm, PortfolioForm, ProjectDeleteForm, ProjectEditForm, ProjectForm, ProjectRestoreForm, ProjectTrashForm, RealtyForm, ) from .mixins import LastAccessMixin class ProjectsView(ListView): model = Project template_name = 'project_list.html' class ProjectView(LastAccessMixin, View): model = Project form_class = AnswerForm template_name = 'contractor_project_detail.html' def get_success_url(self): return reverse('projects:detail', kwargs={'pk': self.object.pk}) def get(self, request, *args, **kwargs): if request.user.is_customer(): self.template_name = 'customer_project_detail.html' project = get_object_or_404(Project, pk=kwargs.get('pk')) return render(request, self.template_name, {'project': project, 'form': self.form_class}) def post(self, request, *args, **kwargs): if not request.user.is_authenticated(): return HttpResponseForbidden() form = self.form_class(request.POST) project = get_object_or_404(Project, pk=kwargs.get('pk')) if form.is_valid(): return self.form_valid(request, form, project) return render(request, self.template_name, {'project': project, 'form': self.form_class}) def form_valid(self, request, form, project): answer = form.save(commit=False) answer.contractor = request.user answer.project = project answer.save() order = Order.objects.create(project=project) return HttpResponseRedirect(reverse('projects:detail', kwargs={'pk': project.pk})) class ProjectTestView(DetailView): model = Project template_name = 'test.html' class ProjectCreateView(PermissionRequiredMixin, View): permission_required = ('projects.add_project',) form_realty = RealtyForm form_project = ProjectForm raise_exception = True permission_denied_message = 'Вы не можете выполнить это действие' template_name = 'customer_project_create.html' def get(self, request, *args, **kwargs): form_project = self.form_project() form_realty = self.form_realty(prefix='form_2') realty_list = Realty.objects.all() return render(request, self.template_name, {'form_project': form_project, 'form_realty': form_realty, 'realty_list': realty_list}) def post(self, request, *args, **kwargs): form_project = self.form_project(request.POST) form_realty = self.form_realty(request.POST, prefix='form_2') if form_project.is_valid() and form_realty.is_valid(): return self.form_valid(request, form_project, form_realty) return render(request, self.template_name, {'form_project': form_project, 'form_realty': form_realty}) def form_valid(self, request, form_project, form_realty): realty = form_realty.save(commit=False) realty.user = request.user realty.save() project = form_project.save(commit=False) project.customer = request.user project.realty = realty project.save() return HttpResponseRedirect('/projects/') class ProjectTrashView(View): form_class = ProjectTrashForm def post(self, req, *args, **kwargs): if req.user.is_authenticated(): form = self.form_class(_.merge({}, req.POST, kwargs), req=req) if form.is_valid(): project = form.cleaned_data.get('pk') project.state = 'trashed' project.save() messages.info(req, 'Проект перемещён в корзину') else: messages.info(req, 'Произошла ошибка:
{msg}
'.format(msg=pformat(form.errors))) else: return HttpResponseForbidden('403 Forbidden') redirect_to = req.POST.get('next') return redirect(redirect_to) class ProjectRestoreView(View): form_class = ProjectRestoreForm def post(self, req, *args, **kwargs): if req.user.is_authenticated(): form = self.form_class(_.merge({}, req.POST, kwargs), req=req) if form.is_valid(): project = form.cleaned_data.get('pk') project.state = 'active' project.save() messages.info(req, 'Проект восстановлен из корзины') else: messages.info(req, 'Произошла ошибка:
{msg}
'.format(msg=pformat(form.errors))) else: return HttpResponseForbidden('403 Forbidden') redirect_to = req.POST.get('next') return redirect(redirect_to) class ProjectDeleteView(View): form_class = ProjectDeleteForm def post(self, req, *args, **kwargs): if req.user.is_authenticated(): form = self.form_class(_.merge({}, req.POST, kwargs), req=req) if form.is_valid(): project = form.cleaned_data.get('pk') project.state = 'deleted' project.save() messages.info(req, 'Проект удалён навсегда') else: messages.info(req, 'Произошла ошибка:
{msg}
'.format(msg=pformat(form.errors))) else: return HttpResponseForbidden('403 Forbidden') redirect_to = req.POST.get('next') return redirect(redirect_to) class ProjectEditView(BaseMixin, View): form_class = ProjectEditForm template_name = 'project_edit_test.html' # 'project_edit.html' def get(self, req, *args, **kwargs): if req.user.is_authenticated(): project = get_object_or_404(req.user.projects, pk=kwargs.get('pk')) form = self.form_class(instance=project, req=req) context = self.get_context_data(**kwargs) context.update({'form': form}) return render(req, self.template_name, context) else: return HttpResponseForbidden('403 Forbidden') def post(self, req, *args, **kwargs): if req.user.is_authenticated(): project = get_object_or_404(req.user.projects, pk=kwargs.get('pk')) form = self.form_class(_.merge({}, req.POST, kwargs), req=req, instance=project) if form.is_valid(): form.save() messages.info(req, 'Проект успешно отредактирован') redirect_to = req.POST.get('next') return redirect(redirect_to) else: context = self.get_context_data(**kwargs) context.update({'form': form}) messages.info(req, 'Произошла ошибка:
{msg}
'.format(msg=pformat(form.errors))) return render(req, self.template_name, context) else: return HttpResponseForbidden('403 Forbidden') class ProjectComparisonView(DetailView): model = Project template_name = 'comparison.html' def get_context_data(self, **kwargs): context = super().get_context_data() return context def add_candidate(request, answer_id, project_id): answer = Answer.objects.get(pk=answer_id) project = Project.objects.get(pk=project_id) candidate = Candidate.objects.create(answer=answer, project=project) return HttpResponseRedirect(reverse('projects:detail',args=[project_id])) class ChatView(View): template_name = 'chat.html' def get(self, request, *args, **kwargs): return render(request, self.template_name) class OfferOrderView(View): template_name = 'chattest.html' def get(self, request, *args, **kwargs): return render(request, self.template_name)