YouTube-DL for Pythonista - download YouTube videos on your iPhone/iPad!

youtube-dl.py

#!python3

'''
Directions:

    - install yt-dlp via Pip (e.g. using (StaSh)[https://github.com/ywangd/stash] - `pip install yt-dlp`)
    - add this script as a Share extension through Settings -> Share Extension Shortcuts
    - while watching a video in the YouTube site or app, just share the video to Pythonista and select this script
    - the video will download, and when it's done you can share the video file itself with any app (e.g. VLC)
Advanced usage:
    - if you specify -audio as the script argument, this script will download audio only
    - if you specify -stream as the script argument, this script will just grab the actual video URL and redirect you
    to VLC, which will stream the video (without interruptions or ads!)
'''

from __future__ import unicode_literals
import yt_dlp as youtube_dl
import appex
import console
import clipboard
import os
import sys

outdir = os.path.expanduser("~/Documents/Downloads")
try:
	os.mkdir(outdir)
except FileExistsError:
	pass
	
if appex.get_attachments():
	# e.g. share from YouTube app
	url = appex.get_attachments()[0]
elif appex.get_urls():
	# e.g. share from Safari
	url = appex.get_urls()[0]
elif appex.get_text():
	url = appex.get_text()
elif clipboard.get():
	url = clipboard.get()
	
print("URL: ", url)
if not url or not url.startswith("http"):
	url = input("No URL found - enter URL to download: ")
	
ydl_opts = {'outtmpl': os.path.join(outdir, '%(title)s.%(ext)s')}

if '--audio' in sys.argv[1:] or '-audio' in sys.argv[1:]:
	ydl_opts['format'] = 'bestaudio'
	
if '--stream' in sys.argv[1:] or '-stream' in sys.argv[1:]:
	with youtube_dl.YoutubeDL(ydl_opts) as ydl:
		info = ydl.extract_info(url, download=False)
		from objc_util import UIApplication, nsurl
		from urllib.parse import urlencode
		app = UIApplication.sharedApplication()
		params = urlencode({'url': info['formats'][-1]['url']})
		#app.openURL_(nsurl('vlc-x-callback://x-callback-url/stream?' + params))
		#app.openURL_(nsurl('infuse://x-callback-url/play?' + params))
		app.openURL_(nsurl(info['formats'][-1]['url']))
else:
	with youtube_dl.YoutubeDL(ydl_opts) as ydl:
		info = ydl.extract_info(url, download=True)
		filepath = ydl.prepare_filename(info)
		
	console.open_in(filepath)