#!/usr/bin/env python3 # -*- coding: utf-8 -*- import logging from typing import Optional from praw.models import Submission from bdfr.exceptions import NotADownloadableLinkError from bdfr.resource import Resource from bdfr.site_authenticator import SiteAuthenticator from bdfr.site_downloaders.fallback_downloaders.fallback_downloader import BaseFallbackDownloader from bdfr.site_downloaders.youtube import Youtube logger = logging.getLogger(__name__) class YtdlpFallback(BaseFallbackDownloader, Youtube): def __init__(self, post: Submission): super(YtdlpFallback, self).__init__(post) def find_resources(self, authenticator: Optional[SiteAuthenticator] = None) -> list[Resource]: logger.debug(f"YtdlpFallback processing URL: {self.post.url}") video_attrs = super().get_video_attributes(self.post.url) logger.debug(f"Video attributes: {video_attrs}") extension = video_attrs.get("ext", None) logger.debug(f"Using extension: {extension}") out = Resource( self.post, self.post.url, super()._download_video({}), extension, ) logger.debug(f"Created resource with extension: {out.extension}") return [out] @staticmethod def can_handle_link(url: str) -> bool: try: attributes = YtdlpFallback.get_video_attributes(url) except NotADownloadableLinkError: return False if attributes: return True