(
self,
schema: str,
table_columns: dict[str, list[str]],
functions: dict[str, None] | dict[str, Any],
)
| 1209 | self._register_schema_completions(schema_name, table_columns, functions) |
| 1210 | |
| 1211 | def _register_schema_completions( |
| 1212 | self, |
| 1213 | schema: str, |
| 1214 | table_columns: dict[str, list[str]], |
| 1215 | functions: dict[str, None] | dict[str, Any], |
| 1216 | ) -> None: |
| 1217 | self.all_completions.add(schema) |
| 1218 | for table, cols in table_columns.items(): |
| 1219 | self.all_completions.add(table) |
| 1220 | for col in cols: |
| 1221 | if col != "*": |
| 1222 | self.all_completions.add(col) |
| 1223 | for func_name in functions: |
| 1224 | self.all_completions.add(func_name) |
| 1225 | |
| 1226 | def reset_completions(self) -> None: |
| 1227 | self.databases: list[str] = [] |
no test coverage detected