osbot-utils 2.31.0__tar.gz → 2.32.0__tar.gz

{osbot_utils-2.31.0 → osbot_utils-2.32.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: osbot_utils
-Version: 2.31.0
+Version: 2.32.0
 Summary: OWASP Security Bot - Utils
 License: MIT
 Author: Dinis Cruz
@@ -23,7 +23,7 @@ Description-Content-Type: text/markdown
 
 Powerful Python util methods and classes that simplify common apis and tasks.
 
-![Current Release](https://img.shields.io/badge/release-v2.31.0-blue)
+![Current Release](https://img.shields.io/badge/release-v2.32.0-blue)
 [![codecov](https://codecov.io/gh/owasp-sbot/OSBot-Utils/graph/badge.svg?token=GNVW0COX1N)](https://codecov.io/gh/owasp-sbot/OSBot-Utils)
 
 

{osbot_utils-2.31.0 → osbot_utils-2.32.0}/README.md

@@ -2,7 +2,7 @@
 
 Powerful Python util methods and classes that simplify common apis and tasks.
 
-![Current Release](https://img.shields.io/badge/release-v2.31.0-blue)
+![Current Release](https://img.shields.io/badge/release-v2.32.0-blue)
 [![codecov](https://codecov.io/gh/owasp-sbot/OSBot-Utils/graph/badge.svg?token=GNVW0COX1N)](https://codecov.io/gh/owasp-sbot/OSBot-Utils)
 
 

{osbot_utils-2.31.0 → osbot_utils-2.32.0}/osbot_utils/helpers/flows/Flow.py

@@ -122,6 +122,11 @@ class Flow(Type_Safe):
     def captured_logs(self):
         return ansis_to_texts(self.captured_exec_logs)
 
+    def durations(self):
+        return self.flow_stats.durations()
+
+    def durations__with_tasks_status(self):
+        return self.flow_stats.durations__with_tasks_status()
 
     async def invoke_flow_target__thread(self, flow):                               # this is a REALLY important method which is used to pin the flow object to the call stack
         return await flow.flow_target(*flow.resolved_args, **flow.resolved_kwargs)          #   which is then used by the Task.find_flow method to find it

{osbot_utils-2.31.0 → osbot_utils-2.32.0}/osbot_utils/helpers/flows/actions/Flow__Stats__Collector.py

@@ -5,6 +5,7 @@ from osbot_utils.helpers.flows.schemas.Schema__Task__Stats   import Schema__Task
 from osbot_utils.helpers.flows.schemas.Schema__Flow__Stats   import Schema__Flow__Stats
 from osbot_utils.type_safe.Type_Safe                         import Type_Safe
 
+FLOW__ERROR_MESSAGE__TASK_FAILED = 'At least one task failed'
 
 class Flow__Stats__Collector(Type_Safe):
     duration               : Duration
@@ -31,6 +32,9 @@ class Flow__Stats__Collector(Type_Safe):
             if flow_error:
                 _.status = Schema__Flow__Status.FAILED
                 _.error_message = str(flow_error)
+            elif _.failed_tasks:
+                _.status = Schema__Flow__Status.FAILED
+                _.error_message = FLOW__ERROR_MESSAGE__TASK_FAILED
             else:
                 _.status = Schema__Flow__Status.COMPLETED
         return self
@@ -44,4 +48,36 @@ class Flow__Stats__Collector(Type_Safe):
         return self
 
     def json(self):                                                                                                     # Return JSON representation of the stats.
-        return self.stats.json()
+        return self.stats.json()
+
+    def durations__with_tasks_status(self):
+        with self.stats as _:
+            tasks_durations = dict()
+            flow_durations  = dict(flow_name     = _.flow_name                ,
+                                   flow_duration = _.duration.duration_seconds,
+                                   flow_status   = _.status.value             ,
+                                   flow_tasks    =  tasks_durations           )
+
+            for task_id, flow_task in self.stats.tasks_stats.items():
+                task_duration   = dict(task_name = flow_task.task_name                     ,
+                                       task_duration  = flow_task.duration.duration_seconds,
+                                       task_status    = flow_task.status.value             )
+                execution_order = flow_task.execution_order
+                tasks_durations[execution_order] = task_duration
+        return flow_durations
+
+    def durations(self):
+        with self.stats as _:
+            if _.status is None:
+                flow_durations = dict()
+            else:
+                tasks_durations = dict()
+                flow_durations  = dict(flow_name     = _.flow_name                ,
+                                       flow_duration = _.duration.duration_seconds,
+                                       flow_status   = _.status.value             ,
+                                       flow_tasks    =  tasks_durations           )
+
+                for task_id, flow_task in self.stats.tasks_stats.items():
+
+                    tasks_durations[flow_task.task_name] = flow_task.duration.duration_seconds
+        return flow_durations

osbot_utils-2.32.0/osbot_utils/version

@@ -0,0 +1 @@
+v2.32.0

{osbot_utils-2.31.0 → osbot_utils-2.32.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name        = "osbot_utils"
-version     = "v2.31.0"
+version     = "v2.32.0"
 description = "OWASP Security Bot - Utils"
 authors     = ["Dinis Cruz <dinis.cruz@owasp.org>"]
 license     = "MIT"

osbot_utils-2.31.0/osbot_utils/version

@@ -1 +0,0 @@
-v2.31.0