import uuid import json from datetime import datetime from pathlib import Path from django.conf import settings from django.db import models from django.utils.text import slugify from django.utils.translation import gettext_lazy as _ from .youtube import get_media_info as get_youtube_media_info from .utils import seconds_to_timestr, parse_media_format class Source(models.Model): ''' A Source is a source of media. Currently, this is either a YouTube channel or a YouTube playlist. ''' SOURCE_TYPE_YOUTUBE_CHANNEL = 'c' SOURCE_TYPE_YOUTUBE_PLAYLIST = 'p' SOURCE_TYPES = (SOURCE_TYPE_YOUTUBE_CHANNEL, SOURCE_TYPE_YOUTUBE_PLAYLIST) SOURCE_TYPE_CHOICES = ( (SOURCE_TYPE_YOUTUBE_CHANNEL, _('YouTube channel')), (SOURCE_TYPE_YOUTUBE_PLAYLIST, _('YouTube playlist')), ) SOURCE_RESOLUTION_360P = '360p' SOURCE_RESOLUTION_480P = '480p' SOURCE_RESOLUTION_720P = '720p' SOURCE_RESOLUTION_1080P = '1080p' SOURCE_RESOLUTION_1440P = '1440p' SOURCE_RESOLUTION_2160P = '2160p' 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_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_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_VCODEC_AVC1 = 'AVC1' SOURCE_VCODEC_VP9 = 'VP9' SOURCE_VCODECS = (SOURCE_VCODEC_AVC1, SOURCE_VCODEC_VP9) SOURCE_VCODECS_PRIORITY = (SOURCE_VCODEC_VP9, SOURCE_VCODEC_AVC1) SOURCE_VCODEC_CHOICES = ( (SOURCE_VCODEC_AVC1, _('AVC1 (H.264)')), (SOURCE_VCODEC_VP9, _('VP9')), ) SOURCE_ACODEC_MP4A = 'MP4A' SOURCE_ACODEC_OPUS = 'OPUS' SOURCE_ACODECS = (SOURCE_ACODEC_MP4A, SOURCE_ACODEC_OPUS) SOURCE_ACODEC_PRIORITY = (SOURCE_ACODEC_OPUS, SOURCE_ACODEC_MP4A) SOURCE_ACODEC_CHOICES = ( (SOURCE_ACODEC_MP4A, _('MP4A')), (SOURCE_ACODEC_OPUS, _('OPUS')), ) FALLBACK_FAIL = 'f' FALLBACK_NEXT_BEST = 'n' FALLBACK_NEXT_BEST_HD = 'h' FALLBACKS = (FALLBACK_FAIL, FALLBACK_NEXT_BEST, FALLBACK_NEXT_BEST_HD) FALLBACK_CHOICES = ( (FALLBACK_FAIL, _('Fail, do not download any media')), (FALLBACK_NEXT_BEST, _('Get next best resolution or codec instead')), (FALLBACK_NEXT_BEST_HD, _('Get next best resolution but at least HD')) ) # Fontawesome icons used for the source on the front end ICONS = { SOURCE_TYPE_YOUTUBE_CHANNEL: '', SOURCE_TYPE_YOUTUBE_PLAYLIST: '', } # Format to use to display a URL for the source URLS = { SOURCE_TYPE_YOUTUBE_CHANNEL: 'https://www.youtube.com/c/{key}', SOURCE_TYPE_YOUTUBE_PLAYLIST: 'https://www.youtube.com/playlist?list={key}', } # Callback functions to get a list of media from the source INDEXERS = { SOURCE_TYPE_YOUTUBE_CHANNEL: get_youtube_media_info, SOURCE_TYPE_YOUTUBE_PLAYLIST: get_youtube_media_info, } # Field names to find the media ID used as the key when storing media KEY_FIELD = { SOURCE_TYPE_YOUTUBE_CHANNEL: 'id', SOURCE_TYPE_YOUTUBE_PLAYLIST: 'id', } class IndexSchedule(models.IntegerChoices): EVERY_HOUR = 3600, _('Every hour') EVERY_2_HOURS = 7200, _('Every 2 hours') EVERY_3_HOURS = 10800, _('Every 3 hours') EVERY_4_HOURS = 14400, _('Every 4 hours') EVERY_5_HOURS = 18000, _('Every 5 hours') EVERY_6_HOURS = 21600, _('Every 6 hours') EVERY_12_HOURS = 43200, _('Every 12 hours') EVERY_24_HOURS = 86400, _('Every 24 hours') uuid = models.UUIDField( _('uuid'), primary_key=True, editable=False, default=uuid.uuid4, help_text=_('UUID of the source') ) created = models.DateTimeField( _('created'), auto_now_add=True, db_index=True, help_text=_('Date and time the source was created') ) last_crawl = models.DateTimeField( _('last crawl'), db_index=True, null=True, blank=True, help_text=_('Date and time the source was last crawled') ) source_type = models.CharField( _('source type'), max_length=1, db_index=True, choices=SOURCE_TYPE_CHOICES, default=SOURCE_TYPE_YOUTUBE_CHANNEL, help_text=_('Source type') ) key = models.CharField( _('key'), max_length=100, db_index=True, unique=True, help_text=_('Source key, such as exact YouTube channel name or playlist ID') ) name = models.CharField( _('name'), max_length=100, db_index=True, unique=True, help_text=_('Friendly name for the source, used locally in TubeSync only') ) directory = models.CharField( _('directory'), max_length=100, db_index=True, unique=True, help_text=_('Directory name to save the media into') ) index_schedule = models.IntegerField( _('index schedule'), choices=IndexSchedule.choices, db_index=True, default=IndexSchedule.EVERY_6_HOURS, help_text=_('Schedule of how often to index the source for new media') ) delete_old_media = models.BooleanField( _('delete old media'), default=False, help_text=_('Delete old media after "days to keep" days?') ) days_to_keep = models.PositiveSmallIntegerField( _('days to keep'), default=14, help_text=_('If "delete old media" is ticked, the number of days after which ' 'to automatically delete media') ) source_resolution = models.CharField( _('source resolution'), max_length=8, db_index=True, choices=SOURCE_RESOLUTION_CHOICES, default=SOURCE_RESOLUTION_1080P, help_text=_('Source resolution, desired video resolution to download') ) source_vcodec = models.CharField( _('source video codec'), max_length=8, db_index=True, choices=SOURCE_VCODEC_CHOICES, default=SOURCE_VCODEC_VP9, help_text=_('Source video codec, desired video encoding format to download (ignored if "resolution" is audio only)') ) source_acodec = models.CharField( _('source audio codec'), max_length=8, db_index=True, choices=SOURCE_ACODEC_CHOICES, default=SOURCE_ACODEC_OPUS, help_text=_('Source audio codec, desired audio encoding format to download') ) prefer_60fps = models.BooleanField( _('prefer 60fps'), default=True, help_text=_('Where possible, prefer 60fps media for this source') ) prefer_hdr = models.BooleanField( _('prefer hdr'), default=False, help_text=_('Where possible, prefer HDR media for this source') ) fallback = models.CharField( _('fallback'), max_length=1, db_index=True, choices=FALLBACK_CHOICES, default=FALLBACK_NEXT_BEST_HD, help_text=_('What do do when media in your source resolution and codecs is not available') ) has_failed = models.BooleanField( _('has failed'), default=False, help_text=_('Source has failed to index media') ) def __str__(self): return self.name class Meta: verbose_name = _('Source') verbose_name_plural = _('Sources') @property def icon(self): return self.ICONS.get(self.source_type) @property def is_audio(self): return self.source_resolution == self.SOURCE_RESOLUTION_AUDIO @property def is_video(self): return not self.is_audio @property def extension(self): ''' The extension is also used by youtube-dl to set the output container. As it is possible to quite easily pick combinations of codecs and containers which are invalid (e.g. OPUS audio in an MP4 container) just set this for people. All video is set to mkv containers, audio-only is set to m4a or ogg depending on audio codec. ''' if self.is_audio: if self.source_acodec == self.SOURCE_ACODEC_M4A: return 'm4a' elif self.source_acodec == self.SOURCE_ACODEC_OPUS: return 'ogg' else: raise ValueError('Unable to choose audio extension, uknown acodec') else: return 'mkv' @classmethod def create_url(obj, source_type, key): url = obj.URLS.get(source_type) return url.format(key=key) @property def url(self): return Source.create_url(self.source_type, self.key) @property def format_summary(self): if self.source_resolution == Source.SOURCE_RESOLUTION_AUDIO: vc = 'none' else: vc = self.source_vcodec ac = self.source_acodec f = '60FPS' if self.prefer_60fps else '' h = 'HDR' if self.prefer_hdr else '' return f'{self.source_resolution} (video:{vc}, audio:{ac}) {f} {h}'.strip() @property def directory_path(self): if self.source_resolution == self.SOURCE_RESOLUTION_AUDIO: return settings.SYNC_AUDIO_ROOT / self.directory else: return settings.SYNC_VIDEO_ROOT / self.directory @property def key_field(self): return self.KEY_FIELD.get(self.source_type, '') @property def source_resolution_height(self): return self.RESOLUTION_MAP.get(self.source_resolution, 0) @property def can_fallback(self): return self.fallback != self.FALLBACK_FAIL def index_media(self): ''' Index the media source returning a list of media metadata as dicts. ''' indexer = self.INDEXERS.get(self.source_type, None) if not callable(indexer): raise Exception(f'Source type f"{self.source_type}" has no indexer') response = indexer(self.url) # Account for nested playlists, such as a channel of playlists of playlists def _recurse_playlists(playlist): videos = [] if not playlist: return videos entries = playlist.get('entries', []) for entry in entries: if not entry: continue subentries = entry.get('entries', []) if subentries: videos = videos + _recurse_playlists(entry) else: videos.append(entry) return videos return _recurse_playlists(response) def get_media_thumb_path(instance, filename): fileid = str(instance.uuid) filename = f'{fileid.lower()}.jpg' prefix = fileid[:2] return Path('thumbs') / prefix / filename _metadata_cache = {} class Media(models.Model): ''' Media is a single piece of media, such as a single YouTube video linked to a Source. ''' # Format to use to display a URL for the media URLS = { Source.SOURCE_TYPE_YOUTUBE_CHANNEL: 'https://www.youtube.com/watch?v={key}', Source.SOURCE_TYPE_YOUTUBE_PLAYLIST: 'https://www.youtube.com/watch?v={key}', } # Maps standardised names to names used in source metdata METADATA_FIELDS = { 'upload_date': { Source.SOURCE_TYPE_YOUTUBE_CHANNEL: 'upload_date', Source.SOURCE_TYPE_YOUTUBE_PLAYLIST: 'upload_date', }, 'title': { Source.SOURCE_TYPE_YOUTUBE_CHANNEL: 'title', Source.SOURCE_TYPE_YOUTUBE_PLAYLIST: 'title', }, 'description': { Source.SOURCE_TYPE_YOUTUBE_CHANNEL: 'description', Source.SOURCE_TYPE_YOUTUBE_PLAYLIST: 'description', }, 'duration': { Source.SOURCE_TYPE_YOUTUBE_CHANNEL: 'duration', Source.SOURCE_TYPE_YOUTUBE_PLAYLIST: 'duration', }, 'formats': { Source.SOURCE_TYPE_YOUTUBE_CHANNEL: 'formats', Source.SOURCE_TYPE_YOUTUBE_PLAYLIST: 'formats', } } uuid = models.UUIDField( _('uuid'), primary_key=True, editable=False, default=uuid.uuid4, help_text=_('UUID of the media') ) created = models.DateTimeField( _('created'), auto_now_add=True, db_index=True, help_text=_('Date and time the media was created') ) source = models.ForeignKey( Source, on_delete=models.CASCADE, related_name='media_source', help_text=_('Source the media belongs to') ) published = models.DateTimeField( _('published'), db_index=True, null=True, blank=True, help_text=_('Date and time the media was published on the source') ) key = models.CharField( _('key'), max_length=100, db_index=True, help_text=_('Media key, such as exact YouTube video ID') ) thumb = models.ImageField( _('thumb'), upload_to=get_media_thumb_path, max_length=100, blank=True, null=True, width_field='thumb_width', height_field='thumb_height', help_text=_('Thumbnail') ) thumb_width = models.PositiveSmallIntegerField( _('thumb width'), blank=True, null=True, help_text=_('Width (X) of the thumbnail') ) thumb_height = models.PositiveSmallIntegerField( _('thumb height'), blank=True, null=True, help_text=_('Height (Y) of the thumbnail') ) metadata = models.TextField( _('metadata'), blank=True, null=True, help_text=_('JSON encoded metadata for the media') ) downloaded = models.BooleanField( _('downloaded'), db_index=True, default=False, help_text=_('Media has been downloaded') ) downloaded_audio_codec = models.CharField( _('downloaded audio codec'), max_length=30, db_index=True, blank=True, null=True, help_text=_('Audio codec of the downloaded media') ) downloaded_video_codec = models.CharField( _('downloaded video codec'), max_length=30, db_index=True, blank=True, null=True, help_text=_('Video codec of the downloaded media') ) downloaded_container = models.CharField( _('downloaded container format'), max_length=30, db_index=True, blank=True, null=True, help_text=_('Container format of the downloaded media') ) downloaded_fps = models.PositiveSmallIntegerField( _('downloaded fps'), db_index=True, blank=True, null=True, help_text=_('FPS of the downloaded media') ) downloaded_hdr = models.BooleanField( _('downloaded hdr'), default=False, help_text=_('Downloaded media has HDR') ) downloaded_filesize = models.PositiveBigIntegerField( _('downloaded filesize'), db_index=True, blank=True, null=True, help_text=_('Size of the downloaded media in bytes') ) def __str__(self): return self.key class Meta: verbose_name = _('Media') verbose_name_plural = _('Media') def get_metadata_field(self, field): fields = self.METADATA_FIELDS.get(field, {}) return fields.get(self.source.source_type, '') def iter_formats(self): for fmt in self.formats: yield parse_media_format(fmt) def get_best_combined_format(self): ''' Attempts to see if there is a single, combined audio and video format that exactly matches the source requirements. This is used over separate audio and video formats if possible. Combined formats are the easiest to check for as they must exactly match the source profile be be valid. ''' for fmt in self.iter_formats(): # Check height matches if self.source.source_resolution.strip().upper() != fmt['format']: continue # Check the video codec matches if self.source.source_vcodec != fmt['vcodec']: continue # Check the audio codec matches if self.source.source_acodec != fmt['acodec']: continue # if the source prefers 60fps, check for it if self.source.prefer_60fps: if not fmt['is_60fps']: continue # If the source prefers HDR, check for it if self.source.prefer_hdr: if not fmt['is_hdr']: continue # If we reach here, we have a combined match! return True, fmt['id'] return False, False def get_best_audio_format(self): ''' Finds the best match for the source required audio format. If the source has a 'fallback' of fail this can return no match. ''' # Order all audio-only formats by bitrate audio_formats = [] for fmt in self.iter_formats(): # If the format has a video stream, skip it if fmt['vcodec']: continue audio_formats.append(fmt) audio_formats = list(reversed(sorted(audio_formats, key=lambda k: k['abr']))) if not audio_formats: # Media has no audio formats at all return False, False # Find the highest bitrate audio format with a matching codec for fmt in audio_formats: if self.source.source_acodec == fmt['acodec']: # Matched! return True, fmt['id'] # No codecs matched if self.source.can_fallback: # Can fallback, find the next highest bitrate non-matching codec return False, audio_formats[0] else: # Can't fallback return False, False def get_best_video_format(self): ''' Finds the best match for the source required video format. If the source has a 'fallback' of fail this can return no match. Resolution is treated as the most important factor to match. ''' min_height = getattr(settings, 'VIDEO_HEIGHT_CUTOFF', 360) fallback_hd_cutoff = getattr(settings, 'VIDEO_HEIGHT_IS_HD', 500) # Filter video-only formats by resolution that matches the source video_formats = [] for fmt in self.iter_formats(): # If the format has an audio stream, skip it if fmt['acodec']: continue if self.source.source_resolution.strip().upper() == fmt['format']: video_formats.append(fmt) # Check we matched some streams if not video_formats: # No streams match the requested resolution, see if we can fallback if self.source.can_fallback: # Find the next-best format matches by height for fmt in self.iter_formats(): # If the format has an audio stream, skip it if fmt['acodec']: continue if (fmt['height'] <= self.source.source_resolution_height and fmt['height'] >= min_height): video_formats.append(fmt) else: # Can't fallback return False, False video_formats = list(reversed(sorted(video_formats, key=lambda k: k['height']))) if not video_formats: # Still no matches return False, False exact_match, best_match = None, None # Of our filtered video formats, check for resolution + codec + hdr + fps match if self.source.prefer_60fps and self.source.prefer_hdr: for fmt in video_formats: # Check for an exact match if (self.source.source_resolution.strip().upper() == fmt['format'] and self.source.source_vcodec == fmt['vcodec'] and fmt['is_hdr'] and fmt['is_60fps']): # Exact match exact_match, best_match = True, fmt break if self.source.can_fallback: if not best_match: for fmt in video_formats: # Check for a codec, hdr and fps match but drop the resolution if (self.source.source_vcodec == fmt['vcodec'] and fmt['is_hdr'] and fmt['is_60fps']): # Close match exact_match, best_match = False, fmt break if not best_match: for fmt in video_formats: # Check for hdr and fps match but drop the resolution and codec if fmt['is_hdr'] and fmt['is_60fps']: exact_match, best_match = False, fmt break if not best_match: for fmt in video_formats: # Check for fps match but drop the resolution and codec and hdr if fmt['is_hdr'] and fmt['is_60fps']: exact_match, best_match = False, fmt break if not best_match: # Match the highest resolution exact_match, best_match = False, video_formats[0] # Check for resolution + codec + fps match if self.source.prefer_60fps and not self.source.prefer_hdr: for fmt in video_formats: # Check for an exact match if (self.source.source_resolution.strip().upper() == fmt['format'] and self.source.source_vcodec == fmt['vcodec'] and fmt['is_60fps']): # Exact match exact_match, best_match = True, fmt break if self.source.can_fallback: if not best_match: for fmt in video_formats: # Check for a codec and fps match but drop the resolution if (self.source.source_vcodec == fmt['vcodec'] and fmt['is_60fps']): exact_match, best_match = False, fmt break if not best_match: for fmt in video_formats: # Check for an fps match but drop the resolution and codec if fmt['is_60fps']: exact_match, best_match = False, fmt break if not best_match: # Match the highest resolution exact_match, best_match = False, video_formats[0] # Check for resolution + codec + hdr if self.source.prefer_hdr and not self.source.prefer_60fps: for fmt in video_formats: # Check for an exact match if (self.source.source_resolution.strip().upper() == fmt['format'] and self.source.source_vcodec == fmt['vcodec'] and fmt['is_hdr']): # Exact match exact_match, best_match = True, fmt break if self.source.can_fallback: if not best_match: for fmt in video_formats: # Check for a codec and hdr match but drop the resolution if (self.source.source_vcodec == fmt['vcodec'] and fmt['is_hdr']): exact_match, best_match = True, fmt break if not best_match: for fmt in video_formats: # Check for an hdr match but drop the resolution and codec if fmt['is_hdr']: exact_match, best_match = False, fmt break if not best_match: # Match the highest resolution exact_match, best_match = False, video_formats[0] # check for resolution + codec if not self.source.prefer_hdr and not self.source.prefer_60fps: for fmt in video_formats: # Check for an exact match if (self.source.source_resolution.strip().upper() == fmt['format'] and self.source.source_vcodec == fmt['vcodec'] and not fmt['is_60fps']): # Exact match exact_match, best_match = True, fmt break if self.source.can_fallback: if not best_match: for fmt in video_formats: # Check for a codec match without 60fps and drop the resolution if (self.source.source_vcodec == fmt['vcodec'] and not fmt['is_60fps']): exact_match, best_match = False, fmt break if not best_match: for fmt in video_formats: # Check for a codec match but drop the resolution if self.source.source_vcodec == fmt['vcodec']: # Close match exact_match, best_match = False, fmt break if not best_match: # Match the highest resolution exact_match, best_match = False, video_formats[0] # See if we found a match if best_match: # Final check to see if the match we found was good enough if exact_match: return True, best_match['id'] elif self.source.can_fallback: # Allow the fallback if it meets requirements if (self.source.fallback == self.source.FALLBACK_NEXT_BEST_HD and best_match['height'] >= fallback_hd_cutoff): return False, best_match['id'] elif self.source.fallback == self.source.FALLBACK_NEXT_BEST: return False, best_match['id'] # Nope, failed to find match return False, False def get_format_str(self): ''' Returns a youtube-dl compatible format string for the best matches combination of source requirements and available audio and video formats. ''' if self.source.is_audio: audio_format = self.get_best_audio_format() return 'a' else: combined_format = self.get_best_combined_format() if combined_format: return 'c' else: audio_format = self.get_best_audio_format() video_format = self.get_best_video_format() return 'a+v' return False @property def loaded_metadata(self): if self.pk in _metadata_cache: return _metadata_cache[self.pk] try: unpacked_data = json.loads(self.metadata) except Exception: return {} _metadata_cache[self.pk] = unpacked_data return _metadata_cache[self.pk] @property def url(self): url = self.URLS.get(self.source.source_type, '') return url.format(key=self.key) @property def title(self): field = self.get_metadata_field('title') return self.loaded_metadata.get(field, '').strip() @property def name(self): title = self.title return title if title else self.key @property def upload_date(self): field = self.get_metadata_field('upload_date') upload_date_str = self.loaded_metadata.get(field, '').strip() try: return datetime.strptime(upload_date_str, '%Y%m%d') except (AttributeError, ValueError) as e: return None @property def duration(self): field = self.get_metadata_field('duration') return int(self.loaded_metadata.get(field, 0)) @property def duration_formatted(self): duration = self.duration if duration > 0: return seconds_to_timestr(duration) return '??:??:??' @property def formats(self): field = self.get_metadata_field('formats') return self.loaded_metadata.get(field, []) @property def filename(self): upload_date = self.upload_date dateobj = upload_date if upload_date else self.created datestr = dateobj.strftime('%Y-%m-%d') source_name = slugify(self.source.name) name = slugify(self.name.replace('&', 'and').replace('+', 'and')) ext = self.source.extension fn = f'{datestr}_{source_name}_{name}'[:100] return f'{fn}.{ext}' @property def filepath(self): return self.source.directory_path / self.filename