hoelee/tubesync
1
0
Fork 0
mirror of https://github.com/meeb/tubesync.git synced 2025-06-22 13:06:34 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use a class method to generate the mapping dictionary

Browse Source
This commit is contained in:
tcely 2025-02-12 02:46:27 -05:00 committed by GitHub
parent e79218c208
commit b45675b13a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tubesync/sync/choices.py
View File

@ -1,5 +1,3 @@
from enum import Enum
from django.db import models
from django.utils.translation import gettext_lazy as _
@ -81,14 +79,19 @@ class SourceResolution(models.TextChoices):
VIDEO_2160P = '2160p', _('4320p (8K)')
VIDEO_4320P = '4320p', _('4320p (8K)')
@classmethod
def _integer_mapping(cls):
filter_func = lambda s: s.lower().endswith('0p')
return dict(zip(
filter(filter_func, cls.names),
map(
lambda s: int(s[:-1], base=10),
filter(filter_func, cls.values)
)
))
SourceResolutionInteger = dict()
for name in SourceResolution.names:
if name.endswith('0P'):
value = SourceResolution.__dict__.get(name, None)
if isinstance(value, Enum):
value = int(value.value[: -1])
SourceResolutionInteger.update({name: value})
SourceResolutionInteger = SourceResolution._integer_mapping()
# as stolen from: