[SQL Lab] Add function names to autocomplete (#9012)

This commit is contained in:
Erik Ritter
2020-01-24 11:08:26 -08:00
committed by GitHub
parent a85df65f85
commit a4b8e0b9e4
8 changed files with 62 additions and 3 deletions

View File

@@ -161,6 +161,10 @@ class Database(
def allows_subquery(self) -> bool:
return self.db_engine_spec.allows_subqueries
@property
def function_names(self) -> List[str]:
return self.db_engine_spec.get_function_names(self)
@property
def allows_cost_estimate(self) -> bool:
extra = self.get_extra()
@@ -320,7 +324,7 @@ class Database(
return self.get_dialect().identifier_preparer.quote
def get_df( # pylint: disable=too-many-locals
self, sql: str, schema: str, mutator: Optional[Callable] = None
self, sql: str, schema: Optional[str] = None, mutator: Optional[Callable] = None
) -> pd.DataFrame:
sqls = [str(s).strip(" ;") for s in sqlparse.parse(sql)]
source_key = None