skip_next_task method was added

Author: davidBMSTU

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = cotea
-version = 1.3.13
+version = 1.3.14
 author = David Badalyan
 author_email = [email protected]
 description = Tool that provides Python API to run Ansible programmatically.

src/cotea/runner.py

@@ -429,4 +429,9 @@ def add_var_as_extra_var(self, new_var_name, value):
 
 
     def get_stats(self):
-        return self.play_wrp.custom_stats
+        return self.play_wrp.custom_stats
+
+
+    def skip_next_task(self):
+        skipped_task_name = self.task_wrp.skip_next_task()
+        return skipped_task_name

src/cotea/wrappers/get_next_task_wrapper.py

@@ -210,8 +210,22 @@ def get_prev_task_name(self):
 
 
     def skip_next_task(self):
+        skipped_tasks = []
         for host in self.hosts_left:
-            self.ansible_iterator.get_next_task_for_host(host)
+            host_state_block_and_task = self.play_iterator.get_next_task_for_host(host)
+            skipped_task = host_state_block_and_task[len(host_state_block_and_task) - 1]
+            skipped_tasks.append(skipped_task)
+
+        skipped_task_name = ""
+
+        if len(skipped_tasks) > 0:
+            # task should be tha same on all hosts
+            task = skipped_tasks[0]
+
+            if hasattr(task, "get_name"):
+                skipped_task_name = str(task.get_name())
+        
+        return skipped_task_name
 
 
     def dont_add_last_task_after_new(self):