Skip to content

Commit 497ea07

Browse files
committed
feat: add option to save exceptions as JSON
Signed-off-by: ktro2828 <kotaro.uetake@tier4.jp>
1 parent f2e028a commit 497ea07

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

t4_devkit/cli/sanity.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,9 @@
66
from tabulate import tabulate
77
from tqdm import tqdm
88

9+
from t4_devkit.common.io import save_json
910
from t4_devkit.common.sanity import DBException, sanity_check
11+
from t4_devkit.common.serialize import serialize_dataclasses
1012

1113
from .version import version_callback
1214

@@ -41,6 +43,7 @@ def main(
4143
is_eager=True,
4244
),
4345
db_parent: str = typer.Argument(..., help="Path to parent directory of the databases."),
46+
output: str | None = typer.Option(None, "-o", "--output", help="Path to output JSON file."),
4447
revision: str | None = typer.Option(
4548
None, "-rv", "--revision", help="Specify if you want to check the specific version."
4649
),
@@ -57,3 +60,7 @@ def main(
5760
headers = ["DatasetID", "Version", "status", "Message"]
5861
table = [[e.dataset_id, e.version, e.status, e.message] for e in exceptions]
5962
print(tabulate(table, headers=headers, tablefmt="pretty"))
63+
64+
if output:
65+
serialized = serialize_dataclasses(exceptions)
66+
save_json(serialized, output)

0 commit comments

Comments
 (0)