harlequin 2.5.2__tar.gz → 2.6.0__tar.gz

{harlequin-2.5.2 → harlequin-2.6.0}/CHANGELOG.md

@@ -4,6 +4,16 @@ All notable changes to this project will be documented in this file.
 
 ## [Unreleased]
 
+## [2.6.0] - 2026-08-03
+
+### Performance
+
+- Autocomplete no longer re-sorts the full completion list every time the Data Catalog lazy-loads a node's children; new completions are staged and merged at most once per second. This keeps the UI responsive while the catalog loads in the background, especially against high-latency (remote) databases.
+
+### Bug Fixes
+
+- Refreshes the file tree in the Data Catalog after Harlequin writes a file, either by saving from the editor or exporting data, so newly written files appear without a manual refresh ([#871](https://github.com/tconbeer/harlequin/issues/871)).
+
 ## [2.5.2] - 2026-03-25
 
 ### Bug Fixes
@@ -873,7 +883,8 @@ All notable changes to this project will be documented in this file.
 
 - Use the DuckDB CLI.
 
-[unreleased]: https://github.com/tconbeer/harlequin/compare/2.5.2...HEAD
+[unreleased]: https://github.com/tconbeer/harlequin/compare/2.6.0...HEAD
+[2.6.0]: https://github.com/tconbeer/harlequin/compare/2.5.2...2.6.0
 [2.5.2]: https://github.com/tconbeer/harlequin/compare/2.5.1...2.5.2
 [2.5.1]: https://github.com/tconbeer/harlequin/compare/2.5.0...2.5.1
 [2.5.0]: https://github.com/tconbeer/harlequin/compare/2.4.1...2.5.0

{harlequin-2.5.2 → harlequin-2.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: harlequin
-Version: 2.5.2
+Version: 2.6.0
 Summary: The SQL IDE for Your Terminal.
 Project-URL: Homepage, https://harlequin.sh
 Project-URL: Documentation, https://harlequin.sh/docs/getting-started/index

{harlequin-2.5.2 → harlequin-2.6.0}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 
 name = "harlequin"
-version = "2.5.2"
+version = "2.6.0"
 requires-python = ">=3.10"
 description = "The SQL IDE for Your Terminal."
 authors = [
@@ -68,14 +68,15 @@ dev = [
 ]
 static = [
     "ruff>=0.14.1",
-    "mypy==1.18.2,<2",
+    "mypy==2.0.0,<3",
     "pandas-stubs>=2,<3",
     "boto3-stubs>=1.34.23,<2",
 ]
 test = [
     "pytest>=9.0.2,<9.1.0",
     "pytest-asyncio==1.3.0",
-    "pytest-textual-snapshot @ https://github.com/tconbeer/pytest-textual-snapshot.git#main",
+    "pytest-xdist>=3.6",
+    "pytest-textual-snapshot @ git+https://github.com/tconbeer/pytest-textual-snapshot.git@a2c6763b45283f76757a123342900accefcf04e8",
     "boto3==1.34.22",
     # extension tests require consistent version of duckdb.
     "duckdb==1.5.1",
@@ -186,6 +187,10 @@ strict_equality = true
 
 [tool.pytest.ini_options]
 
+# Tests run in parallel, one worker per CPU. Pass -n0 to run them in a single
+# process, e.g. when using a debugger or -s.
+addopts = ["-n", "auto"]
+
 markers = [
     "online: requeries internet connection (deselect with '-m \"not online\"')",
     "use_cache: override the autouse fixture the disables the buffer cache",

{harlequin-2.5.2 → harlequin-2.6.0}/src/harlequin/app.py

@@ -27,6 +27,7 @@ from textual.lazy import Lazy
 from textual.message import Message
 from textual.reactive import reactive
 from textual.screen import Screen, ScreenResultCallbackType, ScreenResultType
+from textual.timer import Timer
 from textual.types import CSSPathType
 from textual.widget import AwaitMount, Widget
 from textual.widgets import Button, Footer, Input
@@ -229,6 +230,7 @@ class Harlequin(AppBase):
         self.query_timer: Union[float, None] = None
         self.connection: HarlequinConnection | None = None
         self.harlequin_driver = HarlequinDriver(app=self)
+        self._completer_merge_timer: Timer | None = None
 
         if keymap_names is None:
             keymap_names = ("vscode",)
@@ -889,11 +891,15 @@ class Harlequin(AppBase):
         if table is None:
             show_export_error(error=ValueError("You must execute a query first."))
             return
-        notify = partial(self.notify, "Data exported successfully.")
+
+        def on_export_success() -> None:
+            self.notify("Data exported successfully.")
+            self.data_catalog.update_file_tree()
+
         callback = partial(
             export_callback,
             table=table,
-            success_callback=notify,
+            success_callback=on_export_success,
             error_callback=show_export_error,
         )
         self.app.push_screen(
@@ -1168,12 +1174,27 @@ class Harlequin(AppBase):
             self.editor_collection.word_completer is not None
             and self.editor_collection.member_completer is not None
         ):
+            # appending to the catalog is cheap, but merging the full
+            # completion list is O(n log n), so when the Data Catalog's lazy
+            # loader delivers a stream of items (one message per node), we
+            # defer the merge and run it at most once per second.
             self.editor_collection.word_completer.extend_catalog(
-                parent=parent, items=items
+                parent=parent, items=items, defer_merge=True
             )
             self.editor_collection.member_completer.extend_catalog(
-                parent=parent, items=items
+                parent=parent, items=items, defer_merge=True
             )
+            if self._completer_merge_timer is None:
+                self._completer_merge_timer = self.set_timer(
+                    1.0, self._merge_completers
+                )
+
+    def _merge_completers(self) -> None:
+        self._completer_merge_timer = None
+        if self.editor_collection.word_completer is not None:
+            self.editor_collection.word_completer.merge()
+        if self.editor_collection.member_completer is not None:
+            self.editor_collection.member_completer.merge()
 
     def update_completers(self, catalog: Catalog) -> None:
         if self.connection is None:

{harlequin-2.5.2 → harlequin-2.6.0}/src/harlequin/autocomplete/completers.py

@@ -25,12 +25,8 @@ class WordCompleter:
         self._function_completions = function_completions
         self._catalog_completions = catalog_completions
         self._extra_completions = extra_completions or []
-        self.completions: list[HarlequinCompletion] = self._merge_completions(
-            self._keyword_completions,
-            self._function_completions,
-            self._catalog_completions,
-            self._extra_completions,
-        )
+        self.completions: list[HarlequinCompletion] = []
+        self.merge()
 
     def __call__(self, prefix: str) -> list[tuple[tuple[str, str], str]]:
         """
@@ -64,18 +60,29 @@ class WordCompleter:
 
     def update_catalog(self, catalog: Catalog) -> None:
         self._catalog_completions = build_catalog_completions(catalog=catalog)
-        self.completions = self._merge_completions(
-            self._keyword_completions,
-            self._function_completions,
-            self._catalog_completions,
-            self._extra_completions,
-        )
+        self.merge()
 
-    def extend_catalog(self, parent: CatalogItem, items: list[CatalogItem]) -> None:
+    def extend_catalog(
+        self,
+        parent: CatalogItem,
+        items: list[CatalogItem],
+        defer_merge: bool = False,
+    ) -> None:
         # TODO: dedupe/merge on the parent's unique key, so we can load items from
         # a cache and update them later when they are lazy-loaded.
         new_completions = _build_children_completions(items=items, context=parent.label)
         self._catalog_completions.extend(new_completions)
+        if not defer_merge:
+            self.merge()
+
+    def merge(self) -> None:
+        """
+        Rebuild the merged, sorted completions list. This is O(n log n) in the
+        total number of completions, so callers that extend the catalog many
+        times in quick succession (e.g., the Data Catalog's lazy loader) should
+        pass defer_merge=True to extend_catalog and call this once, on a
+        debounce.
+        """
         self.completions = self._merge_completions(
             self._keyword_completions,
             self._function_completions,

{harlequin-2.5.2 → harlequin-2.6.0}/src/harlequin/components/code_editor.py

@@ -81,6 +81,8 @@ class CodeEditor(TextEditor, inherit_bindings=False):
 
     def on_text_area_saved(self, message: TextAreaSaved) -> None:
         self.app.notify(f"Editor contents saved to {message.path}")
+        if hasattr(self.app, "data_catalog"):
+            self.app.data_catalog.update_file_tree()
 
     def on_text_area_clipboard_error(self) -> None:
         if not self.has_shown_clipboard_error: