gen.py

#!/usr/bin/env python3

import yaml
import json
import sys
import shutil

if len(sys.argv) != 2:
    print('expecting ce or ee as arg1')
    exit(1)

if sys.argv[1] != r'ce' and sys.argv[1] != r'ee':
    print('expecting ce or ee as arg1')
    exit(2)

## check if the 'lang' field matches expected input
## when no 'lang' is defined, it matches both 'en' and 'cn'
def is_lang_match(i, en_or_cn):
    if 'lang' in i:
        return i['lang'] == en_or_cn
    else:
        return True

EDITION = sys.argv[1]

# parser document env file
def parse_env_file(file_path):
    config = {}
    with open(file_path, 'r') as file:
        for line in file:
            line = line.strip()
            if line:
                key, value = line.split('=')
                config[key] = value
    return config

## check if the 'edition' field matches expected input
## when no 'edition' is defined, it matches both 'ce' and 'ee'
def is_edition_match(i, ce_or_ee):
    if 'edition' in i:
        return i['edition'] == ce_or_ee
    else:
        return True

def read_title_from_md(lang, path):
    if lang == 'en':
        dir = 'en_US'
    else:
        dir = 'zh_CN'
    path = dir + '/' + path + '.md'
    with open(path) as f:
        for line in f:
            if line.strip():
                return line.strip('\n').strip('#').strip()

def parse(children, lang, edition):
    acc=[]
    for i in range(len(children)):
        child = children[i]
        if 'path' in child:
            child['path'] = child['path'].replace('${edition}', edition)
        if isinstance(child, str):
            child = child.replace('${edition}', edition)
        if not is_lang_match(child, lang):
            continue
        if not is_edition_match(child, edition):
            continue

        if 'title_en' in child:
            title = child['title_en']
            if lang == 'cn' and 'title_cn' in child:
                title = child['title_cn']
        else:
            title = read_title_from_md(lang, child)
        _child = {'title': title}

        if isinstance(child, str):
            _child['path'] = child
        else:
            if 'path' in child:
                _child['path'] = child['path']

            if 'collapsed' in child:
                _child['collapsed'] = child['collapsed']

            if 'children' in child:
                godeep = parse(child['children'], lang, edition)
                _child['children'] = godeep

        acc.append(_child)
    return acc

def move_manual(lang, edition):
    if lang == 'cn':
        lang = 'zh'
    baseDir = 'en_US' if lang == 'en' else 'zh_CN'
    source_path = f'cfg-manual-docgen/configuration-manual-{edition}-{lang}.md'
    target_path = f'{baseDir}/configuration/configuration-manual.md'
    shutil.copyfile(source_path, target_path)

with open(r'dir.yaml', encoding='utf-8') as file:
    # Read file and replace the str with env variable
    content = file.read()
    version = parse_env_file(r'current-version.env')
    for key in version:
        content = content.replace('${' + key + '}', version[key])

    # The FullLoader parameter handles the conversion from YAML
    # scalar values to Python the dictionary format
    all = yaml.load(content, Loader=yaml.FullLoader)

    move_manual('en', EDITION)
    move_manual('cn', EDITION)
    en = parse(all, 'en', EDITION)
    cn = parse(all, 'cn', EDITION)
    res ={'en': en, 'cn': cn}
    json.dump(res, sys.stdout, indent=2, ensure_ascii=False)