backend/module/utils/image_translator/devscripts/make_readme.py

#!/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),
    )))