69 lines
2.7 KiB
Python
69 lines
2.7 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/>.
|
|
|
|
from django.utils.translation import ugettext as _
|
|
|
|
from taiga.base.exceptions import ValidationError
|
|
from taiga.base.api import serializers
|
|
from taiga.base.api import validators
|
|
from taiga.projects.notifications.validators import WatchersValidator
|
|
from taiga.projects.userstories.models import UserStory
|
|
from taiga.projects.validators import DuplicatedNameInProjectValidator
|
|
from taiga.projects.validators import ProjectExistsValidator
|
|
from . import models
|
|
|
|
|
|
class MilestoneExistsValidator:
|
|
def validate_milestone_id(self, attrs, source):
|
|
value = attrs[source]
|
|
if not models.Milestone.objects.filter(pk=value).exists():
|
|
msg = _("There's no milestone with that id")
|
|
raise ValidationError(msg)
|
|
return attrs
|
|
|
|
|
|
class MilestoneValidator(WatchersValidator, DuplicatedNameInProjectValidator, validators.ModelValidator):
|
|
class Meta:
|
|
model = models.Milestone
|
|
read_only_fields = ("id", "created_date", "modified_date")
|
|
|
|
|
|
# bulk validators
|
|
class _UserStoryMilestoneBulkValidator(validators.Validator):
|
|
us_id = serializers.IntegerField()
|
|
order = serializers.IntegerField()
|
|
|
|
|
|
class UpdateMilestoneBulkValidator(MilestoneExistsValidator,
|
|
ProjectExistsValidator,
|
|
validators.Validator):
|
|
project_id = serializers.IntegerField()
|
|
milestone_id = serializers.IntegerField()
|
|
bulk_stories = _UserStoryMilestoneBulkValidator(many=True)
|
|
|
|
def validate_bulk_stories(self, attrs, source):
|
|
filters = {
|
|
"project__id": attrs["project_id"],
|
|
"id__in": [us["us_id"] for us in attrs[source]]
|
|
}
|
|
|
|
if UserStory.objects.filter(**filters).count() != len(filters["id__in"]):
|
|
raise ValidationError(_("All the user stories must be from the same project"))
|
|
|
|
return attrs
|