X-Git-Url: https://jfr.im/git/yt-dlp.git/blobdiff_plain/310f762636e192a065696232bb5a0f15938398b1..add96eb9f84cfffe85682bf2fb85135746994ee8:/devscripts/make_supportedsites.py diff --git a/devscripts/make_supportedsites.py b/devscripts/make_supportedsites.py index 0ae6f8aa3..01548ef97 100644 --- a/devscripts/make_supportedsites.py +++ b/devscripts/make_supportedsites.py @@ -1,45 +1,19 @@ -#!/usr/bin/env python -from __future__ import unicode_literals +#!/usr/bin/env python3 -import io -import optparse +# Allow direct execution import os import sys +sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) -# Import youtube_dlc -ROOT_DIR = os.path.join(os.path.dirname(__file__), '..') -sys.path.insert(0, ROOT_DIR) -import youtube_dlc + +from devscripts.utils import get_filename_args, write_file +from yt_dlp.extractor import list_extractor_classes def main(): - parser = optparse.OptionParser(usage='%prog OUTFILE.md') - options, args = parser.parse_args() - if len(args) != 1: - parser.error('Expected an output filename') - - outfile, = args - - def gen_ies_md(ies): - for ie in ies: - ie_md = '**{0}**'.format(ie.IE_NAME) - ie_desc = getattr(ie, 'IE_DESC', None) - if ie_desc is False: - continue - if ie_desc is not None: - ie_md += ': {0}'.format(ie.IE_DESC) - if not ie.working(): - ie_md += ' (Currently broken)' - yield ie_md - - ies = sorted(youtube_dlc.gen_extractors(), key=lambda i: i.IE_NAME.lower()) - out = '# Supported sites\n' + ''.join( - ' - ' + md + '\n' - for md in gen_ies_md(ies)) - - with io.open(outfile, 'w', encoding='utf-8') as outf: - outf.write(out) + out = '\n'.join(ie.description() for ie in list_extractor_classes() if ie.IE_DESC is not False) + write_file(get_filename_args(), f'# Supported sites\n{out}\n') if __name__ == '__main__':