From 2f6565ec872577a85bd47de13fa1f9241bd60d76 Mon Sep 17 00:00:00 2001
From: Fabio Manganiello <blacklight86@gmail.com>
Date: Sun, 29 Apr 2018 22:25:31 +0200
Subject: [PATCH] Added support for if-else constructs in procedures

---
 platypush/procedure/__init__.py | 90 +++++++++++++++++++++++++++++++++
 1 file changed, 90 insertions(+)

diff --git a/platypush/procedure/__init__.py b/platypush/procedure/__init__.py
index 958235edf..e64bc629f 100644
--- a/platypush/procedure/__init__.py
+++ b/platypush/procedure/__init__.py
@@ -30,6 +30,7 @@ class Procedure(object):
     def build(cls, name, async, requests, backend=None, id=None, **kwargs):
         reqs = []
         loop_count = 0
+        if_count = 0
 
         for request_config in requests:
             # Check if this request is a for loop
@@ -56,6 +57,25 @@ class Procedure(object):
                     reqs.append(loop)
                     continue
 
+            # Check if this request is an if-else
+            if len(request_config.keys()) >= 1:
+                key = list(request_config.keys())[0]
+                m = re.match('\s*(if)\s+\$\{(.*)\}\s*', key)
+                if m:
+                    if_count += 1
+                    if_name = '{}__if_{}'.format(name, if_count)
+                    condition = m.group(2)
+                    else_branch = request_config['else'] if 'else' in request_config else []
+
+                    if_proc = IfProcedure(name=if_name,
+                                          requests=request_config[key],
+                                          condition=condition,
+                                          else_branch=else_branch,
+                                          backend=backend, id=id)
+
+                    reqs.append(if_proc)
+                    continue
+
             request_config['origin'] = Config.get('device_id')
             request_config['id'] = id
             if 'target' not in request_config:
@@ -136,5 +156,75 @@ class LoopProcedure(Procedure):
 
         return response
 
+
+class IfProcedure(Procedure):
+    """
+    Models an if-else construct.
+
+    Example:
+
+        procedure.sync.process_results:
+            -
+                action: http.get
+                args:
+                    url: https://some-service/some/json/endpoint
+                    # Example response: { "sensors": [ {"temperature": 18 } ] }
+            -
+                if ${sensors['temperature'] < 20}:
+                    -
+                        action: shell.exec
+                        args:
+                            cmd: '/path/turn_on_heating.sh'
+                else:
+                    -
+                        action: shell.exec
+                        args:
+                            cmd: '/path/turn_off_heating.sh'
+    """
+
+    context = {}
+
+    def __init__(self, name, condition, requests, else_branch=[], backend=None, id=None, **kwargs):
+        kwargs['async'] = False
+        self.condition = condition
+        self.else_branch = []
+        reqs = []
+
+        for req in requests:
+            req['origin'] = Config.get('device_id')
+            req['id'] = id
+            if 'target' not in req:
+                req['target'] = req['origin']
+
+            reqs.append(Request.build(req))
+
+        for req in else_branch:
+            req['origin'] = Config.get('device_id')
+            req['id'] = id
+            if 'target' not in req:
+                req['target'] = req['origin']
+
+            self.else_branch.append(Request.build(req))
+
+        super(). __init__(name=name, requests=reqs, backend=None, **kwargs)
+
+
+    def execute(self, **context):
+        condition_true = eval(self.condition)
+        response = Response()
+
+        if condition_true:
+            response = super().execute(**context)
+        elif self.else_branch:
+            try:
+                reqs = self.requests
+                self.requests = self.else_branch
+                response = super().execute(**context)
+            finally:
+                self.requests = reqs
+
+        return response
+
+
 # vim:sw=4:ts=4:et: