File size: 3,059 Bytes
9dce458 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
#!/usr/bin/env python3
# Adapted from https://github.com/yt-dlp/yt-dlp/tree/master/devscripts
import os
import sys
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import functools
import re
from devscripts.utils import read_file, write_file
from manga_translator.args import HelpFormatter, parser
READMES = (
[
'README.md',
'## Options',
'<!-- Auto generated end -->',
],
[
'README_CN.md',
'## 选项',
'<!-- Auto generated end -->',
],
)
ALLOWED_OVERSHOOT = 2
DISABLE_PATCH = object()
HelpFormatter.INDENT_INCREMENT = 0
HelpFormatter.MAX_HELP_POSITION = 45
HelpFormatter.WIDTH = 100
def take_section(text, start=None, end=None, *, shift=0):
return text[
text.index(start) + shift if start else None:
text.index(end) + shift if end else None
]
def apply_patch(text, patch):
return text if patch[0] is DISABLE_PATCH else re.sub(*patch, text)
options = take_section(parser.format_help(), '\noptions:', shift=len('\noptions:'))
max_width = max(map(len, options.split('\n')))
switch_col_width = len(re.search(r'(?m)^\s{5,}', options).group())
delim = f'\n{" " * switch_col_width}'
PATCHES = (
# ( # Headings
# r'(?m)^ (\w.+\n)( (?=\w))?',
# r'## \1'
# ),
( # Fixup `--date` formatting
rf'(?m)( --date DATE.+({delim}[^\[]+)*)\[.+({delim}.+)*$',
(rf'\1[now|today|yesterday][-N[day|week|month|year]].{delim}'
f'E.g. "--date today-2weeks" downloads only{delim}'
'videos uploaded on the same day two weeks ago'),
),
( # Do not split URLs
rf'({delim[:-1]})? (?P<label>\[\S+\] )?(?P<url>https?({delim})?:({delim})?/({delim})?/(({delim})?\S+)+)\s',
lambda mobj: ''.join((delim, mobj.group('label') or '', re.sub(r'\s+', '', mobj.group('url')), '\n'))
),
( # Do not split "words"
rf'(?m)({delim}\S+)+$',
lambda mobj: ''.join((delim, mobj.group(0).replace(delim, '')))
),
# ( # Allow overshooting last line
# rf'(?m)^(?P<prev>.+)${delim}(?P<current>.+)$(?!{delim})',
# lambda mobj: (mobj.group().replace(delim, ' ')
# if len(mobj.group()) - len(delim) + 1 <= max_width + ALLOWED_OVERSHOOT
# else mobj.group())
# ),
# ( # Avoid newline when a space is available b/w switch and description
# DISABLE_PATCH, # This creates issues with prepare_manpage
# r'(?m)^(\s{4}-.{%d})(%s)' % (switch_col_width - 6, delim),
# r'\1 '
# ),
# ( # Replace brackets with a Markdown link
# r'SponsorBlock API \((http.+)\)',
# r'[SponsorBlock API](\1)'
# ),
)
for file, options_start, options_end in READMES:
readme = read_file(file)
write_file(file, ''.join((
take_section(readme, end=options_start, shift=len(options_start)),
'\n\n```text',
functools.reduce(apply_patch, PATCHES, options),
'```\n\n',
take_section(readme, options_end),
)))
|