mirror of
https://github.com/anatolykopyl/youtube-cdl.git
synced 2026-03-26 12:55:11 +00:00
✨ Initial commit
This commit is contained in:
47
main.py
Normal file
47
main.py
Normal file
@@ -0,0 +1,47 @@
|
||||
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"])])
|
||||
Reference in New Issue
Block a user