mygpo/episodestates/tasks.py

   1 from celery.utils.log import get_task_logger
   2
   3 from mygpo.celery import celery
   4 from mygpo.history.models import EpisodeHistoryEntry
   5 from mygpo.episodestates.models import EpisodeState
   6
   7 logger = get_task_logger(__name__)
   8
   9
  10 @celery.task
  11 def update_episode_state(historyentry_pk):
  12     """ Updates the episode state with the saved EpisodeHistoryEntry """
  13
  14     # previously an EpisodeHistoryEntry was passed as parameters directly;
  15     # as there can still be tasks like this in the queue, we should still
  16     # be able to handle it
  17     if isinstance(historyentry_pk, EpisodeHistoryEntry):
  18         historyentry = historyentry_pk
  19     else:
  20         historyentry = EpisodeHistoryEntry.objects.get(pk=historyentry_pk)
  21
  22     user = historyentry.user
  23     episode = historyentry.episode
  24
  25     logger.info('Updating Episode State for {user} / {episode}'.format(
  26         user=user, episode=episode))
  27
  28     state = EpisodeState.objects.update_or_create(
  29         user=user,
  30         episode=episode,
  31         defaults={
  32             'action': historyentry.action,
  33             'timestamp': historyentry.timestamp,
  34         }
  35     )