From 0efca3cadbad5517a59884576ddb90cfe7ac30f8 Mon Sep 17 00:00:00 2001
From: Diego Russo <diego.russo@arm.com>
Date: Mon, 30 May 2022 13:34:14 +0100
Subject: Add MLIA codebase

Add MLIA codebase including sources and tests.

Change-Id: Id41707559bd721edd114793618d12ccd188d8dbd
---
 src/mlia/cli/common.py | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 src/mlia/cli/common.py

(limited to 'src/mlia/cli/common.py')

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
-- 
cgit v1.2.1