mygpo/suggestions/tasks.py

   1 from itertools import chain
   2 from collections import Counter
   3
   4 from django.contrib.auth import get_user_model
   5
   6 from django_db_geventpool.utils import close_connection
   7
   8 from celery import shared_task
   9
  10 from mygpo.celery import celery
  11 from mygpo.subscriptions import get_subscribed_podcasts
  12 from mygpo.suggestions.models import PodcastSuggestion
  13
  14 from celery.utils.log import get_task_logger
  15
  16 logger = get_task_logger(__name__)
  17
  18
  19 @shared_task
  20 @close_connection
  21 def update_suggestions(user_pk, max_suggestions=15):
  22     """updates the suggestions of a user"""
  23
  24     User = get_user_model()
  25     user = User.objects.get(pk=user_pk)
  26
  27     logger.info("Updating suggestions of user {user.username}".format(user=user))
  28
  29     # calculate possible suggestions
  30     subscribed_podcasts = [sp.podcast for sp in get_subscribed_podcasts(user)]
  31     logger.debug(
  32         "Found {num_podcasts} subscribed podcasts".format(
  33             num_podcasts=len(subscribed_podcasts)
  34         )
  35     )
  36
  37     # TODO: update related_podcasts of the subscribed_podcasts?
  38
  39     related = list(
  40         chain.from_iterable([p.related_podcasts.all() for p in subscribed_podcasts])
  41     )
  42
  43     # get most relevant
  44     counter = Counter(related)
  45     logger.debug(
  46         "Found {num_related} related podcasts".format(num_related=len(counter))
  47     )
  48
  49     suggested = [p for p, count in counter.most_common(max_suggestions)]
  50
  51     for suggested_podcast in suggested:
  52         ps, created = PodcastSuggestion.objects.get_or_create(
  53             suggested_to=user, podcast=suggested_podcast
  54         )
  55         if created:
  56             logger.info(
  57                 "Created suggestion for {podcast}".format(podcast=suggested_podcast)
  58             )
  59
  60     user.profile.suggestions_up_to_date = True
  61     user.profile.save()