tf.py

import argparse
from datetime import datetime
import json
import os

from api.domain.usecase import build_tf

TF_STORAGE_DIR = os.path.dirname(os.path.realpath(__file__)) + '/storage/tf'


def main(tf_data: str, tf_name: str = None) -> None:
    if tf_name is None:
        tf_name = datetime.now().strftime("%Y-%m-%d-%H-%M-%S")

    tf_path = f'{TF_STORAGE_DIR}/{tf_name}'
    os.mkdir(tf_path)

    build_tf.execute(
        tf_data=json.loads(tf_data),
        tf_path=tf_path,
    )

    return print(f'Create build in {tf_path}')


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description='Test builder'
    )
    parser.add_argument(
        'tf_data',
        type=str,
        help='The tf schema in json',
    )
    parser.add_argument(
        '--tf_name',
        type=str,
        help='Build folder name',
    )

    data = vars(parser.parse_args())

    main(**data)