bf1c24229c074bd3586ed983ee25f67c907055e3
[wolnelektury.git] / src / social / utils.py
1 # -*- coding: utf-8 -*-
2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
4 #
5 from collections import defaultdict
6 from django.contrib.contenttypes.models import ContentType
7 from django.db.models import Q
8 from django.utils.functional import lazy
9 from catalogue.models import Book, Tag
10 from catalogue import utils
11 from catalogue.tasks import touch_tag
12 from social.models import Cite
13
14
15 def likes(user, work, request=None):
16     if not user.is_authenticated():
17         return False
18
19     if request is None:
20         return work.tags.filter(category='set', user=user).exists()
21
22     if not hasattr(request, 'social_likes'):
23         # tuple: unchecked, checked, liked
24         request.social_likes = defaultdict(lambda:(set(), set(), set()))
25
26     ct = ContentType.objects.get_for_model(type(work))
27     likes_t = request.social_likes[ct.pk]
28     if work.pk in likes_t[1]:
29         return work.pk in likes_t[2]
30     else:
31         likes_t[0].add(work.pk)
32         def _likes():
33             if likes_t[0]:
34                 ids = tuple(likes_t[0])
35                 likes_t[0].clear()
36                 likes_t[2].update(Tag.intermediary_table_model.objects.filter(
37                     content_type_id=ct.pk, tag__user_id=user.pk,
38                     object_id__in=ids
39                 ).distinct().values_list('object_id', flat=True))
40                 likes_t[1].update(ids)
41             return work.pk in likes_t[2]
42         return lazy(_likes, bool)()
43
44
45 def get_set(user, name):
46     """Returns a tag for use by the user. Creates it, if necessary."""
47     try:
48         tag = Tag.objects.get(category='set', user=user, name=name)
49     except Tag.DoesNotExist:
50         tag = Tag.objects.create(category='set', user=user, name=name,
51                 slug=utils.get_random_hash(name), sort_key=name.lower())
52     return tag
53
54
55 def set_sets(user, work, sets):
56     """Set tags used for given work by a given user."""
57
58     old_sets = list(work.tags.filter(category='set', user=user))
59
60     work.tags = sets + list(
61             work.tags.filter(~Q(category='set') | ~Q(user=user)))
62
63     for shelf in [shelf for shelf in old_sets if shelf not in sets]:
64         touch_tag(shelf)
65     for shelf in [shelf for shelf in sets if shelf not in old_sets]:
66         touch_tag(shelf)
67
68     # delete empty tags
69     Tag.objects.filter(category='set', user=user, items=None).delete()
70
71
72 def cites_for_tags(tags):
73     """Returns a QuerySet with all Cites for books with given tags."""
74     return Cite.objects.filter(book__in=Book.tagged.with_all(tags))