mygpo/publisher/models.py

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