835 lines
33 KiB
Python
835 lines
33 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (C) 2014-2017 Andrey Antukh <niwi@niwi.nz>
|
|
# Copyright (C) 2014-2017 Jesús Espino <jespinog@gmail.com>
|
|
# Copyright (C) 2014-2017 David Barragán <bameda@dbarragan.com>
|
|
# Copyright (C) 2014-2017 Alejandro Alonso <alejandro.alonso@kaleidos.net>
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import uuid
|
|
from easy_thumbnails.source_generators import pil_image
|
|
from dateutil.relativedelta import relativedelta
|
|
|
|
from django.apps import apps
|
|
from django.conf import settings
|
|
from django.http import Http404
|
|
from django.utils.translation import ugettext as _
|
|
from django.utils import timezone
|
|
|
|
from django_pglocks import advisory_lock
|
|
|
|
from taiga.base import filters
|
|
from taiga.base import exceptions as exc
|
|
from taiga.base import response
|
|
from taiga.base.api import ModelCrudViewSet, ModelListViewSet
|
|
from taiga.base.api.mixins import BlockedByProjectMixin, BlockeableSaveMixin, BlockeableDeleteMixin
|
|
from taiga.base.api.permissions import AllowAnyPermission
|
|
from taiga.base.api.utils import get_object_or_404
|
|
from taiga.base.decorators import list_route
|
|
from taiga.base.decorators import detail_route
|
|
from taiga.base.utils.slug import slugify_uniquely
|
|
|
|
from taiga.permissions import services as permissions_services
|
|
|
|
from taiga.projects.epics.models import Epic
|
|
from taiga.projects.history.mixins import HistoryResourceMixin
|
|
from taiga.projects.issues.models import Issue
|
|
from taiga.projects.likes.mixins.viewsets import LikedResourceMixin, FansViewSetMixin
|
|
from taiga.projects.notifications.mixins import WatchersViewSetMixin
|
|
from taiga.projects.notifications.choices import NotifyLevel
|
|
from taiga.projects.mixins.on_destroy import MoveOnDestroyMixin
|
|
from taiga.projects.mixins.ordering import BulkUpdateOrderMixin
|
|
from taiga.projects.tasks.models import Task
|
|
from taiga.projects.tagging.api import TagsColorsResourceMixin
|
|
from taiga.projects.userstories.models import UserStory, RolePoints
|
|
from taiga.users import services as users_services
|
|
|
|
from . import filters as project_filters
|
|
from . import models
|
|
from . import permissions
|
|
from . import serializers
|
|
from . import validators
|
|
from . import services
|
|
from . import utils as project_utils
|
|
from . import throttling
|
|
|
|
######################################################
|
|
# Project
|
|
######################################################
|
|
|
|
|
|
class ProjectViewSet(LikedResourceMixin, HistoryResourceMixin,
|
|
BlockeableSaveMixin, BlockeableDeleteMixin,
|
|
TagsColorsResourceMixin, ModelCrudViewSet):
|
|
validator_class = validators.ProjectValidator
|
|
queryset = models.Project.objects.all()
|
|
permission_classes = (permissions.ProjectPermission, )
|
|
filter_backends = (project_filters.UserOrderFilterBackend,
|
|
project_filters.QFilterBackend,
|
|
project_filters.CanViewProjectObjFilterBackend,
|
|
project_filters.DiscoverModeFilterBackend)
|
|
|
|
filter_fields = (("member", "members"),
|
|
"is_looking_for_people",
|
|
"is_featured",
|
|
"is_backlog_activated",
|
|
"is_kanban_activated")
|
|
|
|
ordering = ("name", "id")
|
|
order_by_fields = ("total_fans",
|
|
"total_fans_last_week",
|
|
"total_fans_last_month",
|
|
"total_fans_last_year",
|
|
"total_activity",
|
|
"total_activity_last_week",
|
|
"total_activity_last_month",
|
|
"total_activity_last_year")
|
|
|
|
def is_blocked(self, obj):
|
|
return obj.blocked_code is not None
|
|
|
|
def _get_order_by_field_name(self):
|
|
order_by_query_param = project_filters.CanViewProjectObjFilterBackend.order_by_query_param
|
|
order_by = self.request.QUERY_PARAMS.get(order_by_query_param, None)
|
|
if order_by is not None and order_by.startswith("-"):
|
|
return order_by[1:]
|
|
|
|
def get_queryset(self):
|
|
qs = super().get_queryset()
|
|
qs = qs.select_related("owner")
|
|
if self.request.QUERY_PARAMS.get('discover_mode', False):
|
|
qs = project_utils.attach_members(qs)
|
|
qs = project_utils.attach_notify_policies(qs)
|
|
qs = project_utils.attach_is_fan(qs, user=self.request.user)
|
|
qs = project_utils.attach_my_role_permissions(qs, user=self.request.user)
|
|
qs = project_utils.attach_my_role_permissions(qs, user=self.request.user)
|
|
qs = project_utils.attach_closed_milestones(qs)
|
|
else:
|
|
qs = project_utils.attach_extra_info(qs, user=self.request.user)
|
|
|
|
# If filtering an activity period we must exclude the activities not updated recently enough
|
|
now = timezone.now()
|
|
order_by_field_name = self._get_order_by_field_name()
|
|
if order_by_field_name == "total_fans_last_week":
|
|
qs = qs.filter(totals_updated_datetime__gte=now - relativedelta(weeks=1))
|
|
elif order_by_field_name == "total_fans_last_month":
|
|
qs = qs.filter(totals_updated_datetime__gte=now - relativedelta(months=1))
|
|
elif order_by_field_name == "total_fans_last_year":
|
|
qs = qs.filter(totals_updated_datetime__gte=now - relativedelta(years=1))
|
|
elif order_by_field_name == "total_activity_last_week":
|
|
qs = qs.filter(totals_updated_datetime__gte=now - relativedelta(weeks=1))
|
|
elif order_by_field_name == "total_activity_last_month":
|
|
qs = qs.filter(totals_updated_datetime__gte=now - relativedelta(months=1))
|
|
elif order_by_field_name == "total_activity_last_year":
|
|
qs = qs.filter(totals_updated_datetime__gte=now - relativedelta(years=1))
|
|
|
|
return qs
|
|
|
|
def retrieve(self, request, *args, **kwargs):
|
|
qs = self.get_queryset()
|
|
if self.action == "by_slug":
|
|
self.lookup_field = "slug"
|
|
# If we retrieve the project by slug we want to filter by user the
|
|
# permissions and return 404 in case the user don't have access
|
|
flt = filters.get_filter_expression_can_view_projects(
|
|
self.request.user)
|
|
|
|
qs = qs.filter(flt)
|
|
|
|
self.object = get_object_or_404(qs, **kwargs)
|
|
|
|
self.check_permissions(request, 'retrieve', self.object)
|
|
|
|
if self.object is None:
|
|
raise Http404
|
|
|
|
serializer = self.get_serializer(self.object)
|
|
return response.Ok(serializer.data)
|
|
|
|
def get_serializer_class(self):
|
|
if self.action == "list":
|
|
return serializers.ProjectSerializer
|
|
|
|
return serializers.ProjectDetailSerializer
|
|
|
|
@detail_route(methods=["POST"])
|
|
def change_logo(self, request, *args, **kwargs):
|
|
"""
|
|
Change logo to this project.
|
|
"""
|
|
self.object = get_object_or_404(self.get_queryset(), **kwargs)
|
|
self.check_permissions(request, "change_logo", self.object)
|
|
|
|
logo = request.FILES.get('logo', None)
|
|
if not logo:
|
|
raise exc.WrongArguments(_("Incomplete arguments"))
|
|
try:
|
|
pil_image(logo)
|
|
except Exception:
|
|
raise exc.WrongArguments(_("Invalid image format"))
|
|
|
|
self.pre_conditions_on_save(self.object)
|
|
|
|
self.object.logo = logo
|
|
self.object.save(update_fields=["logo"])
|
|
|
|
serializer = self.get_serializer(self.object)
|
|
return response.Ok(serializer.data)
|
|
|
|
@detail_route(methods=["POST"])
|
|
def remove_logo(self, request, *args, **kwargs):
|
|
"""
|
|
Remove the logo of a project.
|
|
"""
|
|
self.object = get_object_or_404(self.get_queryset(), **kwargs)
|
|
self.check_permissions(request, "remove_logo", self.object)
|
|
self.pre_conditions_on_save(self.object)
|
|
self.object.logo = None
|
|
self.object.save(update_fields=["logo"])
|
|
|
|
serializer = self.get_serializer(self.object)
|
|
return response.Ok(serializer.data)
|
|
|
|
@detail_route(methods=["POST"])
|
|
def watch(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "watch", project)
|
|
self.pre_conditions_on_save(project)
|
|
notify_level = request.DATA.get("notify_level", NotifyLevel.involved)
|
|
project.add_watcher(self.request.user, notify_level=notify_level)
|
|
return response.Ok()
|
|
|
|
@detail_route(methods=["POST"])
|
|
def unwatch(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "unwatch", project)
|
|
self.pre_conditions_on_save(project)
|
|
user = self.request.user
|
|
project.remove_watcher(user)
|
|
return response.Ok()
|
|
|
|
@list_route(methods=["POST"])
|
|
def bulk_update_order(self, request, **kwargs):
|
|
if self.request.user.is_anonymous():
|
|
return response.Unauthorized()
|
|
|
|
validator = validators.UpdateProjectOrderBulkValidator(data=request.DATA, many=True)
|
|
if not validator.is_valid():
|
|
return response.BadRequest(validator.errors)
|
|
|
|
data = validator.data
|
|
services.update_projects_order_in_bulk(data, "user_order", request.user)
|
|
return response.NoContent(data=None)
|
|
|
|
@detail_route(methods=["POST"])
|
|
def create_template(self, request, **kwargs):
|
|
template_name = request.DATA.get('template_name', None)
|
|
template_description = request.DATA.get('template_description', None)
|
|
|
|
if not template_name:
|
|
raise response.BadRequest(_("Not valid template name"))
|
|
|
|
if not template_description:
|
|
raise response.BadRequest(_("Not valid template description"))
|
|
|
|
with advisory_lock("create-project-template"):
|
|
template_slug = slugify_uniquely(template_name, models.ProjectTemplate)
|
|
|
|
project = self.get_object()
|
|
|
|
self.check_permissions(request, 'create_template', project)
|
|
|
|
template = models.ProjectTemplate(
|
|
name=template_name,
|
|
slug=template_slug,
|
|
description=template_description,
|
|
)
|
|
|
|
template.load_data_from_project(project)
|
|
|
|
template.save()
|
|
return response.Created(serializers.ProjectTemplateSerializer(template).data)
|
|
|
|
@detail_route(methods=['POST'])
|
|
def leave(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, 'leave', project)
|
|
self.pre_conditions_on_save(project)
|
|
services.remove_user_from_project(request.user, project)
|
|
return response.Ok()
|
|
|
|
def _regenerate_csv_uuid(self, project, field):
|
|
uuid_value = uuid.uuid4().hex
|
|
setattr(project, field, uuid_value)
|
|
project.save()
|
|
return uuid_value
|
|
|
|
@detail_route(methods=["POST"])
|
|
def regenerate_epics_csv_uuid(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "regenerate_epics_csv_uuid", project)
|
|
self.pre_conditions_on_save(project)
|
|
data = {"uuid": self._regenerate_csv_uuid(project, "epics_csv_uuid")}
|
|
return response.Ok(data)
|
|
|
|
@detail_route(methods=["POST"])
|
|
def regenerate_userstories_csv_uuid(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "regenerate_userstories_csv_uuid", project)
|
|
self.pre_conditions_on_save(project)
|
|
data = {"uuid": self._regenerate_csv_uuid(project, "userstories_csv_uuid")}
|
|
return response.Ok(data)
|
|
|
|
@detail_route(methods=["POST"])
|
|
def regenerate_tasks_csv_uuid(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "regenerate_tasks_csv_uuid", project)
|
|
self.pre_conditions_on_save(project)
|
|
data = {"uuid": self._regenerate_csv_uuid(project, "tasks_csv_uuid")}
|
|
return response.Ok(data)
|
|
|
|
@detail_route(methods=["POST"])
|
|
def regenerate_issues_csv_uuid(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "regenerate_issues_csv_uuid", project)
|
|
self.pre_conditions_on_save(project)
|
|
data = {"uuid": self._regenerate_csv_uuid(project, "issues_csv_uuid")}
|
|
return response.Ok(data)
|
|
|
|
@list_route(methods=["GET"])
|
|
def by_slug(self, request, *args, **kwargs):
|
|
slug = request.QUERY_PARAMS.get("slug", None)
|
|
return self.retrieve(request, slug=slug)
|
|
|
|
@detail_route(methods=["GET", "PATCH"])
|
|
def modules(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, 'modules', project)
|
|
modules_config = services.get_modules_config(project)
|
|
|
|
if request.method == "GET":
|
|
return response.Ok(modules_config.config)
|
|
|
|
else:
|
|
self.pre_conditions_on_save(project)
|
|
modules_config.config.update(request.DATA)
|
|
modules_config.save()
|
|
return response.NoContent()
|
|
|
|
@detail_route(methods=["GET"])
|
|
def stats(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "stats", project)
|
|
return response.Ok(services.get_stats_for_project(project))
|
|
|
|
@detail_route(methods=["GET"])
|
|
def member_stats(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "member_stats", project)
|
|
return response.Ok(services.get_member_stats_for_project(project))
|
|
|
|
@detail_route(methods=["GET"])
|
|
def issues_stats(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "issues_stats", project)
|
|
return response.Ok(services.get_stats_for_project_issues(project))
|
|
|
|
@detail_route(methods=["POST"])
|
|
def transfer_validate_token(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "transfer_validate_token", project)
|
|
token = request.DATA.get('token', None)
|
|
services.transfer.validate_project_transfer_token(token, project, request.user)
|
|
return response.Ok()
|
|
|
|
@detail_route(methods=["POST"])
|
|
def transfer_request(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "transfer_request", project)
|
|
services.request_project_transfer(project, request.user)
|
|
return response.Ok()
|
|
|
|
@detail_route(methods=['post'])
|
|
def transfer_start(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "transfer_start", project)
|
|
|
|
user_id = request.DATA.get('user', None)
|
|
if user_id is None:
|
|
raise exc.WrongArguments(_("Invalid user id"))
|
|
|
|
user_model = apps.get_model("users", "User")
|
|
try:
|
|
user = user_model.objects.get(id=user_id)
|
|
except user_model.DoesNotExist:
|
|
return response.BadRequest(_("The user doesn't exist"))
|
|
|
|
# Check the user is a membership from the project
|
|
if not project.memberships.filter(user=user).exists():
|
|
return response.BadRequest(_("The user must be already a project member"))
|
|
|
|
reason = request.DATA.get('reason', None)
|
|
services.start_project_transfer(project, user, reason)
|
|
return response.Ok()
|
|
|
|
@detail_route(methods=["POST"])
|
|
def transfer_accept(self, request, pk=None):
|
|
token = request.DATA.get('token', None)
|
|
if token is None:
|
|
raise exc.WrongArguments(_("Invalid token"))
|
|
|
|
project = self.get_object()
|
|
self.check_permissions(request, "transfer_accept", project)
|
|
|
|
(can_transfer, error_message) = services.check_if_project_can_be_transfered(
|
|
project,
|
|
request.user,
|
|
)
|
|
if not can_transfer:
|
|
members = project.memberships.count()
|
|
raise exc.NotEnoughSlotsForProject(project.is_private, members, error_message)
|
|
|
|
reason = request.DATA.get('reason', None)
|
|
services.accept_project_transfer(project, request.user, token, reason)
|
|
return response.Ok()
|
|
|
|
@detail_route(methods=["POST"])
|
|
def transfer_reject(self, request, pk=None):
|
|
token = request.DATA.get('token', None)
|
|
if token is None:
|
|
raise exc.WrongArguments(_("Invalid token"))
|
|
|
|
project = self.get_object()
|
|
self.check_permissions(request, "transfer_reject", project)
|
|
|
|
reason = request.DATA.get('reason', None)
|
|
services.reject_project_transfer(project, request.user, token, reason)
|
|
return response.Ok()
|
|
|
|
@detail_route(methods=["POST"])
|
|
def duplicate(self, request, pk=None):
|
|
project = self.get_object()
|
|
self.check_permissions(request, "duplicate", project)
|
|
if project.blocked_code is not None:
|
|
raise exc.Blocked(_("Blocked element"))
|
|
|
|
validator = validators.DuplicateProjectValidator(data=request.DATA)
|
|
if not validator.is_valid():
|
|
return response.BadRequest(validator.errors)
|
|
|
|
data = validator.data
|
|
|
|
# Validate if the project can be imported
|
|
is_private = data.get('is_private', False)
|
|
total_memberships = len(data.get("users", [])) + 1
|
|
(enough_slots, error_message) = users_services.has_available_slot_for_new_project(
|
|
self.request.user,
|
|
is_private,
|
|
total_memberships
|
|
)
|
|
if not enough_slots:
|
|
raise exc.NotEnoughSlotsForProject(is_private, total_memberships, error_message)
|
|
|
|
new_project = services.duplicate_project(
|
|
project=project,
|
|
owner=request.user,
|
|
name=data["name"],
|
|
description=data["description"],
|
|
is_private=data["is_private"],
|
|
users=data["users"]
|
|
)
|
|
new_project = get_object_or_404(self.get_queryset(), id=new_project.id)
|
|
serializer = self.get_serializer(new_project)
|
|
return response.Created(serializer.data)
|
|
|
|
def _raise_if_blocked(self, project):
|
|
if self.is_blocked(project):
|
|
raise exc.Blocked(_("Blocked element"))
|
|
|
|
def _set_base_permissions(self, obj):
|
|
update_permissions = False
|
|
if not obj.id:
|
|
if not obj.is_private:
|
|
# Creating a public project
|
|
update_permissions = True
|
|
else:
|
|
if self.get_object().is_private != obj.is_private:
|
|
# Changing project public state
|
|
update_permissions = True
|
|
|
|
if update_permissions:
|
|
permissions_services.set_base_permissions_for_project(obj)
|
|
|
|
def pre_save(self, obj):
|
|
if not obj.id:
|
|
obj.owner = self.request.user
|
|
obj.template = self.request.QUERY_PARAMS.get('template', None)
|
|
|
|
if not obj.id or self.get_object().is_private != obj.is_private:
|
|
# Validate if the owner have enought slots to create the project
|
|
# or if you are changing the privacity
|
|
(can_create_or_update, error_message) = services.check_if_project_can_be_created_or_updated(obj)
|
|
if not can_create_or_update:
|
|
members = max(obj.memberships.count(), 1)
|
|
raise exc.NotEnoughSlotsForProject(obj.is_private, members, error_message)
|
|
|
|
self._set_base_permissions(obj)
|
|
super().pre_save(obj)
|
|
|
|
def destroy(self, request, *args, **kwargs):
|
|
obj = self.get_object_or_none()
|
|
self.check_permissions(request, 'destroy', obj)
|
|
|
|
if obj is None:
|
|
raise Http404
|
|
|
|
self.pre_delete(obj)
|
|
self.pre_conditions_on_delete(obj)
|
|
|
|
services.orphan_project(obj)
|
|
if settings.CELERY_ENABLED:
|
|
services.delete_project.delay(obj.id)
|
|
else:
|
|
services.delete_project(obj.id)
|
|
|
|
return response.NoContent()
|
|
|
|
|
|
class ProjectFansViewSet(FansViewSetMixin, ModelListViewSet):
|
|
permission_classes = (permissions.ProjectFansPermission,)
|
|
resource_model = models.Project
|
|
|
|
|
|
class ProjectWatchersViewSet(WatchersViewSetMixin, ModelListViewSet):
|
|
permission_classes = (permissions.ProjectWatchersPermission,)
|
|
resource_model = models.Project
|
|
|
|
|
|
######################################################
|
|
## Custom values for selectors
|
|
######################################################
|
|
|
|
class EpicStatusViewSet(MoveOnDestroyMixin, BlockedByProjectMixin,
|
|
ModelCrudViewSet, BulkUpdateOrderMixin):
|
|
|
|
model = models.EpicStatus
|
|
serializer_class = serializers.EpicStatusSerializer
|
|
validator_class = validators.EpicStatusValidator
|
|
permission_classes = (permissions.EpicStatusPermission,)
|
|
filter_backends = (filters.CanViewProjectFilterBackend,)
|
|
filter_fields = ('project',)
|
|
bulk_update_param = "bulk_epic_statuses"
|
|
bulk_update_perm = "change_epicstatus"
|
|
bulk_update_order_action = services.bulk_update_epic_status_order
|
|
move_on_destroy_related_class = Epic
|
|
move_on_destroy_related_field = "status"
|
|
move_on_destroy_project_default_field = "default_epic_status"
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
project_id = request.DATA.get("project", 0)
|
|
with advisory_lock("epic-status-creation-{}".format(project_id)):
|
|
return super().create(request, *args, **kwargs)
|
|
|
|
|
|
class UserStoryStatusViewSet(MoveOnDestroyMixin, BlockedByProjectMixin,
|
|
ModelCrudViewSet, BulkUpdateOrderMixin):
|
|
|
|
model = models.UserStoryStatus
|
|
serializer_class = serializers.UserStoryStatusSerializer
|
|
validator_class = validators.UserStoryStatusValidator
|
|
permission_classes = (permissions.UserStoryStatusPermission,)
|
|
filter_backends = (filters.CanViewProjectFilterBackend,)
|
|
filter_fields = ('project',)
|
|
bulk_update_param = "bulk_userstory_statuses"
|
|
bulk_update_perm = "change_userstorystatus"
|
|
bulk_update_order_action = services.bulk_update_userstory_status_order
|
|
move_on_destroy_related_class = UserStory
|
|
move_on_destroy_related_field = "status"
|
|
move_on_destroy_project_default_field = "default_us_status"
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
project_id = request.DATA.get("project", 0)
|
|
with advisory_lock("epic-user-story-status-creation-{}".format(project_id)):
|
|
return super().create(request, *args, **kwargs)
|
|
|
|
|
|
class PointsViewSet(MoveOnDestroyMixin, BlockedByProjectMixin,
|
|
ModelCrudViewSet, BulkUpdateOrderMixin):
|
|
|
|
model = models.Points
|
|
serializer_class = serializers.PointsSerializer
|
|
validator_class = validators.PointsValidator
|
|
permission_classes = (permissions.PointsPermission,)
|
|
filter_backends = (filters.CanViewProjectFilterBackend,)
|
|
filter_fields = ('project',)
|
|
bulk_update_param = "bulk_points"
|
|
bulk_update_perm = "change_points"
|
|
bulk_update_order_action = services.bulk_update_points_order
|
|
move_on_destroy_related_class = RolePoints
|
|
move_on_destroy_related_field = "points"
|
|
move_on_destroy_project_default_field = "default_points"
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
project_id = request.DATA.get("project", 0)
|
|
with advisory_lock("points-creation-{}".format(project_id)):
|
|
return super().create(request, *args, **kwargs)
|
|
|
|
|
|
class TaskStatusViewSet(MoveOnDestroyMixin, BlockedByProjectMixin,
|
|
ModelCrudViewSet, BulkUpdateOrderMixin):
|
|
|
|
model = models.TaskStatus
|
|
serializer_class = serializers.TaskStatusSerializer
|
|
validator_class = validators.TaskStatusValidator
|
|
permission_classes = (permissions.TaskStatusPermission,)
|
|
filter_backends = (filters.CanViewProjectFilterBackend,)
|
|
filter_fields = ("project",)
|
|
bulk_update_param = "bulk_task_statuses"
|
|
bulk_update_perm = "change_taskstatus"
|
|
bulk_update_order_action = services.bulk_update_task_status_order
|
|
move_on_destroy_related_class = Task
|
|
move_on_destroy_related_field = "status"
|
|
move_on_destroy_project_default_field = "default_task_status"
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
project_id = request.DATA.get("project", 0)
|
|
with advisory_lock("task-status-creation-{}".format(project_id)):
|
|
return super().create(request, *args, **kwargs)
|
|
|
|
|
|
class SeverityViewSet(MoveOnDestroyMixin, BlockedByProjectMixin,
|
|
ModelCrudViewSet, BulkUpdateOrderMixin):
|
|
|
|
model = models.Severity
|
|
serializer_class = serializers.SeveritySerializer
|
|
validator_class = validators.SeverityValidator
|
|
permission_classes = (permissions.SeverityPermission,)
|
|
filter_backends = (filters.CanViewProjectFilterBackend,)
|
|
filter_fields = ("project",)
|
|
bulk_update_param = "bulk_severities"
|
|
bulk_update_perm = "change_severity"
|
|
bulk_update_order_action = services.bulk_update_severity_order
|
|
move_on_destroy_related_class = Issue
|
|
move_on_destroy_related_field = "severity"
|
|
move_on_destroy_project_default_field = "default_severity"
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
project_id = request.DATA.get("project", 0)
|
|
with advisory_lock("severity-creation-{}".format(project_id)):
|
|
return super().create(request, *args, **kwargs)
|
|
|
|
|
|
class PriorityViewSet(MoveOnDestroyMixin, BlockedByProjectMixin,
|
|
ModelCrudViewSet, BulkUpdateOrderMixin):
|
|
model = models.Priority
|
|
serializer_class = serializers.PrioritySerializer
|
|
validator_class = validators.PriorityValidator
|
|
permission_classes = (permissions.PriorityPermission,)
|
|
filter_backends = (filters.CanViewProjectFilterBackend,)
|
|
filter_fields = ("project",)
|
|
bulk_update_param = "bulk_priorities"
|
|
bulk_update_perm = "change_priority"
|
|
bulk_update_order_action = services.bulk_update_priority_order
|
|
move_on_destroy_related_class = Issue
|
|
move_on_destroy_related_field = "priority"
|
|
move_on_destroy_project_default_field = "default_priority"
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
project_id = request.DATA.get("project", 0)
|
|
with advisory_lock("priority-creation-{}".format(project_id)):
|
|
return super().create(request, *args, **kwargs)
|
|
|
|
|
|
class IssueTypeViewSet(MoveOnDestroyMixin, BlockedByProjectMixin,
|
|
ModelCrudViewSet, BulkUpdateOrderMixin):
|
|
model = models.IssueType
|
|
serializer_class = serializers.IssueTypeSerializer
|
|
validator_class = validators.IssueTypeValidator
|
|
permission_classes = (permissions.IssueTypePermission,)
|
|
filter_backends = (filters.CanViewProjectFilterBackend,)
|
|
filter_fields = ("project",)
|
|
bulk_update_param = "bulk_issue_types"
|
|
bulk_update_perm = "change_issuetype"
|
|
bulk_update_order_action = services.bulk_update_issue_type_order
|
|
move_on_destroy_related_class = Issue
|
|
move_on_destroy_related_field = "type"
|
|
move_on_destroy_project_default_field = "default_issue_type"
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
project_id = request.DATA.get("project", 0)
|
|
with advisory_lock("issue-type-creation-{}".format(project_id)):
|
|
return super().create(request, *args, **kwargs)
|
|
|
|
|
|
class IssueStatusViewSet(MoveOnDestroyMixin, BlockedByProjectMixin,
|
|
ModelCrudViewSet, BulkUpdateOrderMixin):
|
|
model = models.IssueStatus
|
|
serializer_class = serializers.IssueStatusSerializer
|
|
validator_class = validators.IssueStatusValidator
|
|
permission_classes = (permissions.IssueStatusPermission,)
|
|
filter_backends = (filters.CanViewProjectFilterBackend,)
|
|
filter_fields = ("project",)
|
|
bulk_update_param = "bulk_issue_statuses"
|
|
bulk_update_perm = "change_issuestatus"
|
|
bulk_update_order_action = services.bulk_update_issue_status_order
|
|
move_on_destroy_related_class = Issue
|
|
move_on_destroy_related_field = "status"
|
|
move_on_destroy_project_default_field = "default_issue_status"
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
project_id = request.DATA.get("project", 0)
|
|
with advisory_lock("issue-status-creation-{}".format(project_id)):
|
|
return super().create(request, *args, **kwargs)
|
|
|
|
|
|
######################################################
|
|
## Project Template
|
|
######################################################
|
|
|
|
class ProjectTemplateViewSet(ModelCrudViewSet):
|
|
model = models.ProjectTemplate
|
|
serializer_class = serializers.ProjectTemplateSerializer
|
|
validator_class = validators.ProjectTemplateValidator
|
|
permission_classes = (permissions.ProjectTemplatePermission,)
|
|
|
|
def get_queryset(self):
|
|
return models.ProjectTemplate.objects.all()
|
|
|
|
|
|
######################################################
|
|
## Members & Invitations
|
|
######################################################
|
|
|
|
class MembershipViewSet(BlockedByProjectMixin, ModelCrudViewSet):
|
|
model = models.Membership
|
|
admin_serializer_class = serializers.MembershipAdminSerializer
|
|
serializer_class = serializers.MembershipSerializer
|
|
validator_class = validators.MembershipValidator
|
|
permission_classes = (permissions.MembershipPermission,)
|
|
filter_backends = (filters.CanViewProjectFilterBackend,)
|
|
filter_fields = ("project", "role")
|
|
throttle_classes = (throttling.MembershipsRateThrottle,)
|
|
|
|
def get_serializer_class(self):
|
|
use_admin_serializer = False
|
|
|
|
if self.action == "create":
|
|
use_admin_serializer = True
|
|
|
|
if self.action == "retrieve":
|
|
use_admin_serializer = permissions_services.is_project_admin(self.request.user, self.object.project)
|
|
|
|
project_id = self.request.QUERY_PARAMS.get("project", None)
|
|
if self.action == "list" and project_id is not None:
|
|
project = get_object_or_404(models.Project, pk=project_id)
|
|
use_admin_serializer = permissions_services.is_project_admin(self.request.user, project)
|
|
|
|
if use_admin_serializer:
|
|
return self.admin_serializer_class
|
|
|
|
else:
|
|
return self.serializer_class
|
|
|
|
def _check_if_project_can_have_more_memberships(self, project, total_new_memberships):
|
|
(can_add_memberships, error_type) = services.check_if_project_can_have_more_memberships(
|
|
project,
|
|
total_new_memberships
|
|
)
|
|
if not can_add_memberships:
|
|
raise exc.NotEnoughSlotsForProject(
|
|
project.is_private,
|
|
total_new_memberships,
|
|
error_type
|
|
)
|
|
|
|
@list_route(methods=["POST"])
|
|
def bulk_create(self, request, **kwargs):
|
|
context = {
|
|
"request": request
|
|
}
|
|
validator = validators.MembersBulkValidator(data=request.DATA, context=context)
|
|
if not validator.is_valid():
|
|
return response.BadRequest(validator.errors)
|
|
|
|
data = validator.data
|
|
project = models.Project.objects.get(id=data["project_id"])
|
|
invitation_extra_text = data.get("invitation_extra_text", None)
|
|
self.check_permissions(request, 'bulk_create', project)
|
|
if project.blocked_code is not None:
|
|
raise exc.Blocked(_("Blocked element"))
|
|
|
|
if "bulk_memberships" in data and isinstance(data["bulk_memberships"], list):
|
|
total_new_memberships = len(data["bulk_memberships"])
|
|
self._check_if_project_can_have_more_memberships(project, total_new_memberships)
|
|
|
|
try:
|
|
with advisory_lock("membership-creation-{}".format(project.id)):
|
|
members = services.create_members_in_bulk(data["bulk_memberships"],
|
|
project=project,
|
|
invitation_extra_text=invitation_extra_text,
|
|
callback=self.post_save,
|
|
precall=self.pre_save)
|
|
except exc.ValidationError as err:
|
|
return response.BadRequest(err.message_dict)
|
|
|
|
members_serialized = self.admin_serializer_class(members, many=True)
|
|
return response.Ok(members_serialized.data)
|
|
|
|
@detail_route(methods=["POST"])
|
|
def resend_invitation(self, request, **kwargs):
|
|
invitation = self.get_object()
|
|
|
|
self.check_permissions(request, 'resend_invitation', invitation.project)
|
|
self.pre_conditions_on_save(invitation)
|
|
|
|
services.send_invitation(invitation=invitation)
|
|
return response.NoContent()
|
|
|
|
def pre_delete(self, obj):
|
|
if obj.user is not None and not services.can_user_leave_project(obj.user, obj.project):
|
|
raise exc.BadRequest(_("The project must have an owner and at least one of the users "
|
|
"must be an active admin"))
|
|
|
|
def pre_save(self, obj):
|
|
if not obj.id:
|
|
self._check_if_project_can_have_more_memberships(obj.project, 1)
|
|
|
|
if not obj.token:
|
|
obj.token = str(uuid.uuid1())
|
|
|
|
obj.invited_by = self.request.user
|
|
obj.user = services.find_invited_user(obj.email, default=obj.user)
|
|
super().pre_save(obj)
|
|
|
|
def post_save(self, object, created=False):
|
|
super().post_save(object, created=created)
|
|
|
|
if not created:
|
|
return
|
|
|
|
# Send email only if a new membership is created
|
|
services.send_invitation(invitation=object)
|
|
|
|
|
|
class InvitationViewSet(ModelListViewSet):
|
|
"""
|
|
Only used by front for get invitation by it token.
|
|
"""
|
|
queryset = models.Membership.objects.filter(user__isnull=True)
|
|
serializer_class = serializers.MembershipSerializer
|
|
lookup_field = "token"
|
|
permission_classes = (AllowAnyPermission,)
|
|
|
|
def list(self, *args, **kwargs):
|
|
raise exc.PermissionDenied(_("You don't have permisions to see that."))
|