Cleaning: timezone issues, deprecated urls.py imports, missing notes.
[wolnelektury.git] / apps / 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 django.db.models import Q
6 from catalogue.models import Book, Tag
7 from catalogue import utils
8 from catalogue.tasks import touch_tag
9 from social.models import Cite
10
11
12 def likes(user, work):
13     return user.is_authenticated() and work.tags.filter(category='set', user=user).exists()
14
15
16 def get_set(user, name):
17     """Returns a tag for use by the user. Creates it, if necessary."""
18     try:
19         tag = Tag.objects.get(category='set', user=user, name=name)
20     except Tag.DoesNotExist:
21         tag = Tag.objects.create(category='set', user=user, name=name,
22                 slug=utils.get_random_hash(name), sort_key=name.lower())
23     return tag
24
25
26 def set_sets(user, work, sets):
27     """Set tags used for given work by a given user."""
28
29     old_sets = list(work.tags.filter(category='set', user=user))
30
31     work.tags = sets + list(
32             work.tags.filter(~Q(category='set') | ~Q(user=user)))
33
34     for shelf in [shelf for shelf in old_sets if shelf not in sets]:
35         touch_tag(shelf)
36     for shelf in [shelf for shelf in sets if shelf not in old_sets]:
37         touch_tag(shelf)
38
39     # delete empty tags
40     Tag.objects.filter(category='set', user=user, book_count=0).delete()
41
42
43 def cites_for_tags(tags):
44     """Returns a QuerySet with all Cites for books with given tags."""
45     books = Book.tagged.with_all(tags).order_by().values_list('id', flat=True)
46     books = list(books)
47     return Cite.objects.filter(book__id__in=books)