vlc/share/ytdl-extract.py

78 lines
2.0 KiB
Python
Executable File

#! /usr/bin/python3
#
# Copyright (C) 2020 Rémi Denis-Courmont
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation; either version 2.1 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
import sys
import json
import urllib.parse
import yt_dlp
class logger(object):
def debug(self, msg):
pass
def warning(self, msg):
pass
def error(self, msg):
sys.stderr.write(msg + '\n')
def url_extract(url):
opts = {
'extract_flat': 'in_playlist',
'logger': logger(),
'youtube_include_dash_manifest': False,
}
dl = yt_dlp.YoutubeDL(opts)
# Process a given URL
infos = dl.extract_info(url, download=False)
if 'entries' in infos:
for entry in infos['entries']:
if 'ie_key' in entry and entry['ie_key']:
# Flat-extracted playlist entry
url = 'ytdl:///?' + urllib.parse.urlencode(entry)
entry['url'] = url;
print(json.dumps(infos))
def url_process(ie_url):
opts = {
'logger': logger(),
'youtube_include_dash_manifest': False,
}
dl = yt_dlp.YoutubeDL(opts)
# Rebuild the original IE entry
entry = { }
for p in urllib.parse.parse_qsl(url[9:]):
entry[p[0]] = p[1]
infos = dl.process_ie_result(entry, download=False)
print(json.dumps(infos))
url = sys.argv[1]
if url.startswith('ytdl:///?'):
url_process(url)
else:
url_extract(url)