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

youtube-dl.py

#!python3

'''
Directions:
    
    - install youtube-dl via Pip (e.g. using the StaSh command: https://github.com/ywangd/stash)
    - 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 --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 youtube_dl
import appex
import console
import clipboard
import os
import sys


class MyLogger(object):
    def debug(self, msg):
        pass

    def warning(self, msg):
        pass

    def error(self, msg):
        print(msg)


def postdownload(d):
	if d['status'] == 'finished':
        print('Done downloading  ...')
		console.quicklook(d['filename'])
		remove(d['filename'])
		appex.finish()


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(f"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'),
	'format': 'bestaudio/best',
	'nocheckcertificate': True,
    'logger': MyLogger(),
	# source for postdownload
	# https://gist.github.com/solarfl4re/0a3647788f8ec2f375f2da3db55fb0fe
	'progress_hooks': [postdownload]
}

if sys.argv[1:] == ['--stream']:
	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))
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)