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)