namuan/api-mocker-generator

View on GitHub
api_mocker_generator/api_mocker_generator.py

Summary

Maintainability
A
0 mins
Test Coverage
import argparse
import os

from .logger import *
from .orchestra import play

__version__ = '0.0.6'


def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--swagger",
        required=True,
        help="Location of swagger document. Could be HTTP endpoint or local file system",
    )
    parser.add_argument(
        "--output",
        required=True,
        type=str,
        help='Output directory for configuration and mocked responses'
    )
    parser.add_argument("--verbose", action="store_true", default=False)
    return parser.parse_args()


def main():
    args = parse_args()

    if args.verbose:
        set_verbose_logging()
    else:
        logging.basicConfig(level=logging.INFO)

    os.makedirs(args.output, exist_ok=True)

    info("Api-Mocker-Generator version {}".format(__version__))
    play(args.swagger, args.output)