2021-05-02 21:48:25 +12:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
# coding=utf-8
|
|
|
|
|
|
|
|
import logging
|
|
|
|
from typing import Optional
|
|
|
|
|
|
|
|
import youtube_dl
|
|
|
|
from praw.models import Submission
|
|
|
|
|
|
|
|
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 YoutubeDlFallback(BaseFallbackDownloader, Youtube):
|
|
|
|
def __init__(self, post: Submission):
|
|
|
|
super(YoutubeDlFallback, self).__init__(post)
|
|
|
|
|
|
|
|
def find_resources(self, authenticator: Optional[SiteAuthenticator] = None) -> list[Resource]:
|
|
|
|
out = super()._download_video({})
|
|
|
|
return [out]
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
def can_handle_link(url: str) -> bool:
|
|
|
|
yt_logger = logging.getLogger('youtube-dl')
|
|
|
|
yt_logger.setLevel(logging.CRITICAL)
|
|
|
|
with youtube_dl.YoutubeDL({
|
|
|
|
'logger': yt_logger,
|
|
|
|
}) as ydl:
|
|
|
|
try:
|
|
|
|
result = ydl.extract_info(url, download=False)
|
|
|
|
if result:
|
|
|
|
return True
|
2021-05-06 01:56:34 +12:00
|
|
|
except Exception as e:
|
|
|
|
logger.exception(e)
|
|
|
|
return False
|
2021-05-02 21:48:25 +12:00
|
|
|
return False
|