mygpo/publisher/models.py

   1 """ This module contains models for the publisher pages """
   2
   3
   4
   5 from django.db import models
   6 from django.conf import settings
   7
   8 from mygpo.podcasts.models import Podcast
   9
  10 import logging
  11 logger = logging.getLogger(__name__)
  12
  13
  14 class PublishedPodcastManager(models.Manager):
  15     """ Manager for the PublishedPodcast model """
  16
  17     def publish_podcasts(self, user, podcasts):
  18         existed, created = 0, 0
  19         for podcast in podcasts:
  20             pp, new = PublishedPodcast.objects.get_or_create(
  21                 publisher=user,
  22                 podcast=podcast,
  23             )
  24
  25             if new:
  26                 created += 1
  27                 logger.info('Created publisher permissions for %r on %r',
  28                             user, podcast)
  29             else:
  30                 existed += 1
  31                 logger.info('Publisher permissions for %r on %r already exist',
  32                             user, podcast)
  33
  34
  35         return created, existed
  36
  37
  38 class PublishedPodcast(models.Model):
  39     publisher = models.ForeignKey(settings.AUTH_USER_MODEL,
  40                                   on_delete=models.CASCADE)
  41     podcast = models.ForeignKey(Podcast, on_delete=models.CASCADE)
  42
  43     class Meta:
  44         unique_together = (
  45             ('publisher', 'podcast'),
  46         )
  47
  48     objects = PublishedPodcastManager()