openMINDS.py

import argparse
import os
import shutil
import sys

from generator.commons import TARGET_PATH, OPENMINDS_VOCAB
from generator.expander import Expander
from generator.generate_html import HTMLGenerator
from generator.generate_json_schema import JsonSchemaGenerator
from generator.generate_plantuml import PlantUMLGenerator
from generator.instance_locator import InstanceLocator
from generator.vocab_extractor import VocabExtractor

parser = argparse.ArgumentParser(prog=sys.argv[0], description='Generate various sources out of the EBRAINS openMINDS schema templates')
parser.add_argument('--reinit', dest='reinit', action='store_true')
parser.add_argument('--current', help="The current element")
parser.add_argument('--allTags', help="A comma separated list of all available tags")
parser.add_argument('--allVersionBranches', help="A comma separated list of all available version branches")
parser.set_defaults(reinit=False)
parser.add_argument('--path', help="The path to the source", default=".")
parser.add_argument('--ignore', help="Names of schema groups to ignore", default=[], action='append')
parser.add_argument('--vocab', help="Names of schema groups to ignore", default=OPENMINDS_VOCAB)
args = vars(parser.parse_args())


def main():
    print("***************************************")
    print(f"Triggering the generation of sources for openMINDS ...")
    print("***************************************")
    print()
    print("Expanding the schemas...")
    expander = Expander(args["path"], args["vocab"], ignore=args["ignore"])
    expander.expand()
    print("Extracting the vocab...")
    types_file, properties_file = VocabExtractor(expander.schemas, args["path"], args["reinit"], args["current"], args["vocab"]).extract()
    expander.enrich_with_vocab(types_file, properties_file)
    instances = InstanceLocator(args["path"]).find_instances()
    print("Clear target directory")
    if os.path.exists(TARGET_PATH):
        shutil.rmtree(TARGET_PATH)

    print(f"Generating JSON schemas for...")
    JsonSchemaGenerator(expander.schemas, args["vocab"]).generate(ignore=args["ignore"])
    print("Generating HTML documentation...")
    HTMLGenerator(expander.schemas, instances, current=args["current"], all_tags=[x for x in args["allTags"].split(",") if x], all_version_branches=[x for x in args["allVersionBranches"].split(",") if x]).generate(ignore=args["ignore"])
    print("Generating UML documentation...")
    PlantUMLGenerator(expander.schemas).generate(ignore=args["ignore"])



if __name__ == "__main__":
    main()