[lazy_extractors] Fix for search IEs

[yt-dlp.git] / devscripts / make_readme.py

diff --git a/devscripts/make_readme.py b/devscripts/make_readme.py
index a24b9b1d805d111d6c4c41c65a37e33736fe9ec0..3f56af74470b5dbb3396a5fc52c29cac77f40e57 100755 (executable)
--- a/devscripts/make_readme.py
+++ b/devscripts/make_readme.py
@@ -1,3 +1,10 @@
+#!/usr/bin/env python3
+
+# yt-dlp --help | make_readme.py
+# This must be run in a console of correct width
+
+from __future__ import unicode_literals
+
 import io
 import sys
 import re
@@ -5,15 +12,18 @@
 README_FILE = 'README.md'
 helptext = sys.stdin.read()
 
+if isinstance(helptext, bytes):
+    helptext = helptext.decode('utf-8')
+
 with io.open(README_FILE, encoding='utf-8') as f:
     oldreadme = f.read()
 
-header = oldreadme[:oldreadme.index('# OPTIONS')]
+header = oldreadme[:oldreadme.index('## General Options:')]
 footer = oldreadme[oldreadme.index('# CONFIGURATION'):]
 
-options = helptext[helptext.index('  General Options:')+19:]
-options = re.sub(r'^  (\w.+)$', r'## \1', options, flags=re.M)
-options = '# OPTIONS\n' + options + '\n'
+options = helptext[helptext.index('  General Options:'):]
+options = re.sub(r'(?m)^  (\w.+)$', r'## \1', options)
+options = options + '\n'
 
 with io.open(README_FILE, 'w', encoding='utf-8') as f:
     f.write(header)