wmill 1.394.6__py3-none-any.whl → 1.396.0__py3-none-any.whl

wmill/client.py

@@ -346,6 +346,36 @@ class Windmill:
     def set_state(self, value: Any):
         self.set_resource(value, path=self.state_path, resource_type="state")
 
+    def set_progress(self, value: int, job_id: Optional[str] = None):
+        workspace = get_workspace()
+        flow_id = os.environ.get("WM_FLOW_JOB_ID")
+        job_id = job_id or os.environ.get("WM_JOB_ID")
+
+        if job_id != None:
+            job = self.get_job(job_id)
+            flow_id = job.get("parent_job")
+
+        self.post(
+            f"/w/{workspace}/job_metrics/set_progress/{job_id}",
+            json={
+                "percent": value,
+                "flow_job_id": flow_id or None,
+            },
+        )        
+
+    def get_progress(self, job_id: Optional[str] = None ) -> Any:
+        workspace = get_workspace()
+        job_id = job_id or os.environ.get("WM_JOB_ID")
+
+        r = self.get(
+            f"/w/{workspace}/job_metrics/get_progress/{job_id}",
+        )        
+        if r.status_code == 404:
+            print(f"Job {job_id} does not exist")
+            return None
+        else:
+            return r.json()
+
     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()
@@ -838,6 +868,20 @@ def set_state(value: Any) -> None:
     """
     return _client.set_state(value)
 
+@init_global_client
+def set_progress(value: int, job_id: Optional[str] = None) -> None:
+    """
+    Set the progress
+    """
+    return _client.set_progress(value, job_id)
+
+@init_global_client
+def get_progress(job_id: Optional[str] = None) -> Any:
+    """
+    Get the progress
+    """
+    return _client.get_progress(job_id)
+
 
 def set_shared_state_pickle(value: Any, path="state.pickle") -> None:
     """

{wmill-1.394.6.dist-info → wmill-1.396.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wmill
-Version: 1.394.6
+Version: 1.396.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.394.6.dist-info → wmill-1.396.0.dist-info}/RECORD

@@ -1,8 +1,8 @@
 wmill/__init__.py,sha256=nGZnQPezTdrBnBW1D0JqUtm75Gdf_xi3tAcPGwHRZ5A,46
-wmill/client.py,sha256=wHMpt-K1a1I-9-Hhby9PDL-XmkyGu4qoX2g_WM5s8NY,33045
+wmill/client.py,sha256=PCt3IXmAEkOldz08LtkDw1S4r5GHfEtZDibKMAZp9XM,34327
 wmill/py.typed,sha256=8PjyZ1aVoQpRVvt71muvuq5qE-jTFZkK-GLHkhdebmc,26
 wmill/s3_reader.py,sha256=izHlg2Xsg0Sr_LkDDEC35VuEijJcuPBDIm-xj21KsgU,1668
 wmill/s3_types.py,sha256=XVjPyc5sjWesTy6nnRReojhyPgPsaT0EY3X57mWUaJ4,1173
-wmill-1.394.6.dist-info/METADATA,sha256=VQOHmz2S2mMdY9w9UZWMBEZV2_ujOGlaFh5a8JBRJys,2693
-wmill-1.394.6.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
-wmill-1.394.6.dist-info/RECORD,,
+wmill-1.396.0.dist-info/METADATA,sha256=bSEtdlJ-DXttZ6JoRB2K1PYliSD95jh-PamGAax3MJo,2693
+wmill-1.396.0.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
+wmill-1.396.0.dist-info/RECORD,,