Coverage for packages/sql-smith/src/sql_smith/capability/has_from.py: 77%

1from typing import Union 

2 

3from sql_smith.functions import identify_all, listing 

4 

5 

6class HasFromMixin: 

7 def from_(self, *tables: Union[str, "ExpressionInterface"]): 

8 self._from = identify_all(*tables) 

9 return self 

10 

11 def add_from(self, *tables: Union[str, "ExpressionInterface"]): 

12 self._from = (*self._from, *identify_all(*tables)) 

13 return self 

14 

15 def _apply_from(self, query: "ExpressionInterface") -> "ExpressionInterface": 

16 if self._from: 

17 return query.append("FROM {}", listing(self._from)) 

18 return query