main.py

import json
import logging
import sys

from fastapi import FastAPI

import api.InfrastructureTemplateController
from api.InfrastructureTemplateController import compress_file
from controller.PluginOrchestrator import create_infrastructure_files
from icgparser import ModelParser

fast_api = FastAPI()

fast_api.include_router(api.InfrastructureTemplateController.api_router)
logging.getLogger().setLevel(logging.INFO)

#-------------------------------------------------------------------------
# Parse parameters
#-------------------------------------------------------------------------
skip_next = False
doml_directory = "./doml"
model_filename = "./nginx-openstack_v2_multiecores.domlx"
load_split_model = True
output_file_name = "iac_files.tar.gz"

# get metamodel directory from command line
def param_dir(pos, list):
    global doml_directory
    global skip_next
    doml_directory = list[pos+1]
    print(f"    doml_directory = {doml_directory} model_filename = {model_filename}")
    skip_next = True

def param_help(pos, list):
    print(f"\nUsage: {sys.argv[0]} [-h] [-d <doml_directory>] [--single] <model_filename>\n")
    sys.exit()

# indicate to load the single-file metamodel (doml.ecore) instead of the split one
def param_single(pos, list):
    global load_split_model
    print(f"--> param_single({pos},{list}")
    load_split_model = False

options = {'-d': param_dir, '-h': param_help, '--single_mmodel': param_single, '--single': param_single, '--output': output_file_name}
argc = len(sys.argv)
paramlist = sys.argv[1:]
for i, param in enumerate(paramlist):
    print(f"i={i} param={param} skip_next={skip_next}")
    if param in options:
        options[param](i, paramlist)
        continue
    if skip_next:
        skip_next = False
        continue
    else:
        model_filename = param

if __name__ == '__main__':
    ModelParser.parse_model(model_filename, load_split_model, doml_directory)
    with open("input_file_generated/ir.json") as json_file:
        data = json.load(json_file)
        template_generated_folder = create_infrastructure_files(data)
        compress_file_folder = compress_file(template_generated_folder, output_file_name)