add method to get emissions using a nonstop zone as playlist

[django-panik-emissions.git] / emissions / models.py

diff --git a/emissions/models.py b/emissions/models.py
index aba341d871bf9cfbd36ffe97a9a54fd1e2f643da..cd5783703f471f08450350c68204aac05abba36b 100644 (file)
--- a/emissions/models.py
+++ b/emissions/models.py
@@ -870,6 +870,14 @@ class Nonstop(models.Model):
     def get_public_label(self):
         return self.title.split('(')[0].strip()
 
+    def get_playlist_emissions(self):
+        emissions = {}
+        for recurring_playlist in self.recurring_playlist_zones.all().select_related(
+            'schedule', 'schedule__emission'
+        ):
+            emissions[recurring_playlist.schedule.emission_id] = recurring_playlist.schedule.emission
+        return emissions.values()
+
 
 class Focus(models.Model):
     title = models.CharField(_('Alternate Title'), max_length=50, null=True, blank=True)