1 files changed, 38 insertions, 0 deletions
diff --git a/src/mlia/cli/common.py b/src/mlia/cli/common.py
new file mode 100644
index 0000000..54bd457
--- /dev/null
+++ b/src/mlia/cli/common.py
@@ -0,0 +1,38 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""CLI common module."""
+import argparse
+from dataclasses import dataclass
+from typing import Callable
+from typing import List
+
+
+@dataclass
+class CommandInfo:
+    """Command description."""
+
+    func: Callable
+    aliases: List[str]
+    opt_groups: List[Callable[[argparse.ArgumentParser], None]]
+    is_default: bool = False
+
+    @property
+    def command_name(self) -> str:
+        """Return command name."""
+        return self.func.__name__
+
+    @property
+    def command_name_and_aliases(self) -> List[str]:
+        """Return list of command name and aliases."""
+        return [self.command_name, *self.aliases]
+
+    @property
+    def command_help(self) -> str:
+        """Return help message for the command."""
+        assert self.func.__doc__, "Command function does not have a docstring"
+        func_help = self.func.__doc__.splitlines()[0].rstrip(".")
+
+        if self.is_default:
+            func_help = f"{func_help} [default]"
+
+        return func_help