28 lines
705 B
Python
28 lines
705 B
Python
from common import logger
|
|
from engine.plugins import FFmpegdl
|
|
from ykdl.common import url_to_module
|
|
# from engine.plugins import SDownload
|
|
|
|
|
|
# class Generic(SDownload):
|
|
# pass
|
|
class Generic(FFmpegdl):
|
|
def __init__(self, fname, url, suffix='flv'):
|
|
super().__init__(fname, url, suffix)
|
|
|
|
def check_stream(self):
|
|
logger.debug(self.fname)
|
|
site, url = url_to_module(self.url)
|
|
try:
|
|
info = site.parser(url)
|
|
except AssertionError:
|
|
return
|
|
stream_id = info.stream_types[0]
|
|
urls = info.streams[stream_id]['src']
|
|
self.ydl_opts['absurl'] = urls[0]
|
|
# print(info.title)
|
|
return True
|
|
|
|
|
|
__plugin__ = Generic
|