From 118009aa185320476d2ef88684195dbd0a1570d5 Mon Sep 17 00:00:00 2001 From: ArturBaybulatov Date: Fri, 26 Aug 2016 20:44:11 +0300 Subject: [PATCH] #ARC-9 --- api/urls.py | 4 +- .../management/commands/generate_projects.py | 2 +- archilance/util.py | 10 +++++ projects/models.py | 5 ++- projects/views.py | 12 +++--- users/models.py | 10 ++--- users/templates/contractor_office.html | 30 +++++++-------- users/views.py | 38 ++++++++++++++----- .../migrations/0006_remove_worksell_team.py | 19 ++++++++++ work_sell/models.py | 1 - 10 files changed, 89 insertions(+), 42 deletions(-) create mode 100644 work_sell/migrations/0006_remove_worksell_team.py diff --git a/api/urls.py b/api/urls.py index 6fc2a21..e0a3b38 100755 --- a/api/urls.py +++ b/api/urls.py @@ -8,6 +8,7 @@ from .views import ( LocationViewSet, MessageViewSet, NoteViewSet, + OrderViewSet, PortfolioPhotoViewSet, PortfolioViewSet, ProjectViewSet, @@ -17,7 +18,6 @@ from .views import ( StageViewSet, TeamViewSet, UserViewSet, - OrderViewSet, ) @@ -30,10 +30,10 @@ router.register(r'documents', DocumentViewSet) router.register(r'locations', LocationViewSet) router.register(r'message', MessageViewSet) router.register(r'note', NoteViewSet) +router.register(r'orders', OrderViewSet) router.register(r'portfolio-photos', PortfolioPhotoViewSet) router.register(r'portfolios', PortfolioViewSet) router.register(r'projects', ProjectViewSet) -router.register(r'orders', OrderViewSet) router.register(r'realties', RealtyViewSet) router.register(r'reviews', ReviewViewSet) router.register(r'specializations', SpecializationViewSet) diff --git a/archilance/management/commands/generate_projects.py b/archilance/management/commands/generate_projects.py index 099330c..49daa9b 100644 --- a/archilance/management/commands/generate_projects.py +++ b/archilance/management/commands/generate_projects.py @@ -78,7 +78,7 @@ class Command(BaseCommand): project=project, contractor=_.sample((None, User.contractor_objects.order_by('?').first())), secure=_.sample((True, False)), - status=_.sample((True, False)), + status=_.sample(Order.STATUSES)[0], ) return project diff --git a/archilance/util.py b/archilance/util.py index b919d9f..361b400 100644 --- a/archilance/util.py +++ b/archilance/util.py @@ -1,4 +1,5 @@ from django.core import validators +from django.core.exceptions import ObjectDoesNotExist from django.shortcuts import _get_queryset from django.utils import timezone from pprint import pprint, pformat @@ -54,6 +55,15 @@ def get_or_none(klass, *args, **kwargs): return None +def get_related_or_none(obj, attr): + try: + rel = getattr(obj, attr) + except ObjectDoesNotExist: + rel = None + + return rel + + def get_attr_or_none(klass, *args, attr=None, **kwargs): object = get_or_none(klass, *args, **kwargs) diff --git a/projects/models.py b/projects/models.py index 7c1e158..22e32a1 100644 --- a/projects/models.py +++ b/projects/models.py @@ -204,12 +204,13 @@ class Order(models.Model): ('process', 'В процессе'), ('completed', 'Завершен'), ) - contractor = models.ForeignKey(User, null=True, blank=True, related_name='orders') - team = models.ForeignKey(Team, null=True, blank=True, related_name='orders') + + contractor = models.ForeignKey(User, null=True, blank=True, related_name='orders') # Related name should've been "contractor_orders" created = models.DateTimeField(default=timezone.now) project = models.OneToOneField(Project, related_name='order') secure = models.BooleanField(default=False) status = models.CharField(max_length=30, choices=STATUSES, default='created') + team = models.ForeignKey(Team, null=True, blank=True, related_name='orders') def __str__(self): return self.project.name diff --git a/projects/views.py b/projects/views.py index 630ca47..c8ee7dd 100644 --- a/projects/views.py +++ b/projects/views.py @@ -59,8 +59,7 @@ class ProjectDetailWithAnswerView(BaseMixin, View): answer = _.first(_.filter(project_answers, lambda a: a.author == contractor)) if not answer: - try: team = contractor.team - except Team.DoesNotExist: team = None + team = util.get_related_or_none(contractor, 'team') if team: answer = _.first(_.filter(project_answers, lambda a: a.author == team)) @@ -68,8 +67,10 @@ class ProjectDetailWithAnswerView(BaseMixin, View): context.update({'answer': answer}) if not answer: - try: contractor.team; context.update({'can_answer_as_team': True}) - except Team.DoesNotExist: pass + 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}) @@ -154,8 +155,7 @@ class ProjectAnswerCreateMessageView(BaseMixin, View): if isinstance(answer.author, User) and answer.author == request.user: message.contractor_or_team = request.user elif isinstance(answer.author, Team): - try: team = request.user.team - except Team.DoesNotExist: team = None + team = util.get_related_or_none(request.user, 'team') if team and answer.author == team: message.contractor_or_team = team diff --git a/users/models.py b/users/models.py index 432e1f2..f33bb5e 100644 --- a/users/models.py +++ b/users/models.py @@ -147,11 +147,11 @@ class User(AbstractBaseUser, PermissionsMixin): if self.location: if self.location.level == 3: return self.location.parent.parent.name + ', ' + self.location.name - else: - return None - else: - return None - + elif self.location.level == 2: + return self.location.parent.name + ', ' + self.location.name + elif self.location.level == 1: + return self.location.name + @property def is_staff(self): return self.is_superuser diff --git a/users/templates/contractor_office.html b/users/templates/contractor_office.html index 3ccb4b4..6b78b0a 100644 --- a/users/templates/contractor_office.html +++ b/users/templates/contractor_office.html @@ -17,12 +17,12 @@
- - - {% if not contractor.is_owner_team %} + {% if contractor.team %} + + {% else %} @@ -58,7 +58,7 @@
- {% if contractor.is_owner_team %} + {% if contractor.team %}
@@ -82,11 +82,11 @@

