Basic "CLI" (#546)

Summary: Pull Request resolved: #546

Differential Revision: D49758493

fbshipit-source-id: 549797dd0b64f88f6fcb7640a7dc262e1d3dc51e

Author: Olivia-liu

sdk/etdb/TARGETS

@@ -1,3 +1,4 @@
+load("@fbcode_macros//build_defs:python_binary.bzl", "python_binary")
 load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
 
 oncall("executorch")
@@ -44,6 +45,12 @@ python_library(
     ],
 )
 
+python_binary(
+    name = "inspector_cli",
+    main_module = "executorch.sdk.etdb.inspector",
+    deps = [":inspector"],
+)
+
 python_library(
     name = "inspector_utils",
     srcs = [

sdk/etdb/inspector.py

@@ -4,6 +4,7 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
+import argparse
 import dataclasses
 import logging
 from collections import defaultdict, OrderedDict
@@ -489,3 +490,24 @@ def get_exported_program(self, graph: Optional[str] = None) -> ExportedProgram:
         if graph is None:
             return self._etrecord.edge_dialect_program
         return self._etrecord.graph_map.get(graph)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "--etdump_path",
+        required=True,
+        help="Provide an ETDump file path.",
+    )
+    parser.add_argument(
+        "--etrecord_path",
+        required=False,
+        help="Provide an optional ETRecord file path.",
+    )
+
+    args = parser.parse_args()
+
+    inspector = Inspector(
+        etdump_path=args.etdump_path, etrecord_path=args.etrecord_path
+    )
+    inspector.print_data_tabular()