mirrors/metube
1
0
Fork 0
mirror of https://github.com/alexta69/metube.git synced 2025-04-03 20:27:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

recursive add (download entire channels)

Browse source
This commit is contained in:
Alex 2019-12-13 19:22:44 +02:00
parent d5b1c37633
commit 34ec89485f
1 changed files with 29 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

45
app/ytdl.py
View file

@ -121,28 +121,41 @@ class DownloadQueue:
'extract_flat': True, 'extract_flat': True,
}).extract_info(url, download=False) }).extract_info(url, download=False)
async def add(self, url): async def __add_entry(self, entry, already):
log.info(f'adding {url}') etype = entry.get('_type') or 'video'
try:
info = await asyncio.get_running_loop().run_in_executor(None, self.__extract_info, url)
except youtube_dl.utils.YoutubeDLError as exc:
return {'status': 'error', 'msg': str(exc)}
etype = info.get('_type') or 'video'
if etype == 'playlist': if etype == 'playlist':
entries = info['entries'] entries = entry['entries']
log.info(f'playlist detected with {len(entries)} entries') log.info(f'playlist detected with {len(entries)} entries')
elif etype == 'video': results = []
entries = [info] for etr in entries:
log.info('single video detected') results.append(await self.__add_entry(etr, already))
else: if any(res['status'] == 'error' for res in results):
return {'status': 'error', 'msg': f'Unsupported resource requested: {etype}, please enter video or playlist URLs only'} return {'status': 'error', 'msg': ', '.join(res['msg'] for res in results if res['status'] == 'error' and 'msg' in res)}
for entry in entries: return {'status': 'ok'}
elif etype == 'video' or etype == 'url' and 'id' in entry:
if entry['id'] not in self.queue: if entry['id'] not in self.queue:
dl = DownloadInfo(entry['id'], entry['title'], entry.get('webpage_url') or entry['url']) dl = DownloadInfo(entry['id'], entry['title'], entry.get('webpage_url') or entry['url'])
self.queue[entry['id']] = Download(self.config.DOWNLOAD_DIR, dl) self.queue[entry['id']] = Download(self.config.DOWNLOAD_DIR, dl)
self.event.set()
await self.notifier.added(dl) await self.notifier.added(dl)
self.event.set() return {'status': 'ok'}
return {'status': 'ok'} elif etype == 'url':
return await self.add(entry['url'], already)
return {'status': 'error', 'msg': f'Unsupported resource "{etype}"'}
async def add(self, url, already=None):
log.info(f'adding {url}')
already = set() if already is None else already
if url in already:
log.info('recursion detected, skipping')
return {'status': 'ok'}
else:
already.add(url)
try:
entry = await asyncio.get_running_loop().run_in_executor(None, self.__extract_info, url)
except youtube_dl.utils.YoutubeDLError as exc:
return {'status': 'error', 'msg': str(exc)}
return await self.__add_entry(entry, already)
async def cancel(self, ids): async def cancel(self, ids):
for id in ids: for id in ids: