109 lines
4.5 KiB
Python
109 lines
4.5 KiB
Python
# Copyright (C) 2014-2016 Andrey Antukh <niwi@niwi.nz>
|
|
# Copyright (C) 2014-2016 Jesús Espino <jespinog@gmail.com>
|
|
# Copyright (C) 2014-2016 David Barragán <bameda@dbarragan.com>
|
|
# Copyright (C) 2014-2016 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 contextlib import suppress
|
|
from django.core.exceptions import ObjectDoesNotExist
|
|
from django.utils import timezone
|
|
|
|
####################################
|
|
# Signals for cached prev task
|
|
####################################
|
|
|
|
# Define the previous version of the task for use it on the post_save handler
|
|
def cached_prev_task(sender, instance, **kwargs):
|
|
instance.prev = None
|
|
if instance.id:
|
|
instance.prev = sender.objects.get(id=instance.id)
|
|
|
|
|
|
####################################
|
|
# Signals for close US and Milestone
|
|
####################################
|
|
|
|
def try_to_close_or_open_us_and_milestone_when_create_or_edit_task(sender, instance, created, **kwargs):
|
|
_try_to_close_or_open_us_when_create_or_edit_task(instance)
|
|
_try_to_close_or_open_milestone_when_create_or_edit_task(instance)
|
|
|
|
def try_to_close_or_open_us_and_milestone_when_delete_task(sender, instance, **kwargs):
|
|
_try_to_close_or_open_us_when_delete_task(instance)
|
|
_try_to_close_milestone_when_delete_task(instance)
|
|
|
|
|
|
# US
|
|
def _try_to_close_or_open_us_when_create_or_edit_task(instance):
|
|
from taiga.projects.userstories import services as us_service
|
|
|
|
if instance.user_story_id:
|
|
if us_service.calculate_userstory_is_closed(instance.user_story):
|
|
us_service.close_userstory(instance.user_story)
|
|
else:
|
|
us_service.open_userstory(instance.user_story)
|
|
|
|
if instance.prev and instance.prev.user_story_id and instance.prev.user_story_id != instance.user_story_id:
|
|
if us_service.calculate_userstory_is_closed(instance.prev.user_story):
|
|
us_service.close_userstory(instance.prev.user_story)
|
|
else:
|
|
us_service.open_userstory(instance.prev.user_story)
|
|
|
|
|
|
def _try_to_close_or_open_us_when_delete_task(instance):
|
|
from taiga.projects.userstories import services as us_service
|
|
|
|
with suppress(ObjectDoesNotExist):
|
|
if instance.user_story_id:
|
|
if us_service.calculate_userstory_is_closed(instance.user_story):
|
|
us_service.close_userstory(instance.user_story)
|
|
else:
|
|
us_service.open_userstory(instance.user_story)
|
|
|
|
|
|
# Milestone
|
|
def _try_to_close_or_open_milestone_when_create_or_edit_task(instance):
|
|
from taiga.projects.milestones import services as milestone_service
|
|
|
|
if instance.milestone_id:
|
|
if milestone_service.calculate_milestone_is_closed(instance.milestone):
|
|
milestone_service.close_milestone(instance.milestone)
|
|
else:
|
|
milestone_service.open_milestone(instance.milestone)
|
|
|
|
if instance.prev and instance.prev.milestone_id and instance.prev.milestone_id != instance.milestone_id:
|
|
if milestone_service.calculate_milestone_is_closed(instance.prev.milestone):
|
|
milestone_service.close_milestone(instance.prev.milestone)
|
|
else:
|
|
milestone_service.open_milestone(instance.prev.milestone)
|
|
|
|
|
|
def _try_to_close_milestone_when_delete_task(instance):
|
|
from taiga.projects.milestones import services
|
|
|
|
with suppress(ObjectDoesNotExist):
|
|
if instance.milestone_id and services.calculate_milestone_is_closed(instance.milestone):
|
|
services.close_milestone(instance.milestone)
|
|
|
|
####################################
|
|
# Signals for set finished date
|
|
####################################
|
|
|
|
def set_finished_date_when_edit_task(sender, instance, **kwargs):
|
|
if instance.status is None:
|
|
return
|
|
if instance.status.is_closed and not instance.finished_date:
|
|
instance.finished_date = timezone.now()
|
|
elif not instance.status.is_closed and instance.finished_date:
|
|
instance.finished_date = None
|