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