From a994b822a4231061153c6b27d8c5d7307b59f8b0 Mon Sep 17 00:00:00 2001
From: Botond Hende <nettingman@gmail.com>
Date: Sun, 1 Sep 2024 23:33:15 +0200
Subject: ship task setting from request body

---
 modules/task_types.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 modules/task_types.py

(limited to 'modules/task_types.py')

diff --git a/modules/task_types.py b/modules/task_types.py
new file mode 100644
index 0000000..d23ce55
--- /dev/null
+++ b/modules/task_types.py
@@ -0,0 +1,21 @@
+from typing import Annotated
+from pydantic.functional_validators import AfterValidator
+
+IDLE = 'IDLE'
+MINING = 'MINING'
+
+MIA = 'MIA'
+ERROR = 'ERROR'
+
+task_types = [
+    IDLE,
+    MINING,
+]
+
+
+def is_task_type(task: str):
+    assert task in task_types, f"'{task}' is not a valid task type."
+    return task
+
+
+task_type = Annotated[str, AfterValidator(is_task_type)]
-- 
cgit v1.2.3-70-g09d2