import sys
import logging
import argparse
from magicalimport import import_symbol
logger = logging.getLogger(__name__)
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--driver", default="Driver")
parser.add_argument(
"--logging",
default="INFO",
choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
)
parser.add_argument("--full", action="store_true")
parser.add_argument("--legacy", action="store_true")
parser.add_argument("file", default=None)
args = parser.parse_args()
logging.basicConfig(
format="%(levelname)5s:%(name)36s:%(message)s",
level=logging._nameToLevel[args.logging],
)
driver_cls = args.driver
if args.legacy and driver_cls == "Driver": # xxx
logger.info("legacy option is added. output Schema is legacy flavor")
driver_cls = "LegacyDriver"
if ":" not in driver_cls:
driver_cls = "swagger_marshmallow_codegen.driver:{}".format(driver_cls)
if args.full:
options = {"targets": {"schema": True, "input": True, "output": True}}
else:
options = {"targets": {"schema": True}}
driver = import_symbol(driver_cls, cwd=True)(options)
if args.file is None:
driver.run(sys.stdin, sys.stdout)
else:
with open(args.file) as rf:
driver.run(rf, sys.stdout)
