96f0ca0c73df9676caf5f8d5a47a507fad17125b
[wolnelektury.git] / apps / social / utils.py
1 from django.db.models import Q
2 from catalogue.models import Tag
3 from catalogue import utils
4 from catalogue.tasks import touch_tag
5
6
7 def likes(user, work):
8     return user.is_authenticated() and work.tags.filter(category='set', user=user).exists()
9
10
11 def get_set(user, name):
12     """Returns a tag for use by the user. Creates it, if necessary."""
13     try:
14         tag = Tag.objects.get(category='set', user=user, name=name)
15     except Tag.DoesNotExist:
16         tag = Tag.objects.create(category='set', user=user, name=name,
17                 slug=utils.get_random_hash(name), sort_key=name.lower())
18     return tag
19
20
21 def set_sets(user, work, sets):
22     """Set tags used for given work by a given user."""
23
24     old_sets = list(work.tags.filter(category='set', user=user))
25
26     work.tags = sets + list(
27             work.tags.filter(~Q(category='set') | ~Q(user=user)))
28
29     for shelf in [shelf for shelf in old_sets if shelf not in sets]:
30         touch_tag(shelf)
31     for shelf in [shelf for shelf in sets if shelf not in old_sets]:
32         touch_tag(shelf)
33
34     # delete empty tags
35     Tag.objects.filter(category='set', user=user, book_count=0).delete()