diff --git a/bin/ffx/track_delete_screen.py b/bin/ffx/track_delete_screen.py index 929de8d..e707214 100644 --- a/bin/ffx/track_delete_screen.py +++ b/bin/ffx/track_delete_screen.py @@ -48,13 +48,6 @@ class TrackDeleteScreen(Screen): } """ - TRACK_TYPE_LABELS = [ - 'video', - 'audio', - 'subtitle' - ] - - def __init__(self, trackType : TrackType, trackId = None): super().__init__() @@ -82,13 +75,11 @@ class TrackDeleteScreen(Screen): def compose(self): - typeLabel = TrackDeleteScreen.TRACK_TYPE_LABELS[int(self.trackType.value)-1] - yield Header() with Grid(): - yield Static(f"Are you sure to delete the following {typeLabel} stream?", id="toplabel", classes="two") + yield Static(f"Are you sure to delete the following {self.trackType.label()} stream?", id="toplabel", classes="two") # yield Static("", classes="two") # diff --git a/bin/ffx/track_details_screen.py b/bin/ffx/track_details_screen.py index 059d43f..85e3c14 100644 --- a/bin/ffx/track_details_screen.py +++ b/bin/ffx/track_details_screen.py @@ -62,12 +62,6 @@ class TrackDetailsScreen(Screen): } """ - TRACK_TYPE_LABELS = [ - 'video', - 'audio', - 'subtitle' - ] - def __init__(self, trackType : TrackType, streamId = None, patternId = None): super().__init__() @@ -144,14 +138,12 @@ class TrackDetailsScreen(Screen): # # self.subtitleStreamsTable.cursor_type = 'row' - typeLabel = TrackDetailsScreen.TRACK_TYPE_LABELS[int(self.trackType.value)-1] - yield Header() with Grid(): # 1 - yield Static(f"Edit {typeLabel} stream" if self.track_obj else f"New {typeLabel} stream", id="toplabel", classes="five") + yield Static(f"Edit {self.trackType.label()} stream" if self.track_obj else f"New {self.trackType.label()} stream", id="toplabel", classes="five") # yield Input(type="text", id="pattern_input", classes="four") # 2 diff --git a/bin/ffx/track_type.py b/bin/ffx/track_type.py index e76ce55..510bd7e 100644 --- a/bin/ffx/track_type.py +++ b/bin/ffx/track_type.py @@ -1,6 +1,19 @@ from enum import Enum class TrackType(Enum): + VIDEO = 1 AUDIO = 2 SUBTITLE = 3 + + + def label(self): + """Returns the stream type as string""" + + labels = { + TrackType.VIDEO: "video", + TrackType.AUDIO: "audio", + TrackType.SUBTITLE: "subtitle" + } + + return labels.get(self, "undefined")