import json from pprint import pformat import natsort import pydash as _ from django.conf import settings from django.contrib import messages from django.contrib.auth.mixins import LoginRequiredMixin from django.core.exceptions import PermissionDenied from django.core.files.base import ContentFile from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger from django.core.urlresolvers import reverse, reverse_lazy from django.db.models import Q from django.http import HttpResponseForbidden, JsonResponse, HttpResponseRedirect, HttpResponse, Http404 from django.shortcuts import render, get_object_or_404, redirect from django.views.generic import DetailView, CreateView, DeleteView, View, UpdateView, TemplateView from hitcount.models import HitCount from hitcount.views import HitCountMixin import re from archilance import util from archilance.mixins import BaseMixin from common.mixins import NoCsrfMixin from users.models import User, Team from work_sell.models import Picture, WorkSell, WorkSellPhoto from ratings.models import HistoryRating from .models import ( Answer, AnswerFile, AnswerMessage, Arbitration, BuildingClassfication, Candidate, Order, Portfolio, PortfolioPhoto, Project, ProjectFile, TERM_TYPE_MORPHS, Realty ) from .forms import ( ContractorPortfolioTrashForm, CustomerProjectDeleteForm, CustomerProjectEditForm, CustomerProjectEditFormNew, CustomerProjectRestoreForm, CustomerProjectTrashForm, PortfolioEditForm, PortfolioForm, ProjectAnswerForm, ProjectAnswerMessageForm, ProjectFilterForm, ProjectFilterRealtyForm, ProjectWorkTypeSuggestionForm, RealtyForm, RealtyFormNew ) from specializations.models import Specialization from common.models import Location _.map = _.map_ _.filter = _.filter_ class ProjectDetailWithAnswerView(BaseMixin, View): form_class = ProjectAnswerForm template_name = 'project_detail.html' def get(self, request, *args, **kwargs): context = self.get_context_data(**_.merge({}, request.GET, kwargs)) context.update({'TERM_TYPE_MORPHS': TERM_TYPE_MORPHS}) project = get_object_or_404(Project, pk=kwargs.get('pk')) context.update({'project': project}) hit_count = HitCount.objects.get_for_object(project) HitCountMixin.hit_count(request, hit_count) if request.user.is_authenticated() and request.user.is_contractor(): project_answers = project.answers.filter(rejected=False) contractor = request.user answer = _.first(_.filter(project_answers, lambda a: a.author == contractor)) if not answer: team = util.get_related_or_none(contractor, 'team') if team: answer = _.first(_.filter(project_answers, lambda a: a.author == team)) context.update({'answer': answer}) # answer.messages.update() if not answer: team = util.get_related_or_none(contractor, 'team') if team: context.update({'can_answer_as_team': True}) if request.GET.get('answer_as_team') == 'on': context.update({'answer_as_team': True}) form = self.form_class(request=request, answer_as_team=True, project=project) else: form = self.form_class(request=request, project=project) context.update({'form': form}) return render(request, self.template_name, context) def post(self, request, *args, **kwargs): if request.user.is_authenticated() and request.user.is_contractor(): context = self.get_context_data(**kwargs) answer_as_team = None project = get_object_or_404(Project, pk=kwargs.get('pk')) # TODO: Does this work? if request.POST.get('answer_as_team') == 'on': answer_as_team = True # Check for duplicate answers: if answer_as_team and util.has_related(request.user, 'team'): if project.answers.filter(object_id=request.user.team.pk, content_type__model='team').exists(): raise PermissionDenied('Повторный отклик') else: if project.answers.filter(object_id=request.user.pk, content_type__model='user').exists(): raise PermissionDenied('Повторный отклик') if answer_as_team: form = self.form_class(request.POST, request=request, answer_as_team=True, project=project) else: form = self.form_class(request.POST, request=request, project=project) project = get_object_or_404(Project, pk=kwargs.get('pk')) context.update({'project': project}) if form.is_valid(): answer = form.save(commit=False) answer.project = project answer.author = request.user.team if answer_as_team else request.user answer.save() form.save_m2m() for file in request.FILES.getlist('new_files'): if len(file.name) <= 255: AnswerFile.objects.create(file=file, name=file.name, answer=answer) message = AnswerMessage(text=form.cleaned_data.get('text'), is_sender_customer=False) message.answer = answer message.contractor_or_team = answer.author message.save() messages.info(request, 'Отклик успешно размещён') redirect_to = request.POST.get('next') return redirect(redirect_to) else: if form.errors: messages.info(request, ( '
Произошла ошибка (form)
' '{form}'
).format(form=pformat(form.errors)))
context.update({'form': form})
return render(request, self.template_name, context)
else:
raise PermissionDenied
class ProjectAnswerCreateMessageView(BaseMixin, View):
form_class = ProjectAnswerMessageForm
def dispatch(self, request, *args, **kwargs):
if request.user.is_authenticated():
return super().dispatch(request, *args, **kwargs)
else:
raise PermissionDenied
def post(self, request, *args, **kwargs):
form = self.form_class(request.POST, request=request)
answer = get_object_or_404(Answer, pk=kwargs.get('pk'))
if form.is_valid():
message = form.save(commit=False)
if request.user.is_contractor():
message.answer = answer
message.is_sender_customer = False
if isinstance(answer.author, User) and answer.author == request.user:
message.contractor_or_team = request.user
elif isinstance(answer.author, Team):
team = util.get_related_or_none(request.user, 'team')
if team and answer.author == team:
message.contractor_or_team = team
elif request.user.is_customer():
message.answer = answer
message.is_sender_customer = True
message.save()
form.save_m2m()
messages.info(request, 'Сообщение успешно размещено')
else:
if form.errors:
messages.info(request, (
'Произошла ошибка (form)
' '{form}'
).format(form=pformat(form.errors)))
redirect_to = request.POST.get('next')
return redirect(redirect_to)
class RejectProjectAnswerView(NoCsrfMixin, LoginRequiredMixin, BaseMixin, View):
def post(self, request, *args, **kwargs):
if request.user.is_contractor():
answer = None
contractor_answer = util.get_or_none(request.user.contractor_answers, pk=kwargs.get('pk'))
if contractor_answer:
answer = contractor_answer
elif request.user.team:
answer = util.get_or_none(request.user.team.answers, pk=kwargs.get('pk'))
if not answer:
raise Http404
elif request.user.is_customer():
project = get_object_or_404(request.user.customer_projects, answers__pk=kwargs.get('pk'))
answer = get_object_or_404(project.answers, pk=kwargs.get('pk'))
answer.rejected = True
answer.save()
candidate = Candidate.objects.filter(answer=answer)
if candidate:
candidate.delete()
messages.info(request, 'Успешный отказ от проекта')
redirect_to = request.POST.get('next')
return redirect(redirect_to)
class RestoreProjectAnswerView(NoCsrfMixin, LoginRequiredMixin, BaseMixin, View):
def post(self, request, *args, **kwargs):
if request.user.is_contractor():
answer = None
contractor_answer = util.get_or_none(request.user.contractor_answers, pk=kwargs.get('pk'))
if contractor_answer:
answer = contractor_answer
elif request.user.team:
answer = util.get_or_none(request.user.team.answers, pk=kwargs.get('pk'))
if not answer:
raise Http404
elif request.user.is_customer():
project = get_object_or_404(request.user.customer_projects, answers__pk=kwargs.get('pk'))
answer = get_object_or_404(project.answers, pk=kwargs.get('pk'))
answer.rejected = False
answer.save()
messages.info(request, 'Проект успешно восстановлен')
redirect_to = request.POST.get('next')
return redirect(redirect_to)
def build_query(id_str, _class, *args):
query = Q()
id_list = tuple(filter(None, re.split(r'\s|,|;', id_str)))
for id in id_list:
obj = _class.objects.get(id=id)
kwargs = dict(zip(args, (obj.lft, obj.rght)))
query |= Q(**kwargs)
return query
class ProjectFilterView(BaseMixin, View):
template_name = 'project_filter.html'
# form_class = ProjectFilterForm
# renderer_classes = (TemplateHTMLRenderer,)
PROJECT_ORDER_CHOICES = ( # "Упорядочить по"...
('name', 'названию'),
('budget', 'цене'),
('created', 'дате размещения'),
('views', 'просмотрам'),
)
def filter(self, projects, data):
# print("filter data = ", data)
specialization = data.get('specialization')
keywords = data.get('keywords')
building_classification = data.get('building_classification')
construction_type = data.get('construction_type')
location = data.get('location')
cro = data.get('cro')
if keywords:
# print("keywords = ", keywords)
keywords = tuple(filter(None, re.split(r'\s|,|;', keywords)))
for k in keywords:
projects = projects.filter(Q(name__icontains=k.lower()) | Q(text__icontains=k.lower()))
if specialization:
query = build_query(specialization, Specialization,
'specialization__lft__gte', 'specialization__rght__lte')
projects = projects.filter(query)
if building_classification:
query = build_query(building_classification, BuildingClassfication,
'realty__building_classification__lft__gte',
'realty__building_classification__rght__lte')
projects = projects.filter(query)
if construction_type:
projects = projects.filter(realty__construction_type__in=
tuple(filter(None, re.split(r'\s|,|;', construction_type))))
if location:
query = build_query(location, Location,
'realty__location__lft__gte',
'realty__location__rght__lte')
projects = projects.filter(query)
if cro:
projects = projects.filter(cro=True)
return projects
def sort_by(self, projects):
# order_by = request.GET.get('order_by')
# last_order_by = request.GET.get('last_order_by')
# reverse_order = request.GET.get('reverse_order')
# ord = None
# manual_sort = None
#
# # ORDERS
# if order_by:
# reverse_order = not reverse_order if order_by == last_order_by else False
# ord = order_by
# last_order_by = ord
# elif last_order_by:
# ord = last_order_by
#
# if ord and ord == 'views':
# projects = natsort.natsorted(projects.all(), key=lambda p: p.hit_count.hits, reverse=reverse_order)
# manual_sort = True
# elif ord:
# projects = projects.order_by('-%s' % ord if reverse_order else ord)
return projects
def pagination(self, projects, page):
paginator = Paginator(projects, settings.PAGE_SIZE)
# print("pag page = ", page)
try:
projects = paginator.page(page)
except PageNotAnInteger:
projects = paginator.page(1)
except EmptyPage:
projects = paginator.page(paginator.num_pages)
return projects
def get_context(self, request):
context = request.dict()
projects = Project.objects.filter(state='active')
projects = self.filter(projects, request)
num_results = projects.count()
projects = self.sort_by(projects)
projects = self.pagination(projects, request.get("page"))
has_additional_fields = bool(context.get("building_classification") or context.get(
"construction_type") or context.get("location") or context.get("cro"))
context.update({
'choices': self.PROJECT_ORDER_CHOICES,
'projects': projects,
'is_paginated': True,
'page_obj': projects,
'num_results': num_results,
'has_additional_fields': has_additional_fields
})
return context
def get(self, request):
return render(request, self.template_name, self.get_context(request.GET))
def post(self, request):
return render(request, 'partials/inc-projects-results.html', self.get_context(request.POST))
# class CustomerProjectCreateView(BaseMixin, View):
# form_class = CustomerProjectEditForm
# realty_form = RealtyForm
# work_type_suggestion_form = ProjectWorkTypeSuggestionForm
# template_name = 'customer_project_create.html'
#
# def dispatch(self, request, *args, **kwargs):
# if request.user.is_authenticated() and request.user.is_customer():
# return super().dispatch(request, *args, **kwargs)
# else:
# raise PermissionDenied
#
# def get(self, request, *args, **kwargs):
# context = self.get_context_data(**_.merge({}, request.GET, kwargs))
#
# form = self.form_class(request=request)
# realty_form = self.realty_form(request=request, prefix='realty_form')
# work_type_suggestion_form = self.work_type_suggestion_form(request=request, prefix='work_type_suggestion')
#
# context.update({
# 'form': form,
# 'realty_form': realty_form,
# 'work_type_suggestion_form': work_type_suggestion_form,
# })
#
# return render(request, self.template_name, context)
#
# def post(self, request, *args, **kwargs):
# form = self.form_class(request.POST,
# request=request) # Passing `request.FILES` seems unnecessary here. Files are added manually below
#
# form.is_valid()
# realty = form.cleaned_data.get('realty')
#
# if realty:
# realty_form = self.realty_form(request.POST, instance=realty, request=request, prefix='realty_form')
# else:
# realty_form = self.realty_form(request.POST, request=request, prefix='realty_form')
#
# if form.is_valid() and realty_form.is_valid():
# project = form.save(commit=False)
# project.customer = request.user
# project.save()
# form.save_m2m()
#
# Order.objects.create(project=project, secure=project.deal_type == 'secure_deal')
#
# for file in request.FILES.getlist('new_files'):
# ProjectFile.objects.create(file=file, project=project)
#
# if realty:
# realty_form.save()
# else:
# realty = realty_form.save(commit=False)
# realty.user = request.user
# realty.save()
# realty_form.save_m2m()
#
# project.realty = realty # Connect a realty with a project
# project.save()
#
# messages.info(request, 'Проект успешно создан')
# redirect_to = reverse('projects:detail', kwargs={'pk': project.pk})
# return redirect(redirect_to)
# else:
# if form.errors:
# messages.info(request, (
# 'Произошла ошибка (form)
' # '{form}'
# ).format(form=pformat(form.errors)))
#
# if realty_form and realty_form.errors:
# messages.info(request, (
# 'Произошла ошибка (realty_form)
' # '{realty_form}'
# ).format(realty_form=pformat(realty_form.errors)))
#
# context = self.get_context_data(**kwargs)
# context.update({'form': form, 'realty_form': realty_form})
# return render(request, self.template_name, context)
class CustomerProjectCreateView(BaseMixin, View):
form_class = CustomerProjectEditFormNew
realty_form = RealtyFormNew
work_type_suggestion_form = ProjectWorkTypeSuggestionForm
template_name = 'customer_project_create.html'
def dispatch(self, request, *args, **kwargs):
if request.user.is_authenticated() and request.user.is_customer():
return super().dispatch(request, *args, **kwargs)
else:
raise PermissionDenied
def get(self, request, *args, **kwargs):
context = self.get_context_data(**_.merge({}, request.GET, kwargs))
form = self.form_class(request=request)
realty_form = self.realty_form(request=request, prefix='realty_form')
work_type_suggestion_form = self.work_type_suggestion_form(request=request, prefix='work_type_suggestion')
context.update({
'form': form,
'realty_form': realty_form,
'work_type_suggestion_form': work_type_suggestion_form,
})
return render(request, self.template_name, context)
def post(self, request, *args, **kwargs):
form = self.form_class(request.POST, request=request) # Passing `request.FILES` seems unnecessary here. Files are added manually below
print('POST = ', request.POST)
form.is_valid()
realty = form.cleaned_data.get('realty')
if realty:
realty_form = self.realty_form(request.POST, instance=realty, request=request, prefix='realty_form')
else:
realty_form = self.realty_form(request.POST, request=request, prefix='realty_form')
if form.is_valid() and realty_form.is_valid():
project = form.save(commit=False)
project.customer = request.user
project.save()
form.save_m2m()
Order.objects.create(project=project, secure=project.deal_type == 'secure_deal')
for file in request.FILES.getlist('new_files'):
ProjectFile.objects.create(file=file, project=project)
if realty:
realty_form.save()
else:
realty = realty_form.save(commit=False)
realty.user = request.user
print('new_realty_name = ', request.POST.get('new_realty_name'))
if not request.POST.get('new_realty_name'):
realty.is_virtual = True
realty.name = request.POST.get('new_realty_name')
realty.save()
realty_form.save_m2m()
project.realty = realty # Connect a realty with a project
project.save()
messages.info(request, 'Проект успешно создан')
redirect_to = reverse('projects:detail', kwargs={'pk': project.pk})
return redirect(redirect_to)
else:
if form.errors:
messages.info(request, (
'Произошла ошибка (form)
' '{form}'
).format(form=pformat(form.errors)))
if realty_form and realty_form.errors:
messages.info(request, (
'Произошла ошибка (realty_form)
' '{realty_form}'
).format(realty_form=pformat(realty_form.errors)))
context = self.get_context_data(**kwargs)
context.update({'form': form, 'realty_form': realty_form})
return render(request, self.template_name, context)
class CustomerProjectEditView(BaseMixin, View):
form_class = CustomerProjectEditForm
realty_form = RealtyForm
work_type_suggestion_form = ProjectWorkTypeSuggestionForm
template_name = 'customer_project_edit.html'
def dispatch(self, request, *args, **kwargs):
if request.user.is_authenticated() and request.user.is_customer():
# Prevent editing when project's taken:
project = get_object_or_404(request.user.customer_projects, pk=kwargs.get('pk'))
if project.order.contractor or project.order.team:
raise PermissionDenied('Заказ уже находится в работе. Вы не можете его редактировать.')
else:
return super().dispatch(request, *args, **kwargs)
raise PermissionDenied
def get(self, request, *args, **kwargs):
project = get_object_or_404(request.user.customer_projects, pk=kwargs.get('pk'))
form = self.form_class(instance=project, request=request)
work_type_suggestion_form = self.work_type_suggestion_form(request=request, prefix='work_type_suggestion')
realty = project.realty
if realty:
realty_form = self.realty_form(instance=project.realty, request=request, prefix='realty_form')
else:
realty_form = self.realty_form(request=request, prefix='realty_form')
context = self.get_context_data(**_.merge({}, request.GET, kwargs))
context.update({
'form': form,
'realty_form': realty_form,
'work_type_suggestion_form': work_type_suggestion_form,
})
return render(request, self.template_name, context)
def post(self, request, *args, **kwargs):
project = get_object_or_404(request.user.customer_projects, pk=kwargs.get('pk'))
form = self.form_class(request.POST, request.FILES, request=request, instance=project)
form.is_valid()
realty = form.cleaned_data.get('realty')
if realty:
realty_form = self.realty_form(request.POST, instance=realty, request=request, prefix='realty_form')
else:
realty_form = self.realty_form(request.POST, request=request, prefix='realty_form')
if form.is_valid() and realty_form.is_valid():
project = form.save(commit=False)
project.customer = request.user
project.files = form.cleaned_data.get(
'files') # TODO: Should we somehow get rid of this explicit assignment?
project.save()
form.save_m2m()
for file in request.FILES.getlist('new_files'):
proj_file = ProjectFile.objects.create(file=file, project=project)
proj_file.save()
if realty:
realty_form.save()
else:
realty = realty_form.save(commit=False)
realty.user = request.user
realty.save()
realty_form.save_m2m()
project.realty = realty # Connect a realty with a project
project.save()
messages.info(request, 'Проект успешно отредактирован')
redirect_to = request.POST.get('next')
return redirect(redirect_to)
else:
if form.errors:
messages.info(request, (
'Произошла ошибка (form)
' '{form}'
).format(form=pformat(form.errors)))
if realty_form and realty_form.errors:
messages.info(request, (
'Произошла ошибка (realty_form)
' '{realty_form}'
).format(realty_form=pformat(realty_form.errors)))
context = self.get_context_data(**kwargs)
context.update({'form': form, 'realty_form': realty_form})
return render(request, self.template_name, context)
class ContractorAnswerArchiveView(View):
def post(self, request, *args, **kwargs):
project_pk = request.POST.get('project_pk')
user_pk = request.POST.get('user_pk')
answer = Answer.objects.filter(project_id=project_pk, object_id=user_pk, content_type__model='user').first()
answer.is_archive = True
answer.save()
redirect_to = request.POST.get('next')
return redirect(redirect_to)
class ContractorPortfolioTrashView(View):
form_class = ContractorPortfolioTrashForm
def post(self, request, *args, **kwargs):
form = self.form_class(_.merge({}, request.POST, kwargs), request=request)
if form.is_valid():
portfolio = form.cleaned_data.get('pk')
portfolio.delete()
messages.info(request, 'Портфолио удалено')
else:
messages.info(request, 'Произошла ошибка: {msg}'.format(msg=pformat(form.errors)))
redirect_to = request.POST.get('next')
return redirect(redirect_to)
class CustomerProjectTrashView(View):
form_class = CustomerProjectTrashForm
def dispatch(self, request, *args, **kwargs):
if request.user.is_authenticated() and request.user.is_customer():
# Prevent editing when project's taken:
project = get_object_or_404(request.user.customer_projects, pk=kwargs.get('pk'))
if project.order.contractor or project.order.team:
raise PermissionDenied('Заказ уже находится в работе. Вы не можете его переместить в корзину.')
else:
return super().dispatch(request, *args, **kwargs)
raise PermissionDenied
def post(self, req, *args, **kwargs):
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)))
redirect_to = req.POST.get('next')
return redirect(redirect_to)
class CustomerProjectRestoreView(View):
form_class = CustomerProjectRestoreForm
def dispatch(self, request, *args, **kwargs):
if request.user.is_authenticated() and request.user.is_customer():
return super().dispatch(request, *args, **kwargs)
else:
raise PermissionDenied
def post(self, req, *args, **kwargs):
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)))
redirect_to = req.POST.get('next')
return redirect(redirect_to)
class CustomerProjectDeleteView(View):
form_class = CustomerProjectDeleteForm
def dispatch(self, request, *args, **kwargs):
if request.user.is_authenticated() and request.user.is_customer():
return super().dispatch(request, *args, **kwargs)
else:
raise PermissionDenied
def post(self, req, *args, **kwargs):
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)))
redirect_to = req.POST.get('next')
return redirect(redirect_to)
class ProjectComparisonView(DetailView):
model = Project
template_name = 'comparison.html'
def get_context_data(self, **kwargs):
c = super().get_context_data(**kwargs)
c['TERM_TYPE_MORPHS'] = TERM_TYPE_MORPHS
return c
def add_candidate(request, answer_id, project_id):
answer = Answer.objects.get(pk=answer_id)
project = Project.objects.get(pk=project_id)
count_answers = Candidate.objects.filter(project=project).count()
count_answers += 1
candidate = Candidate.objects.filter(answer=answer).first()
if not candidate:
Candidate.objects.create(answer=answer, project=project, position=count_answers)
redirect_to = '%s%s' % (reverse('projects:detail', kwargs={'pk': project_id}), '#answers')
return redirect(redirect_to)
class CandidateDeleteView(DeleteView):
model = Candidate
def get_success_url(self):
return reverse('projects:comparison', kwargs={'pk': self.object.project_id})
def get(self, *args, **kwargs):
return self.post(*args, **kwargs)
def sort_candidates(request):
if request.is_ajax():
items = request.POST.getlist('items[]')
i = 1
for item in items:
candidate = Candidate.objects.get(pk=item)
candidate.position = i
candidate.save()
i += 1
data = {
'success': 'ok',
}
return HttpResponse(json.dumps(data), content_type='application/json')
class CustomerOfferOrderView(View):
template_name = 'chattest.html'
def post(self, request, *args, **kwargs):
project_id = kwargs.get('project_id')
answer_id = kwargs.get('answer_id')
try:
project = Project.objects.get(pk=project_id)
except Project.DoesNotExist:
project = None
try:
answer = Answer.objects.get(pk=answer_id)
except Answer.DoesNotExist:
answer = None
order = project.order
status = None
if not order.contractor and not order.team:
if isinstance(answer.author, User):
order.contractor = answer.author
order.save()
status = True
elif isinstance(answer.author, Team):
order.team = answer.author
order.save()
status = True
if status:
hs_rating = HistoryRating()
hs_rating.user = request.user
hs_rating.rating = 1
hs_rating.type = 'choice_contractor'
hs_rating.description = 'Заказчик выбрал исполнителя'
hs_rating.save()
redirect_url = reverse('chat:chat-user') + '#order' + str(order.pk)
return HttpResponseRedirect(redirect_url)
def contractor_portfolio_create(request): # TODO: pekopt: shit. rewrite using generic
if request.is_ajax():
form = PortfolioForm(data=request.POST)
# import code; code.interact(local=dict(globals(), **locals()))
if form.is_valid():
duplicate = form.cleaned_data.get('duplicate')
instance = form.save(commit=False)
instance.user = request.user
instance.save()
if duplicate:
work_sell = WorkSell()
work_sell.name = instance.name
work_sell.budget = instance.budget
work_sell.building_classification = instance.building_classification
work_sell.construction_type = instance.construction_type
work_sell.currency = instance.currency
work_sell.description = instance.description
work_sell.term = instance.term
work_sell.term_type = instance.term_type
work_sell.contractor = instance.user
work_sell.save()
images_ids = request.POST.get('images_ids').split(';')[:-1]
for pk in images_ids:
picture = Picture.objects.get(pk=pk)
temp_file = ContentFile(picture.file.read())
temp_file.name = picture.file.name
p_photo = PortfolioPhoto()
p_photo.img = temp_file
p_photo.portfolio = instance
p_photo.save()
if duplicate:
w_photo = WorkSellPhoto()
w_photo.img = temp_file
w_photo.worksell = work_sell
w_photo.save()
data = {'status': 'ok'}
else:
data = {'status': 'no', 'form_errors': form.errors}
return HttpResponse(json.dumps(data), content_type='application/json')
else:
raise Http404
class ContractorPortfolioUpdateView(BaseMixin, UpdateView):
model = Portfolio
form_class = PortfolioEditForm
template_name = 'contractor_portfolio_edit.html'
def get_success_url(self):
return reverse('projects:contractor-portfolio-detail', kwargs={'pk': self.object.pk})
def form_valid(self, form):
portfolio = form.instance
photos = form.cleaned_data['photos']
# # Doesn't work:
#
# portfolio.photos = photos
# portfolio.save()
PortfolioPhoto.objects.filter(portfolio=portfolio).delete()
for photo in photos:
PortfolioPhoto.objects.create(img=photo.img, portfolio=portfolio)
live_images = form.cleaned_data['live_images']
for live_image in live_images:
new_image = ContentFile(live_image.file.read())
new_image.name = live_image.file.name
PortfolioPhoto.objects.create(img=new_image, portfolio=portfolio)
live_image.file.delete()
live_image.delete()
return super().form_valid(form)
class PortfolioDelete(DeleteView):
model = Portfolio
success_url = reverse_lazy('users:contractor-profile')
class PortfolioDetail(DetailView):
model = Portfolio
template_name = 'portfolio_detail.html'
def dispatch(self, request, *args, **kwargs):
if not self.get_object().user:
return HttpResponseForbidden('404 Not Found')
return super().dispatch(request, *args, **kwargs)
class ArbitrationCreateView(CreateView):
model = Arbitration
fields = (
'order',
'user',
'text',
)
def form_valid(self, form):
if self.request.is_ajax():
self.object = form.save()
data = {
'status': 'ok',
}
return JsonResponse(data)
return super().form_valid(form)
def form_invalid(self, form):
if self.request.is_ajax():
return JsonResponse(form.errors, status=400)
return super().form_invalid(form)
class ProjectWorkTypeSuggestionView(View):
form_class = ProjectWorkTypeSuggestionForm
template_name = 'customer_project_work_type_suggestion.html'
# def get(self, request, *args, **kwargs):
# form = self.form_class(request=request, prefix='work_type_suggestion')
# context = {'form': form}
# return render(request, self.template_name, context)
def post(self, request, *args, **kwargs):
form = self.form_class(request.POST, request=request, prefix='work_type_suggestion')
if form.is_valid():
form.save()
return JsonResponse({'status': 'success'})
else:
form_errors = {'.-error-%s' % bfield.html_name: bfield.errors for bfield in form}
return JsonResponse({
'status': 'error',
'form_errors': form_errors,
})
# import code; code.interact(local=dict(globals(), **locals()))