-#!/usr/bin/env python
-from __future__ import unicode_literals
-
-import io
+#!/usr/bin/env python3
import optparse
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 yt_dlp.extractor import list_extractor_classes
def main():
parser = optparse.OptionParser(usage='%prog OUTFILE.md')
- options, args = parser.parse_args()
+ _, 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)
+
+ with open(args[0], 'w', encoding='utf-8') as outf:
+ outf.write(f'# Supported sites\n{out}\n')
if __name__ == '__main__':