codegen.py

#!/usr/bin/env python3
import sys
import yaml
import os
import re
import requests

class NoAliasDumper(yaml.Dumper):
    def ignore_aliases(self, data):
        return True

def remove_oneof(data):
    """ Removes oneOf key from a dict and recursively calls this
        function on other dict values.
    """
    if 'oneOf' in data:
        del data['oneOf']
    for key in data:
        if isinstance(data[key], dict):
            remove_oneof(data[key])
    return data

def remove_non_json_apis(data):
    """ The R openapi generator does not handle non-json responses
    """
    blacklist = [
        "/rest/GetFrame/{id}",
        "/rest/StateGraphic/{id}"
    ]
    paths = data['paths']
    for key in blacklist:
        del paths[key]
    return data

def replace_description_newlines(data):
    """ The way R docs and mustache templates interact
        does not allow for newlines, it has to be html break tags
    """
    if 'description' in data:
        if isinstance(data['description'], str):
            data['description'] = re.sub("\\n|\n", "\n#' ", data['description'])
    for key in data:
        if isinstance(data[key], dict):
            replace_description_newlines(data[key])
    return data

filepath = sys.argv[1]
if not os.path.exists(filepath):
        response = requests.get("https://cloud.tator.io/schema")
        assert response.status_code == 200
        with open(filepath, 'wb') as f:
            f.write(response.content)
with open(filepath, 'r') as f:
    schema = yaml.load(f, Loader=yaml.FullLoader)
    schema = remove_oneof(schema)
    schema = remove_non_json_apis(schema)
    schema = replace_description_newlines(schema)
with open(filepath, 'w') as f:
    yaml.dump(schema, f, Dumper=NoAliasDumper)