From 4495dd6694a22c7ae64da34541be9e8002b69b12 Mon Sep 17 00:00:00 2001
From: Tim-Oliver Husser <thusser@uni-goettingen.de>
Date: Sat, 9 Dec 2023 09:27:56 +0100
Subject: [PATCH] added ConditionalRunner

---
 pyobs/robotic/scripts/__init__.py    |  1 +
 pyobs/robotic/scripts/conditional.py | 58 ++++++++++++++++++++++++++++
 2 files changed, 59 insertions(+)
 create mode 100644 pyobs/robotic/scripts/conditional.py

diff --git a/pyobs/robotic/scripts/__init__.py b/pyobs/robotic/scripts/__init__.py
index 5ad2dfc6..5c0baac5 100644
--- a/pyobs/robotic/scripts/__init__.py
+++ b/pyobs/robotic/scripts/__init__.py
@@ -4,3 +4,4 @@
 from .sequential import SequentialRunner
 from .parallel import ParallelRunner
 from .selector import SelectorScript
+from .conditional import ConditionalRunner
diff --git a/pyobs/robotic/scripts/conditional.py b/pyobs/robotic/scripts/conditional.py
new file mode 100644
index 00000000..1f240e6d
--- /dev/null
+++ b/pyobs/robotic/scripts/conditional.py
@@ -0,0 +1,58 @@
+from __future__ import annotations
+
+import datetime
+import logging
+from typing import Any, Dict, Optional, List, TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from pyobs.robotic import TaskRunner, TaskSchedule, TaskArchive
+from pyobs.robotic.scripts import Script
+
+log = logging.getLogger(__name__)
+
+
+class ConditionalRunner(Script):
+    """Script for running an if condition."""
+
+    __module__ = "pyobs.modules.robotic"
+
+    def __init__(
+        self,
+        condition: str,
+        true: Dict[str, Any],
+        false: Optional[Dict[str, Any]] = None,
+        **kwargs: Any,
+    ):
+        """Initialize a new ConditionalRunner.
+
+        Args:
+            condition: condition to check
+            true: script to run if condition is evaluated as True
+            false: script to run otherwise.
+        """
+        Script.__init__(self, **kwargs)
+        self.condition = condition
+        self.true = true
+        self.false = false
+
+    async def can_run(self) -> bool:
+        return True
+
+    async def run(
+        self,
+        task_runner: TaskRunner,
+        task_schedule: Optional[TaskSchedule] = None,
+        task_archive: Optional[TaskArchive] = None,
+    ) -> None:
+        # evaluate condition
+        ret = eval(self.condition, {"now": datetime.datetime.utcnow()})
+
+        # run scripts
+        if ret:
+            await self.get_object(self.true, Script).run(task_runner, task_schedule, task_archive)
+        else:
+            if self.false is not None:
+                await self.get_object(self.false, Script).run(task_runner, task_schedule, task_archive)
+
+
+__all__ = ["ConditionalRunner"]