only available gpio modes

1 files changed, 13 insertions, 0 deletions

diff --git a/kvmd/plugins/ugpio/__init__.py b/kvmd/plugins/ugpio/__init__.py
index 1e7725e8..9ed48a5f 100644
--- a/kvmd/plugins/ugpio/__init__.py
+++ b/kvmd/plugins/ugpio/__init__.py
@@ -20,6 +20,7 @@
 # ========================================================================== #
 
 
+from typing import Set
 from typing import Type
 from typing import Optional
 from typing import Any
@@ -47,6 +48,14 @@ class GpioDriverOfflineError(GpioOperationError):
 
 
 # =====
+class UserGpioModes:
+    INPUT = "input"
+    OUTPUT = "output"
+
+    ALL = set([INPUT, OUTPUT])
+
+
+# =====
 class BaseUserGpioDriver(BasePlugin):
     def __init__(  # pylint: disable=super-init-not-called
         self,
@@ -61,6 +70,10 @@ class BaseUserGpioDriver(BasePlugin):
     def get_instance_id(self) -> str:
         return self._instance_name
 
+    @classmethod
+    def get_modes(cls) -> Set[str]:
+        return set(UserGpioModes.ALL)
+
     def register_input(self, pin: int) -> None:
         raise NotImplementedError