79 lines
3.3 KiB
Python
79 lines
3.3 KiB
Python
|
from django.conf import settings
|
||
|
|
from django.core.validators import (MaxLengthValidator, MinValueValidator,
|
||
|
|
MaxValueValidator)
|
||
|
|
from django.db import models
|
||
|
|
from django.db.models.signals import post_save
|
||
|
|
from django.dispatch import receiver
|
||
|
|
from django.utils.translation import ugettext_lazy as _
|
||
|
|
|
||
|
|
from core.behaviors import Timestampable
|
||
|
|
from radio.models import Song
|
||
|
|
from .managers import RequestManager
|
||
|
|
|
||
|
|
|
||
|
|
class RadioProfile(Timestampable, models.Model):
|
||
|
|
user = models.OneToOneField(settings.AUTH_USER_MODEL,
|
||
|
|
on_delete=models.CASCADE,
|
||
|
|
null=True,
|
||
|
|
blank=True)
|
||
|
|
favorites = models.ManyToManyField(Song, related_name='song_favorites')
|
||
|
|
ratings = models.ManyToManyField(Song,
|
||
|
|
related_name='song_ratings',
|
||
|
|
through='Rating')
|
||
|
|
song_requests = models.ManyToManyField(Song,
|
||
|
|
related_name='song_requests',
|
||
|
|
through='SongRequest')
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return "{}'s profile".format(self.user.get_username())
|
||
|
|
|
||
|
|
|
||
|
|
@receiver(post_save, sender=settings.AUTH_USER_MODEL)
|
||
|
|
def create_profile(sender, instance, created, **kwargs):
|
||
|
|
if created:
|
||
|
|
profile, new = RadioProfile.objects.get_or_create(user=instance)
|
||
|
|
|
||
|
|
|
||
|
|
class Rating(Timestampable, models.Model):
|
||
|
|
profile = models.ForeignKey(RadioProfile,
|
||
|
|
on_delete=models.CASCADE,
|
||
|
|
related_name='rating_profile')
|
||
|
|
song = models.ForeignKey(Song, on_delete=models.CASCADE)
|
||
|
|
value = models.PositiveIntegerField(_('song rating'),
|
||
|
|
validators=[MinValueValidator(1),
|
||
|
|
MaxValueValidator(5)])
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return "{} - {}'s Rating: {}".format(self.song.title,
|
||
|
|
self.profile.user.get_username(),
|
||
|
|
self.value)
|
||
|
|
|
||
|
|
|
||
|
|
class SongRequest(Timestampable, models.Model):
|
||
|
|
profile = models.ForeignKey(RadioProfile,
|
||
|
|
on_delete=models.SET_NULL,
|
||
|
|
null=True,
|
||
|
|
blank=True,
|
||
|
|
related_name='request_profile')
|
||
|
|
song = models.ForeignKey(Song,
|
||
|
|
on_delete=models.SET_NULL,
|
||
|
|
null=True,
|
||
|
|
blank=True)
|
||
|
|
queued_at = models.DateTimeField(_('song queued at'),
|
||
|
|
default=None,
|
||
|
|
blank=True,
|
||
|
|
null=True)
|
||
|
|
played_at = models.DateTimeField(_('song played at'),
|
||
|
|
default=None,
|
||
|
|
blank=True,
|
||
|
|
null=True)
|
||
|
|
|
||
|
|
objects = models.Manager()
|
||
|
|
music = RequestManager()
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
req_user = self.profile.user.get_username()
|
||
|
|
return "{} - Requested by {} at {}".format(self.song.title,
|
||
|
|
req_user,
|
||
|
|
self.created_date)
|