From bc0ac0e48780c9964caf39be0469063aba37c434 Mon Sep 17 00:00:00 2001
From: David Trattnig <david.trattnig@o94.at>
Date: Mon, 8 Jun 2020 13:05:49 +0200
Subject: [PATCH] Utility to get playlist type by ID.

---
 modules/base/utils.py | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/modules/base/utils.py b/modules/base/utils.py
index b749a409..873eaec7 100644
--- a/modules/base/utils.py
+++ b/modules/base/utils.py
@@ -21,7 +21,7 @@ import datetime
 import time
 
 from enum import Enum
-from modules.base.enum import Channel, ChannelType
+from modules.base.enum import Channel, ChannelType, PlaylistType
 
 
 
@@ -67,6 +67,28 @@ class EngineUtil:
 
 
 
+    @staticmethod
+    def get_playlist_type(id):
+        """
+        Converts an playlist type ID to the playlist type object.
+
+        Args:
+            id (String):    playlist type ID
+
+        Returns:
+            type (PlaylistType):  The type
+        """
+        if id == 0:
+            return PlaylistType.DEFAULT
+        elif id == 1:
+            return PlaylistType.SHOW
+        elif id == 2:
+            return PlaylistType.TIMESLOT
+        else:
+            return PlaylistType.STATION
+
+
+
     @staticmethod
     def get_entries_string(entries):
         """
-- 
GitLab