From f762b83d2498535b01375a8393c13dbe79714484 Mon Sep 17 00:00:00 2001
From: Markus Heikkinen <markus.heikkinen@zhinst.com>
Date: Wed, 30 Aug 2023 08:10:51 +0200
Subject: [PATCH] Add Session node setting method

---
 src/labone/core/session.py | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/src/labone/core/session.py b/src/labone/core/session.py
index 3bbc64e..4e31837 100644
--- a/src/labone/core/session.py
+++ b/src/labone/core/session.py
@@ -8,7 +8,8 @@
 import capnp
 from typing_extensions import Literal, NotRequired, TypeAlias, TypedDict
 
-from labone.core import errors
+from labone.core import errors, result
+from labone.core import value as annotated_value
 from labone.core.connection_layer import (
     KernelInfo,
     ServerInfo,
@@ -405,3 +406,25 @@ async def list_nodes_info(
             raise TypeError(msg) from error
         response = await _send_and_wait_request(request)
         return json.loads(response.nodeProps)
+
+    async def set_value(
+        self,
+        value: annotated_value.AnnotatedValue,
+    ) -> annotated_value.AnnotatedValue:
+        """Set the value of a node.
+
+        TODO: Tests
+
+        Args:
+            value: Annotated value of the node.
+                TODO: To accept list of `AnnotatedValue`s
+
+        Raises:
+            LabOneConnectionError: If there is a problem in the connection.
+        """
+        request = self._session.setValue_request()
+        request.path = value.path
+        request.value = value.value
+        response = await _send_and_wait_request(request)
+        res = result.unwrap(response.result)
+        return annotated_value.AnnotatedValue.from_capnp(res)