taiga
/
taiga-back
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed auto creation of notify policy objects

remotes/origin/enhancement/email-actions
Jesús Espino 2014-08-01 13:39:45 +02:00
parent fa150250c9
commit 3f628eba52
2 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
taiga/projects/notifications/api.py
View File

@ -44,10 +44,9 @@ class NotifyPolicyViewSet(ModelCrudViewSet):
projects = Project.objects.filter( projects = Project.objects.filter(
Q(owner=self.request.user) | Q(owner=self.request.user) |
Q(memberships__user=self.request.user) Q(memberships__user=self.request.user)
) ).distinct()
for project in projects: for project in projects:
if not services.notify_policy_exists(project, self.request.user): services.create_notify_policy_if_not_exists(project, self.request.user, NotifyLevel.watch)
services.create_notify_policy(project, self.request.user, NotifyLevel.watch)
def get_queryset(self): def get_queryset(self):
self._build_needed_notify_policies() self._build_needed_notify_policies()
@ -55,5 +54,5 @@ class NotifyPolicyViewSet(ModelCrudViewSet):
qs = models.NotifyPolicy.objects.filter( qs = models.NotifyPolicy.objects.filter(
Q(project__owner=self.request.user) | Q(project__owner=self.request.user) |
Q(project__memberships__user=self.request.user) Q(project__memberships__user=self.request.user)
).order_by("project__name") )
return qs.distinct() return qs.distinct()

14
taiga/projects/notifications/services.py
View File

@ -52,6 +52,20 @@ def create_notify_policy(project, user, level=NotifyLevel.notwatch):
raise exc.IntegrityError("Notify exists for specified user and project") from e raise exc.IntegrityError("Notify exists for specified user and project") from e
def create_notify_policy_if_not_exists(project, user, level=NotifyLevel.notwatch):
"""
Given a project and user, create notification policy for it.
"""
model_cls = get_model("notifications", "NotifyPolicy")
try:
result = model_cls.objects.get_or_create(project=project,
user=user,
defaults={"notify_level": level})
return result[0]
except IntegrityError as e:
raise exc.IntegrityError("Notify exists for specified user and project") from e
def get_notify_policy(project, user): def get_notify_policy(project, user):
""" """
Get notification level for specified project and user. Get notification level for specified project and user.