mirror of
https://github.com/anatolykopyl/youtube-cdl.git
synced 2026-03-26 12:55:11 +00:00
48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
from rich.console import Console
|
|
#from rich import inspect
|
|
from pathlib import Path
|
|
import os
|
|
import json
|
|
from getch import getch
|
|
import keyboard
|
|
import youtube_dl
|
|
from get_channels import retrieve_youtube_subscriptions
|
|
|
|
output_dir = 'output'
|
|
c = Console()
|
|
ydl_opts = {
|
|
'format': 'best',
|
|
'outtmpl': '%(id)s.%(ext)s'
|
|
}
|
|
|
|
all_channels = retrieve_youtube_subscriptions()
|
|
curr_channel = 0
|
|
c.print(f'You will be prompted if you want to download \
|
|
a channel for each of your subscriptions. (total {len(all_channels)})')
|
|
for ch in all_channels:
|
|
curr_channel += 1
|
|
c.print(f'[dim][{curr_channel}/{len(all_channels)}]:[/dim] {ch["title"]} [cyan]\[y/n]')
|
|
while True:
|
|
key = getch()
|
|
if key == "y":
|
|
ch['download'] = True
|
|
break
|
|
elif key == "n":
|
|
ch['download'] = False
|
|
break
|
|
else:
|
|
c.print('Press "y" or "n"', style='orange')
|
|
|
|
c.print('All done! 🎉')
|
|
c.print('Saving to output.json...', style='italic')
|
|
f = open("output.json", "a", encoding='utf-8')
|
|
f.write(json.dumps(all_channels))
|
|
f.close()
|
|
|
|
for ch in all_channels:
|
|
if ch['download']:
|
|
ydl_opts['outtmpl'] = '{}/{}/%(id)s.%(ext)s'.format(output_dir, ch['title'])
|
|
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
|
|
Path(os.path.join(output_dir, ch['title'])).mkdir(parents=True, exist_ok=True)
|
|
ydl.download(['https://www.youtube.com/channel/{}'.format(ch["id"])])
|