def _load_suite(context, suite_name):
if suite_name.endswith(".json"):
suite_name = suite_name[:-5]
try:
suite = context.get_expectation_suite(suite_name)
return suite
except ge_exceptions.DataContextError as e:
cli_message(
"<red>Could not find a suite named `{}`. Please check the name and try again.</red>"
.format(suite_name))
logger.info(e)
sys.exit(1)
def _load_suite(context, suite_name):
if suite_name.endswith(".json"):
suite_name = suite_name[:-5]
try:
suite = context.get_expectation_suite(suite_name)
return suite
except ge_exceptions.DataContextError as e:
cli_message(
f"<red>Could not find a suite named `{suite_name}`.</red> Please check "
"the name by running `great_expectations suite list` and try again."
)
logger.info(e)
sys.exit(1)
def load_expectation_suite(context: DataContext,
suite_name: str) -> ExpectationSuite:
"""
Load an expectation suite from a given context.
Handles a suite name with or without `.json`
"""
if suite_name.endswith(".json"):
suite_name = suite_name[:-5]
try:
suite = context.get_expectation_suite(suite_name)
return suite
except ge_exceptions.DataContextError as e:
cli_message(
f"<red>Could not find a suite named `{suite_name}`.</red> Please check "
"the name by running `great_expectations suite list` and try again."
)
logger.info(e)
# TODO this should try to send a usage statistic failure
sys.exit(1)