splityaml.py

"""
usage:
splityaml.py openapi.yaml en openapi-en.json
splityaml.py openapi.yaml fr openapi-fr.json
"""

LANGUAGES = {'en', 'fr'}
POST_GET = 'post/get'

import json
import yaml
import sys

def split(d, lng):
    if isinstance(d, dict):
        if d.keys() == LANGUAGES:
            return d[lng]
        if d.keys() == {POST_GET}:
            return {
                'get': split(convert_post_to_get_params(d[POST_GET]), lng),
                'post': split(d[POST_GET], lng),
            }
    if isinstance(d, dict):
        return {k: split(v, lng) for (k, v) in d.items()}
    if isinstance(d, list):
        return [split(i, lng) for i in d]
    return d

def convert_post_to_get_params(post):
    props = post['requestBody']['content']['application/json']['schema']['properties']
    params = []
    for k, v in props.items():
        if 'oneOf' in v:
            # XXX: assume first version is the string version of the parameter
            v = v['oneOf'][0]
        p = {
            'name': k,
            'in': 'query',
            'description': v['description'],
            'schema': {
                'type': v['type'],
            },
        }
        if 'examples' in v:
            # need to pop so that post schema validated
            p['examples'] = v.pop('examples')
        if 'default' in v:
            p['schema']['default'] = v.pop('default')
        if 'enum' in v:
            p['schema']['enum'] = v['enum']
        if 'format' in v:
            p['schema']['format'] = v['format']
        params.append(p)

    return {
        'summary': post['summary'],
        'description': post['description'],
        'parameters': params,
        'responses': post['responses'],
        'tags': post['tags'],
    }

with open(sys.argv[1], encoding='utf-8') as r:
    doc = yaml.safe_load(r)

lang = sys.argv[2]
out = split(doc, lang)

with open(sys.argv[3], 'w', encoding='utf-8') as w:
    json.dump(out, w)