<<< autocomplete column names of currently selected *frame

Author: gdementen

larray_editor/editor.py

@@ -474,18 +474,21 @@ def __init__(self, editor_window):
         self.completer.setCaseSensitivity(Qt.CaseSensitivity.CaseInsensitive)
         self.completer.setWidget(self)
         self.completer.activated.connect(self.insert_completion)
-        self.update_completer_options()
-        self.sql_context = pl.SQLContext()
-
-    def update_completer_options(self, data=None):
-        if data is not None:
-            data = self._filter_data_for_sql(data)
-            self.sql_context.register_many(data)
-            var_names = [k for k, v in data.items()]
+        self.sql_context = pl.SQLContext(eager=False)
+        self.update_completer_options({})
+
+    def update_completer_options(self, data, selected=None):
+        data = self._filter_data_for_sql(data)
+        if selected is not None and self._handled_by_polars_sql(selected):
+            data['self'] = selected
+            col_names = selected.collect_schema().names()
         else:
-            var_names = []
-        logger.debug(f"available variables for SQL queries: {var_names}")
-        completions = var_names + self.SQL_KEYWORDS
+            col_names = []
+        self.sql_context.register_many(data)
+        table_names = [k for k, v in data.items()]
+        logger.debug(f"available tables for SQL queries: {table_names}")
+        logger.debug(f"available columns for SQL queries: {col_names}")
+        completions = table_names + col_names + self.SQL_KEYWORDS
         model = QStringListModel(completions, self.completer)
         self.completer.setModel(model)
 
@@ -1246,6 +1249,8 @@ def set_current_array(self, array, expr_text):
         # this method is also called to refresh the widget value because of an
         # inplace setitem
 
+        if self.sql_widget is not None:
+            self.sql_widget.update_completer_options(self.data, selected=array)
         # FIXME: we should never store the current_array but current_adapter instead
         self.current_array = array
         self.arraywidget.set_data(array)