wmill 1.295.4__tar.gz → 1.296.1__tar.gz

{wmill-1.295.4 → wmill-1.296.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wmill
-Version: 1.295.4
+Version: 1.296.1
 Summary: A client library for accessing Windmill server wrapping the Windmill client API
 Home-page: https://windmill.dev
 License: Apache-2.0

{wmill-1.295.4 → wmill-1.296.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "wmill"
-version = "1.295.4"
+version = "1.296.1"
 description = "A client library for accessing Windmill server wrapping the Windmill client API"
 license = "Apache-2.0"
 homepage = "https://windmill.dev"

{wmill-1.295.4 → wmill-1.296.1}/wmill/client.py

@@ -346,6 +346,24 @@ class Windmill:
     def set_state(self, value: Any):
         self.set_resource(value, path=self.state_path, resource_type="state")
 
+    def set_flow_user_state(self, key: str, value: Any) -> None:
+        """Set the user state of a flow at a given key"""
+        flow_id = self.get_root_job_id()
+        r = client.post(f"/w/{self.workspace}/jobs/flow/user_states/{flow_id}/{key}", json=value,  raise_for_status=False)
+        if r.status_code == 404:
+            print(f"Job {flow_id} does not exist or is not a flow")
+
+
+    def get_flow_user_state(self, key: str) -> Any:
+        """Get the user state of a flow at a given key"""
+        flow_id = self.get_root_job_id()
+        r = client.get(f"/w/{self.workspace}/jobs/flow/user_states/{flow_id}/{key}", raise_for_status=False)
+        if r.status_code == 404:
+            print(f"Job {flow_id} does not exist or is not a flow")
+            return None
+        else:
+            return r.json()
+
     @property
     def version(self):
         return self.get("version").text
@@ -851,6 +869,21 @@ def set_variable(path: str, value: str, is_secret: bool = False) -> None:
     return _client.set_variable(path, value, is_secret)
 
 
+@init_global_client
+def get_flow_user_state(key: str) -> Any:
+    """
+    Get the user state of a flow at a given key
+    """
+    return _client.get_flow_user_state(key)
+
+@init_global_client
+def set_flow_user_state(key: str, value: Any) -> None:
+    """
+    Set the user state of a flow at a given key
+    """
+    return _client.set_flow_user_state(key, value)
+
+
 @init_global_client
 def get_state_path() -> str:
     return _client.state_path