Состав группы

- {% for p in participants %} + {% for c in team_members %}
- {% if p.avatar %} - {% thumbnail p.avatar "60x60" crop="center" as im %} + {% if c.avatar %} + {% thumbnail c.avatar "60x60" crop="center" as im %} mess-image {% endthumbnail %} {% else %} @@ -95,10 +95,10 @@

- {{ p.get_full_name }} + {{ c.get_full_name }}

-

{{ p.get_popular_specialization }}

+

{{ c.get_popular_specialization }}

{% empty %} В группе пока нет участников @@ -110,15 +110,15 @@

{{ contractor.team.name }}

-

Россия, Москва

+

{{ contractor.get_location }}

На сайте {{ contractor.team.created }}

- Кол-во человек: {{ participants_count }} + Кол-во человек: {{ team_member_count }}

- Выполненных проектов: 0 + Выполненных проектов: {{ completed_project_count }}

Свободен
diff --git a/users/views.py b/users/views.py index d7b1394..4b50b4b 100644 --- a/users/views.py +++ b/users/views.py @@ -364,24 +364,42 @@ class ContractorOfficeView(DetailView): context_object_name = 'contractor' form_class = TeamForm + def dispatch(self, request, *args, **kwargs): + if request.user.is_authenticated() and request.user.is_contractor() and request.user.pk == int(kwargs.get('pk')): + return super().dispatch(request, *args, **kwargs) + else: + raise PermissionDenied + def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) - context['form_team'] = self.form_class - if self.object.is_owner_team(): - participants = self.object.team.contractors.all() - context['participants'] = participants - context['participants_count'] = len(participants) + contractor = self.object + + if util.get_related_or_none(contractor, 'team'): + members = contractor.team.contractors.all() + context['team_members'] = members + context['team_member_count'] = len(members) - contractors = tuple(itertools.chain(participants, (self.object,))) + compl_proj = [] + portfolios = [] + work_sells = [] - portfolios = Portfolio.objects.filter(user__in=contractors) - context['portfolios'] = portfolios + compl_proj.extend(tuple(o.project for o in contractor.orders.filter(status='completed'))) + portfolios.extend(contractor.portfolios.all()) + work_sells.extend(contractor.work_sell.all()) + + for c in members: + compl_proj.extend(tuple(o.project for o in c.orders.filter(status='completed'))) + portfolios.extend(c.portfolios.all()) + work_sells.extend(c.work_sell.all()) - work_sells = WorkSell.objects.filter(contractor__in=contractors) + context['completed_project_count'] = len(compl_proj) + context['portfolios'] = portfolios context['work_sells'] = work_sells - context['reviews'] = Review.objects.filter(target_contractor__in=contractors) + context['reviews'] = Review.objects.filter(target_contractor__in=itertools.chain((contractor,), members)) + + context['form_team'] = self.form_class return context diff --git a/work_sell/migrations/0006_remove_worksell_team.py b/work_sell/migrations/0006_remove_worksell_team.py new file mode 100644 index 0000000..f8f09fe --- /dev/null +++ b/work_sell/migrations/0006_remove_worksell_team.py @@ -0,0 +1,19 @@ +# -*- coding: utf-8 -*- +# Generated by Django 1.9.7 on 2016-08-26 17:25 +from __future__ import unicode_literals + +from django.db import migrations + + +class Migration(migrations.Migration): + + dependencies = [ + ('work_sell', '0005_remove_worksell_img'), + ] + + operations = [ + migrations.RemoveField( + model_name='worksell', + name='team', + ), + ] diff --git a/work_sell/models.py b/work_sell/models.py index b2a6f68..d861fdb 100644 --- a/work_sell/models.py +++ b/work_sell/models.py @@ -23,7 +23,6 @@ class WorkSell(models.Model): location = TreeForeignKey('common.Location', related_name='worksells', null=True, blank=True) name = models.CharField(max_length=255) specialization = TreeForeignKey(Specialization, related_name='worksells', null=True, blank=True) - team = models.ForeignKey(Team, related_name='work_sells', null=True, blank=True) term = models.IntegerField(default=0, null=True, blank=True) term_type = models.CharField(max_length=20, choices=TERMS, default='hour', null=True, blank=True)