From 427bf3ac82d49b077b950b11fe109a9d90972ad9 Mon Sep 17 00:00:00 2001 From: Bastiaan Welmers Date: Mon, 4 May 2020 23:29:57 +0200 Subject: [PATCH] Add get_short_name_of_transcoding_type() --- videodinges/models.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/videodinges/models.py b/videodinges/models.py index 461aea6..b534533 100644 --- a/videodinges/models.py +++ b/videodinges/models.py @@ -1,6 +1,6 @@ import os from datetime import datetime -from typing import NamedTuple, Optional +from typing import NamedTuple, Optional, Union from django.db import models from django.db.models import constraints @@ -83,3 +83,12 @@ def get_quality_by_name(name: str) -> Optional[Quality]: for quality in qualities: if quality.name == name: return quality + +def get_short_name_of_transcoding_type(transcoding_type: Union[str, TranscodingType]) -> str: + if isinstance(transcoding_type, str): + for type_ in transcoding_types: + if type_.name == transcoding_type: + transcoding_type = type_ + + if isinstance(transcoding_type, TranscodingType): + return transcoding_type.short_name \ No newline at end of file