def __init__(self, parser: ArgumentParser):
CliPlugin.__init__(self, parser)
parser.add_argument(
"-n",
"--pipeline_name",
type=str,
choices=preset_pipelines.keys(),
default="default",
help="Preset pipeline to run",
)
parser.add_argument(
"-d",
"--display",
type=str2bool,
default="yes",
help="Indicates if should display images.",
)
parser.add_argument(
"-s",
"--save",
type=str2bool,
default="yes",
help="Indicates if should save results.",
)
def __init__(self, parser: ArgumentParser):
CliPlugin.__init__(self, parser)
parser.add_argument(
"-n",
"--pipeline_name",
type=str,
choices=preset_pipelines.keys(),
default="default",
help="Preset pipeline to run",
)
def __init__(self, parser: ArgumentParser):
CliPlugin.__init__(self, parser)
subparsers = parser.add_subparsers(dest="schematic")
subparsers.required = True
# We import the schematics list from a json file instead of yaml,
# Python's yaml support is not built in
with open("./src/cli_plugins/schematics/schematics.json",
"r",
encoding="utf8") as f:
schematics_dict = json.load(f)
for schematic_dict in schematics_dict["schematics"]:
subparser = subparsers.add_parser(
schematic_dict["name"],
description=schematic_dict["description"])
module = importlib.import_module("." + schematic_dict["module"],
"src.cli_plugins.schematics")
class_type = getattr(module, schematic_dict["class"])
schematic: Schematic = class_type(subparser)
subparser.set_defaults(schematic=schematic.execute)
def __init__(self, parser: ArgumentParser):
CliPlugin.__init__(self, parser)
