wmill 1.600.0__py3-none-any.whl → 1.609.0__py3-none-any.whl

wmill/client.py

@@ -707,13 +707,25 @@ class Windmill:
             params=params if params else None,
         ).json()
     
-    def set_state(self, value: Any):
+    def set_state(self, value: Any, path: str | None = None) -> None:
         """Set the workflow state.
 
         Args:
             value: State value to set
+            path: Optional state resource path override.
         """
-        self.set_resource(value, path=self.state_path, resource_type="state")
+        self.set_resource(value, path=path or self.state_path, resource_type="state")
+
+    def get_state(self, path: str | None = None) -> Any:
+        """Get the workflow state.
+
+        Args:
+            path: Optional state resource path override.
+
+        Returns:
+            State value or None if not set
+        """
+        return self.get_resource(path=path or self.state_path, none_if_undefined=True)
 
     def set_progress(self, value: int, job_id: Optional[str] = None):
         """Set job progress percentage (0-99).
@@ -963,7 +975,7 @@ class Windmill:
             ).json()
         except Exception as e:
             raise Exception("Could not write file to S3") from e
-        return S3Object(s3=response["file_key"])
+        return S3Object(s3=response["file_key"], storage=s3object["storage"])
 
     def sign_s3_objects(self, s3_objects: list[S3Object | str]) -> list[S3Object]:
         """Sign S3 objects for use by anonymous users in public apps.
@@ -1167,20 +1179,26 @@ class Windmill:
         with open(f"/shared/{path}", "r", encoding="utf-8") as f:
             return json.load(f)
 
-    def get_resume_urls(self, approver: str = None) -> dict:
+    def get_resume_urls(self, approver: str = None, flow_level: bool = None) -> dict:
         """Get URLs needed for resuming a flow after suspension.
 
         Args:
             approver: Optional approver name
+            flow_level: If True, generate resume URLs for the parent flow instead of the
+                specific step. This allows pre-approvals that can be consumed by any later
+                suspend step in the same flow.
 
         Returns:
             Dictionary with approvalPage, resume, and cancel URLs
         """
         nonce = random.randint(0, 1000000000)
         job_id = os.environ.get("WM_JOB_ID") or "NO_ID"
+        params = {"approver": approver}
+        if flow_level is not None:
+            params["flow_level"] = flow_level
         return self.get(
             f"/w/{self.workspace}/jobs/resume_urls/{job_id}/{nonce}",
-            params={"approver": approver},
+            params=params,
         ).json()
 
     def request_interactive_slack_approval(
@@ -1725,12 +1743,11 @@ def whoami() -> dict:
 
 
 @init_global_client
-@deprecate("Windmill().state")
-def get_state() -> Any:
+def get_state(path: str | None = None) -> Any:
     """
     Get the state
     """
-    return _client.state
+    return _client.get_state(path=path)
 
 
 @init_global_client
@@ -1781,11 +1798,11 @@ def list_resources(
 
 
 @init_global_client
-def set_state(value: Any) -> None:
+def set_state(value: Any, path: str | None = None) -> None:
     """
     Set the state
     """
-    return _client.set_state(value)
+    return _client.set_state(value, path=path)
 
 
 @init_global_client
@@ -1876,16 +1893,19 @@ def get_state_path() -> str:
 
 
 @init_global_client
-def get_resume_urls(approver: str = None) -> dict:
+def get_resume_urls(approver: str = None, flow_level: bool = None) -> dict:
     """Get URLs needed for resuming a flow after suspension.
 
     Args:
         approver: Optional approver name
+        flow_level: If True, generate resume URLs for the parent flow instead of the
+            specific step. This allows pre-approvals that can be consumed by any later
+            suspend step in the same flow.
 
     Returns:
         Dictionary with approvalPage, resume, and cancel URLs
     """
-    return _client.get_resume_urls(approver)
+    return _client.get_resume_urls(approver, flow_level)
 
 
 @init_global_client

{wmill-1.600.0.dist-info → wmill-1.609.0.dist-info}/METADATA

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

{wmill-1.600.0.dist-info → wmill-1.609.0.dist-info}/RECORD

@@ -1,8 +1,8 @@
 wmill/__init__.py,sha256=nGZnQPezTdrBnBW1D0JqUtm75Gdf_xi3tAcPGwHRZ5A,46
-wmill/client.py,sha256=QhJbL2osMHO-v_ngjvB1_r8gtasS43_CwzOTsH6e5ek,72774
+wmill/client.py,sha256=CM6DOs7LQFWDnKFKeViiAUX708vWdxRfvfT0sgcMJOQ,73878
 wmill/py.typed,sha256=8PjyZ1aVoQpRVvt71muvuq5qE-jTFZkK-GLHkhdebmc,26
 wmill/s3_reader.py,sha256=ySd1R2F9krbyhFU_-ogGDZAWQsF2CkPgq7K7xCEvhDU,2662
 wmill/s3_types.py,sha256=gQZRdQoDRp7IRS9MTgGAiHZr1MIdtrbWeN-NJ28kIkA,1804
-wmill-1.600.0.dist-info/METADATA,sha256=6PP6MZYMlj025r-xzKWR9PS4wne60N_cHxUkUyJDMIk,2693
-wmill-1.600.0.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
-wmill-1.600.0.dist-info/RECORD,,
+wmill-1.609.0.dist-info/METADATA,sha256=Ir13sM7IFpTWzWNOkBijzBreRZxYmvU6NHsL6BSnZKo,2693
+wmill-1.609.0.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
+wmill-1.609.0.dist-info/RECORD,,