jodhus/spradio-server-django
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add functions to grab random songs/jingles.

Browse source
This commit is contained in:
Josh Washburne 2018-03-28 14:43:03 -04:00
parent 4138015054
commit 0fb7f148bb
1 changed files with 24 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
savepointradio/radio/managers.py
View file

@ -1,5 +1,6 @@
from datetime import timedelta from datetime import timedelta
from decimal import getcontext, Decimal, ROUND_UP from decimal import getcontext, Decimal, ROUND_UP
from random import randint
from django.apps import apps from django.apps import apps
from django.db import models from django.db import models
@ -23,7 +24,13 @@ class SongManager(models.Manager):
""" """
return SongQuerySet(self.model, using=self._db) return SongQuerySet(self.model, using=self._db)
def available(self): def available_jingles(self):
"""
Jingles that are currently published and are enabled.
"""
return self.get_queryset().jingles().enabled().published()
def available_songs(self):
""" """
Songs that are currently published and are enabled. Songs that are currently published and are enabled.
""" """
@ -33,7 +40,8 @@ class SongManager(models.Manager):
""" """
Total length of available songs in the playlist (in seconds). Total length of available songs in the playlist (in seconds).
""" """
return self.available().aggregate(models.Sum('length'))['length__sum'] length = self.available_songs().aggregate(models.Sum('length'))
return length['length__sum']
def wait_total(self): def wait_total(self):
""" """
@ -56,7 +64,7 @@ class SongManager(models.Manager):
published) and they have not been played within the default wait time published) and they have not been played within the default wait time
(or at all). (or at all).
""" """
return self.available().filter( return self.available_songs().filter(
models.Q(last_played__lt=self.datetime_from_wait()) | models.Q(last_played__lt=self.datetime_from_wait()) |
models.Q(last_played__isnull=True) models.Q(last_played__isnull=True)
) )
@ -71,3 +79,16 @@ class SongManager(models.Manager):
requests = SongRequest.music.unplayed().values_list('song__id', requests = SongRequest.music.unplayed().values_list('song__id',
flat=True) flat=True)
return self.playable().exclude(id__in=requests) return self.playable().exclude(id__in=requests)
def get_random_requestable_song(self):
"""
Pick a random requestable song and return it.
"""
return self.requestable()[randint(0, self.requestable().count() - 1)]
def get_random_jingle(self):
"""
Pick a random jingle and return it.
"""
random_index = randint(0, self.available_jingles().count() - 1)
return self.available_jingles()[random_index]