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

Use SourceResolution

Browse Source
This commit is contained in:
tcely 2025-02-12 11:21:29 -05:00 committed by GitHub
parent 668a3c6604
commit 3675496cfa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
tubesync/sync/models.py
View File

@ -27,6 +27,7 @@ from .matching import (get_best_combined_format, get_best_audio_format,
from .mediaservers import PlexMediaServer from .mediaservers import PlexMediaServer
from .fields import CommaSepChoiceField from .fields import CommaSepChoiceField
from .choices import (CapChoices, Fallback, IndexSchedule, MediaServerType, from .choices import (CapChoices, Fallback, IndexSchedule, MediaServerType,
SourceResolution, SourceResolutionInteger,
SponsorBlock_Category, YouTube_SourceType) SponsorBlock_Category, YouTube_SourceType)
media_file_storage = FileSystemStorage(location=str(settings.DOWNLOAD_ROOT), base_url='/media-data/') media_file_storage = FileSystemStorage(location=str(settings.DOWNLOAD_ROOT), base_url='/media-data/')
@ -41,37 +42,10 @@ class Source(models.Model):
SOURCE_TYPE_YOUTUBE_CHANNEL_ID = YouTube_SourceType.CHANNEL_ID.value SOURCE_TYPE_YOUTUBE_CHANNEL_ID = YouTube_SourceType.CHANNEL_ID.value
SOURCE_TYPE_YOUTUBE_PLAYLIST = YouTube_SourceType.PLAYLIST.value SOURCE_TYPE_YOUTUBE_PLAYLIST = YouTube_SourceType.PLAYLIST.value
SOURCE_RESOLUTION_360P = '360p' SOURCE_RESOLUTION_1080P = SourceResolution.VIDEO_1080P.value
SOURCE_RESOLUTION_480P = '480p' SOURCE_RESOLUTION_AUDIO = SourceResolution.AUDIO.value
SOURCE_RESOLUTION_720P = '720p' SOURCE_RESOLUTIONS = SourceResolution.values
SOURCE_RESOLUTION_1080P = '1080p' RESOLUTION_MAP = SourceResolutionInteger
SOURCE_RESOLUTION_1440P = '1440p'
SOURCE_RESOLUTION_2160P = '2160p'
SOURCE_RESOLUTION_4320P = '4320p'
SOURCE_RESOLUTION_AUDIO = 'audio'
SOURCE_RESOLUTIONS = (SOURCE_RESOLUTION_360P, SOURCE_RESOLUTION_480P,
SOURCE_RESOLUTION_720P, SOURCE_RESOLUTION_1080P,
SOURCE_RESOLUTION_1440P, SOURCE_RESOLUTION_2160P,
SOURCE_RESOLUTION_4320P, SOURCE_RESOLUTION_AUDIO)
SOURCE_RESOLUTION_CHOICES = (
(SOURCE_RESOLUTION_360P, _('360p (SD)')),
(SOURCE_RESOLUTION_480P, _('480p (SD)')),
(SOURCE_RESOLUTION_720P, _('720p (HD)')),
(SOURCE_RESOLUTION_1080P, _('1080p (Full HD)')),
(SOURCE_RESOLUTION_1440P, _('1440p (2K)')),
(SOURCE_RESOLUTION_2160P, _('2160p (4K)')),
(SOURCE_RESOLUTION_4320P, _('4320p (8K)')),
(SOURCE_RESOLUTION_AUDIO, _('Audio only')),
)
RESOLUTION_MAP = {
SOURCE_RESOLUTION_360P: 360,
SOURCE_RESOLUTION_480P: 480,
SOURCE_RESOLUTION_720P: 720,
SOURCE_RESOLUTION_1080P: 1080,
SOURCE_RESOLUTION_1440P: 1440,
SOURCE_RESOLUTION_2160P: 2160,
SOURCE_RESOLUTION_4320P: 4320,
}
SOURCE_VCODEC_AVC1 = 'AVC1' SOURCE_VCODEC_AVC1 = 'AVC1'
SOURCE_VCODEC_VP9 = 'VP9' SOURCE_VCODEC_VP9 = 'VP9'
@ -295,8 +269,8 @@ class Source(models.Model):
_('source resolution'), _('source resolution'),
max_length=8, max_length=8,
db_index=True, db_index=True,
choices=SOURCE_RESOLUTION_CHOICES, choices=SourceResolution.choices,
default=SOURCE_RESOLUTION_1080P, default=SourceResolution.VIDEO_1080P,
help_text=_('Source resolution, desired video resolution to download') help_text=_('Source resolution, desired video resolution to download')
) )
source_vcodec = models.CharField( source_vcodec = models.CharField(
@ -401,7 +375,7 @@ class Source(models.Model):
@property @property
def is_audio(self): def is_audio(self):
return self.source_resolution == self.SOURCE_RESOLUTION_AUDIO return self.source_resolution == SourceResolution.AUDIO.value
@property @property
def is_video(self): def is_video(self):
@ -461,7 +435,7 @@ class Source(models.Model):
@property @property
def format_summary(self): def format_summary(self):
if self.source_resolution == Source.SOURCE_RESOLUTION_AUDIO: if self.source_resolution == SourceResolution.AUDIO.value:
vc = 'none' vc = 'none'
else: else:
vc = self.source_vcodec vc = self.source_vcodec
@ -478,7 +452,7 @@ class Source(models.Model):
@property @property
def type_directory_path(self): def type_directory_path(self):
if settings.SOURCE_DOWNLOAD_DIRECTORY_PREFIX: if settings.SOURCE_DOWNLOAD_DIRECTORY_PREFIX:
if self.source_resolution == self.SOURCE_RESOLUTION_AUDIO: if self.source_resolution == SourceResolution.AUDIO.value:
return Path(settings.DOWNLOAD_AUDIO_DIR) / self.directory return Path(settings.DOWNLOAD_AUDIO_DIR) / self.directory
else: else:
return Path(settings.DOWNLOAD_VIDEO_DIR) / self.directory return Path(settings.DOWNLOAD_VIDEO_DIR) / self.directory