BTHLABS-49: WIP

services/packages/workspace_tools/hotpocket_workspace_tools/__init__.py

@@ -0,0 +1,2 @@
+from .constants import WorkspaceMode  # noqa: F401
+from .workspace_mode import get_workspace_mode  # noqa: F401

services/packages/workspace_tools/hotpocket_workspace_tools/constants.py

@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+from __future__ import annotations
+
+import enum
+
+DEFAULT_WORKSPACE_MODE = 'docker'
+
+
+class WorkspaceMode(enum.Enum):
+    DOCKER = 'docker'
+    METAL = 'metal'

services/packages/workspace_tools/hotpocket_workspace_tools/workspace_mode.py

@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+from __future__ import annotations
+
+import os
+
+from .constants import DEFAULT_WORKSPACE_MODE, WorkspaceMode
+
+
+def get_workspace_mode(default: str | None = None) -> WorkspaceMode:
+    return WorkspaceMode(os.getenv(
+        'HOTPOCKET_WORKSPACE_MODE',
+        (
+            default
+            if default is not None
+            else DEFAULT_WORKSPACE_MODE
+        ),
+    ))