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

Improve queries and transaction management for better testability.

remotes/origin/enhancement/email-actions
Andrey Antukh 2014-05-28 18:26:05 +02:00
parent 61aec4676d
commit 21e5280eec
1 changed files with 22 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
taiga/projects/stars/services.py
View File

@ -6,7 +6,6 @@ from django.contrib.auth import get_user_model
from .models import Fan, Stars
@atomic
def star(project, user):
"""Star a project for an user.
@ -16,16 +15,21 @@ def star(project, user):
:param project: :class:`~taiga.projects.models.Project` instance.
:param user: :class:`~taiga.users.models.User` instance.
"""
if not Fan.objects.filter(project=project, user=user).exists():
Fan.objects.create(project=project, user=user)
with atomic():
fan, created = Fan.objects.get_or_create(project=project,
user=user)
if not created:
return
stars, _ = Stars.objects.get_or_create(project=project)
stars.count = F('count') + 1
stars.save()
@atomic
def unstar(project, user):
"""Unstar a project for an user.
"""
Unstar a project for an user.
If the user has not starred the project nothing happens so this function can be considered
idempotent.
@ -33,16 +37,25 @@ def unstar(project, user):
:param project: :class:`~taiga.projects.models.Project` instance.
:param user: :class:`~taiga.users.models.User` instance.
"""
if Fan.objects.filter(project=project, user=user).exists():
Fan.objects.filter(project=project, user=user).delete()
with atomic():
qs = Fan.objects.filter(project=project, user=user)
if not qs.exists():
return
qs.delete()
stars, _ = Stars.objects.get_or_create(project=project)
stars.count = F('count') - 1
stars.save()
def get_stars(project):
"""Get the count of stars a project have."""
return Stars.objects.filter(project=project).count
"""
Get the count of stars a project have.
"""
instance, _ = Stars.objects.get_or_create(project=project)
return instance.count
def get_fans(project_or_id):