Coverage for mindsdb / api / executor / sql_query / steps / multiple_step.py: 29%

1from mindsdb.api.executor.planner.steps import MultipleSteps 

2 

3from mindsdb.api.executor.exceptions import NotSupportedYet 

4 

5from .base import BaseStepCall 

6 

7 

8class MultipleStepsCall(BaseStepCall): 

9 

10 bind = MultipleSteps 

11 

12 def call(self, step): 

13 

14 if step.reduce != 'union': 

15 raise NotSupportedYet(f"Only MultipleSteps with type = 'union' is supported. Got '{step.type}'") 

16 data = None 

17 for substep in step.steps: 

18 subdata = self.sql_query.execute_step(substep) 

19 if data is None: 

20 data = subdata 

21 else: 

22 data.add_from_result_set(subdata) 

23 

24 return data