Justin Ibarra
23 lines
687 B
Python
23 lines
687 B
Python
# Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
|
|
# or more contributor license agreements. Licensed under the Elastic License
|
|
# 2.0; you may not use this file except in compliance with the Elastic License
|
|
# 2.0.
|
|
|
|
import argparse
|
|
import importlib
|
|
import os
|
|
|
|
from . import get_ttp_names
|
|
|
|
parser = argparse.ArgumentParser("rta")
|
|
parser.add_argument("ttp_name")
|
|
|
|
parsed_args, remaining = parser.parse_known_args()
|
|
ttp_name, _ = os.path.splitext(os.path.basename(parsed_args.ttp_name))
|
|
|
|
if ttp_name not in get_ttp_names():
|
|
raise ValueError("Unknown RTA {}".format(ttp_name))
|
|
|
|
module = importlib.import_module("rta." + ttp_name)
|
|
exit(module.main(*remaining))
|