added support for LIST remove_item and remove_at commands, LIST inser…

.gitignore

@@ -6,4 +6,5 @@ __pycache__/
 graph.gv
 graph.gv.pdf
 *.log
-.venv
+.venv
+venv

datastructs.py

@@ -353,6 +353,26 @@ def evaluate(self, conditions, recStack, lookup=None):
                                                                      recStack=recStack)
             return result
 
+        if 'remove_item' in self.getName().lower():
+            print("hello!")
+            arguments = flattenAlgorithmWithConditions(self.commands[0].getChildren(),conditions, recStack=recStack)
+            result = flattenAlgorithmWithConditions(self.depends[0],conditions, recStack=recStack)
+            for argument in arguments:
+                for item in result:
+                    if item[0] == argument.getValue():
+                        result=[i for i in result if i != item]
+            print('result',result)
+            return result;
+
+        if 'remove_at' in self.getName().lower():
+            arguments = self.commands[0].getChildren()
+            result = flattenAlgorithmWithConditions(self.depends[0],conditions, recStack=recStack)
+            for argument in arguments:
+                del result[argument.getValue()]
+            return result;
+
+
+
 
 class ConcatNode(Node):
     def __init__(self, name: str):

test/variable_test.py

@@ -1874,7 +1874,7 @@ def test_list_remove_variable(self):
         list(REMOVE_ITEM foo b)
         """
         self.runTool(text)
-        # self.vmodel.export()
+        self.vmodel.export()
 
     def test_variable_growth(self):
         text = """
@@ -1913,5 +1913,55 @@ def test_sample_1(self):
         """
         self.runTool(text)
 
+
+    def test_conditional_list_remove_variable(self):
+        text = """
+        option(opt1 "des1" YES)
+        set(foo john doe bar)
+        if(opt1)
+            list(REMOVE_ITEM foo john boo)
+        endif() 
+        """
+        self.runTool(text)
+        var = self.lookup.getKey('${foo}')
+        a = flattenAlgorithmWithConditions(var)
+        self.vmodel.export()
+
+    def test_complex_conditional_list_remove_variable(self):
+        text = """
+        option(opt1 "des" YES)
+        option(opt2 "des" YES)
+
+        set(var_a foo bar john doe)
+        if(opt1)
+            list(REMOVE_ITEM var_a bar Farshad)
+        elseif(opt2)
+            list(REMOVE_ITEM var_a john Mehran)
+        endif()
+        """
+        self.runTool(text)
+        self.vmodel.export()
+
+
+    def test_remove_at_conditional_list_remove_variable(self):
+        text = """
+        set(var_a foo bar john doe)
+        set(item 2)
+        list(REMOVE_AT var_a ${item})
+        """
+        self.runTool(text)
+        self.vmodel.export()
+
+
+    def test_insert_list_remove_variable(self):
+        # TODO: need implementation
+        text = """
+        set(var_a foo bar john doe)
+        set(item foo)
+        list(INSERT var_a ${item})
+        """
+        self.runTool(text)
+        self.vmodel.export()
+
 if __name__ == '__main__':
     unittest.main()