Skip to content

Commit dd4e290

Browse files
committed
feat: export SQL engine registration from package
Expose HotdataMarimoEngine and register_hotdata_sql_engine from the top-level package and extend import tests.
1 parent ac68bbc commit dd4e290

2 files changed

Lines changed: 11 additions & 1 deletion

File tree

hotdata_marimo/__init__.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,13 +16,19 @@
1616
recent_results,
1717
run_history,
1818
)
19+
from hotdata_marimo.sql_engine import (
20+
HotdataMarimoEngine,
21+
register_hotdata_sql_engine,
22+
unregister_hotdata_sql_engine,
23+
)
1924
from hotdata_marimo.sql_editor import SqlEditor, sql_editor
2025
from hotdata_marimo.table_browser import TableBrowser, connection_picker, table_browser
2126
from hotdata_marimo.workspace_selector import WorkspaceSelector, workspace_selector_from_env
2227

2328
__all__ = [
2429
"__version__",
2530
"HotdataClient",
31+
"HotdataMarimoEngine",
2632
"QueryResult",
2733
"RecentResults",
2834
"SqlEditor",
@@ -39,11 +45,13 @@
3945
"hotdata_workspace_selector",
4046
"query_result",
4147
"recent_results",
48+
"register_hotdata_sql_engine",
49+
"register_mo_ui_hotdata_aliases",
4250
"run_history",
4351
"sql_editor",
4452
"table_browser",
53+
"unregister_hotdata_sql_engine",
4554
"workspace_selector_from_env",
46-
"register_mo_ui_hotdata_aliases",
4755
]
4856

4957
hotdata_sql_editor = sql_editor

tests/test_imports.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,3 +3,5 @@ def test_package_imports():
33

44
assert hm.HotdataClient is not None
55
assert hm.SqlEditor is not None
6+
assert hm.register_hotdata_sql_engine is not None
7+
assert hm.HotdataMarimoEngine is not None

0 commit comments

Comments
 (0)