@marimo-team/frontend 0.23.9-dev20 → 0.23.9-dev21
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/assets/{CellStatus-cST_KROR.js → CellStatus-BqCsCfF0.js} +1 -1
- package/dist/assets/{JsonOutput-BwgMLuPn.js → JsonOutput-7sNcAFVa.js} +1 -1
- package/dist/assets/{MarimoErrorOutput-DIbgMg0T.js → MarimoErrorOutput-BOi3ico3.js} +1 -1
- package/dist/assets/{RenderHTML-CH8QrLVl.js → RenderHTML-CUf8D25v.js} +1 -1
- package/dist/assets/{add-cell-with-ai-DDmHtapY.js → add-cell-with-ai-DoLVEWe2.js} +1 -1
- package/dist/assets/{add-connection-dialog-CRbrpdQu.js → add-connection-dialog-C9ehEIhe.js} +1 -1
- package/dist/assets/{agent-panel-Ci6kmMWq.js → agent-panel-DfGsNTLw.js} +1 -1
- package/dist/assets/{ai-model-dropdown-CNobct-x.js → ai-model-dropdown-BWdck8rF.js} +1 -1
- package/dist/assets/{app-config-button-DrlCOjyd.js → app-config-button-BJhb3-gi.js} +1 -1
- package/dist/assets/{cell-editor-Bhvnqdc-.js → cell-editor-ZKy9HWcX.js} +1 -1
- package/dist/assets/{cell-link-rh-JpkCu.js → cell-link-DLb_k4Kf.js} +1 -1
- package/dist/assets/{cells-Cwex_Zk1.js → cells-yL2qMkYn.js} +41 -41
- package/dist/assets/{chat-display-DzBxD2g1.js → chat-display-UA8v27kw.js} +1 -1
- package/dist/assets/{chat-panel-ipOkNc7y.js → chat-panel-BGtfb6Wo.js} +1 -1
- package/dist/assets/{chat-ui-DqHdyg0V.js → chat-ui-CfYDiSmX.js} +1 -1
- package/dist/assets/{column-preview-WtwlsutV.js → column-preview-D1138kdA.js} +1 -1
- package/dist/assets/{command-palette-C5YuwxOo.js → command-palette-D9JIsy_G.js} +1 -1
- package/dist/assets/{common-B3E9c-gU.js → common-DqZ15mGE.js} +1 -1
- package/dist/assets/{components-DFkvRpxg.js → components-BK3xAU_5.js} +1 -1
- package/dist/assets/{components-wm4TXPI6.js → components-DXfCmhfc.js} +1 -1
- package/dist/assets/{datasource-rDrNZEmo.js → datasource-CJC51Mx7.js} +1 -1
- package/dist/assets/{dependency-graph-panel-BWe9yDlP.js → dependency-graph-panel-avUgiyMA.js} +1 -1
- package/dist/assets/{documentation-panel-BAxMFFrB.js → documentation-panel-BX7dDKRB.js} +1 -1
- package/dist/assets/{download-NRqwJhRT.js → download-DucsG72p.js} +1 -1
- package/dist/assets/{edit-page-CNuf6Bpw.js → edit-page-BG3s4KtX.js} +3 -3
- package/dist/assets/{error-panel-CByNONxp.js → error-panel-BqoOUjfw.js} +1 -1
- package/dist/assets/{file-explorer-panel-DcCFx-Mu.js → file-explorer-panel-8hhN0YIE.js} +1 -1
- package/dist/assets/{file-icons-CV0odlbf.js → file-icons-CNUmy2Dl.js} +1 -1
- package/dist/assets/{floating-outline-DmddQK6b.js → floating-outline-B_uWfcys.js} +1 -1
- package/dist/assets/{focus-DLnEejrn.js → focus-Cyz1EKwA.js} +1 -1
- package/dist/assets/{form-Dnx_5kyZ.js → form-Dyt6aN_7.js} +1 -1
- package/dist/assets/{home-page-ffvNN1h7.js → home-page-CGoMMRBf.js} +1 -1
- package/dist/assets/{hooks-DuHQuYPv.js → hooks-CQnd87Km.js} +1 -1
- package/dist/assets/{html-to-image-CzpeYelq.js → html-to-image-CKzwKb6F.js} +1 -1
- package/dist/assets/{index-BN0lrf4C.js → index-vrfD6Tng.js} +3 -3
- package/dist/assets/{kiosk-mode-C6QWFFID.js → kiosk-mode-z2bbZ6_w.js} +1 -1
- package/dist/assets/{layout-dxaLq-eR.js → layout-Brpm-l5C.js} +3 -3
- package/dist/assets/{logs-panel-9W91k1zJ.js → logs-panel-CNV8j7by.js} +1 -1
- package/dist/assets/{markdown-renderer-B-iPeXPA.js → markdown-renderer-3mlKZ6Wa.js} +1 -1
- package/dist/assets/{name-cell-input-uFfmqWYT.js → name-cell-input-BnL86ka3.js} +1 -1
- package/dist/assets/{outline-panel-BAWIWb20.js → outline-panel-k7h8ZSSH.js} +1 -1
- package/dist/assets/{packages-panel-DncGPBWC.js → packages-panel-CHPpObHs.js} +1 -1
- package/dist/assets/{panels-ByGgkdwq.js → panels-DlR3fy59.js} +1 -1
- package/dist/assets/{process-output-D06o1mmH.js → process-output-Bi7Z_Agd.js} +1 -1
- package/dist/assets/{radio-group-DUTOVun5.js → radio-group-CdBpUP4h.js} +1 -1
- package/dist/assets/{readonly-python-code-BnaYQo0K.js → readonly-python-code-beWODwjc.js} +1 -1
- package/dist/assets/{reveal-component-BXfhZjrl.js → reveal-component-2Gg2VcRe.js} +1 -1
- package/dist/assets/{run-page-Um3pVJx9.js → run-page-BJKTdA2B.js} +1 -1
- package/dist/assets/{scratchpad-panel-D8kI0OKl.js → scratchpad-panel-D1N-Kxhd.js} +1 -1
- package/dist/assets/{session-panel-07dCccS4.js → session-panel-C5xz5D-N.js} +1 -1
- package/dist/assets/{snippets-panel-s3hpmPTY.js → snippets-panel-rCxyqna5.js} +1 -1
- package/dist/assets/{state-DIFGjBXT.js → state-B6DTPcu0.js} +1 -1
- package/dist/assets/{state-B5L8iIIr.js → state-vUGPvNzI.js} +1 -1
- package/dist/assets/{textarea-D0VOSGcU.js → textarea-DRcRrFRn.js} +1 -1
- package/dist/assets/{tracing-CSRhEdtZ.js → tracing-DKMmxWD9.js} +1 -1
- package/dist/assets/{tracing-panel-DuaSjEhP.js → tracing-panel-KG023vH2.js} +2 -2
- package/dist/assets/{useCellActionButton-CtygwcqH.js → useCellActionButton-ZXUUkx8s.js} +1 -1
- package/dist/assets/{useDeleteCell-DeZ7T5A3.js → useDeleteCell-wNHy5lEM.js} +1 -1
- package/dist/assets/{useDependencyPanelTab-Bpa86UXQ.js → useDependencyPanelTab-B7Qj36AL.js} +1 -1
- package/dist/assets/{useNotebookActions-DuMI8tMy.js → useNotebookActions-CXlvqZKD.js} +1 -1
- package/dist/assets/{useRunCells-CScbBaxj.js → useRunCells-CJadPmdw.js} +1 -1
- package/dist/assets/{useSplitCell-CBlrUeHM.js → useSplitCell-C04l3e__.js} +1 -1
- package/dist/index.html +22 -22
- package/package.json +1 -1
- package/src/components/databases/display.tsx +2 -0
- package/src/components/datasources/__tests__/utils.test.ts +82 -0
- package/src/components/datasources/utils.ts +16 -15
- package/src/core/codemirror/format.ts +1 -0
- package/src/core/codemirror/language/languages/sql/sql.ts +1 -0
- package/src/core/codemirror/language/languages/sql/utils.ts +2 -0
package/dist/index.html
CHANGED
|
@@ -66,7 +66,7 @@
|
|
|
66
66
|
<marimo-server-token data-token="{{ server_token }}" hidden></marimo-server-token>
|
|
67
67
|
<!-- /TODO -->
|
|
68
68
|
<title>{{ title }}</title>
|
|
69
|
-
<script type="module" crossorigin src="./assets/index-
|
|
69
|
+
<script type="module" crossorigin src="./assets/index-vrfD6Tng.js"></script>
|
|
70
70
|
<link rel="modulepreload" crossorigin href="./assets/preload-helper-BPPi7vOr.js">
|
|
71
71
|
<link rel="modulepreload" crossorigin href="./assets/chunk-LvLJmgfZ.js">
|
|
72
72
|
<link rel="modulepreload" crossorigin href="./assets/react-Bj1aDYRI.js">
|
|
@@ -133,7 +133,7 @@
|
|
|
133
133
|
<link rel="modulepreload" crossorigin href="./assets/debounce-DhnxH9Rh.js">
|
|
134
134
|
<link rel="modulepreload" crossorigin href="./assets/database-zap-kIkTfzTX.js">
|
|
135
135
|
<link rel="modulepreload" crossorigin href="./assets/main-B0OX4z33.js">
|
|
136
|
-
<link rel="modulepreload" crossorigin href="./assets/cells-
|
|
136
|
+
<link rel="modulepreload" crossorigin href="./assets/cells-yL2qMkYn.js">
|
|
137
137
|
<link rel="modulepreload" crossorigin href="./assets/ErrorBoundary-DyYDV0HI.js">
|
|
138
138
|
<link rel="modulepreload" crossorigin href="./assets/kbd-CTUAEnEx.js">
|
|
139
139
|
<link rel="modulepreload" crossorigin href="./assets/useInstallPackage-DUF4IRRI.js">
|
|
@@ -147,35 +147,35 @@
|
|
|
147
147
|
<link rel="modulepreload" crossorigin href="./assets/usePress-DQ_tAz5W.js">
|
|
148
148
|
<link rel="modulepreload" crossorigin href="./assets/input-C3Hrdlqq.js">
|
|
149
149
|
<link rel="modulepreload" crossorigin href="./assets/ImperativeModal-B3Th7k4R.js">
|
|
150
|
-
<link rel="modulepreload" crossorigin href="./assets/cell-link-
|
|
150
|
+
<link rel="modulepreload" crossorigin href="./assets/cell-link-DLb_k4Kf.js">
|
|
151
151
|
<link rel="modulepreload" crossorigin href="./assets/multi-map-CUuNtzHt.js">
|
|
152
152
|
<link rel="modulepreload" crossorigin href="./assets/alert-yTS3WpF4.js">
|
|
153
153
|
<link rel="modulepreload" crossorigin href="./assets/chevron-right-CG5QYXYk.js">
|
|
154
154
|
<link rel="modulepreload" crossorigin href="./assets/dropdown-menu-CR7cnzLX.js">
|
|
155
155
|
<link rel="modulepreload" crossorigin href="./assets/links-D1JoyKTt.js">
|
|
156
|
-
<link rel="modulepreload" crossorigin href="./assets/useRunCells-
|
|
156
|
+
<link rel="modulepreload" crossorigin href="./assets/useRunCells-CJadPmdw.js">
|
|
157
157
|
<link rel="modulepreload" crossorigin href="./assets/copy-LK56fFow.js">
|
|
158
158
|
<link rel="modulepreload" crossorigin href="./assets/copy-BwrPA9zQ.js">
|
|
159
159
|
<link rel="modulepreload" crossorigin href="./assets/copy-icon-OjtDb4gO.js">
|
|
160
|
-
<link rel="modulepreload" crossorigin href="./assets/RenderHTML-
|
|
161
|
-
<link rel="modulepreload" crossorigin href="./assets/datasource-
|
|
162
|
-
<link rel="modulepreload" crossorigin href="./assets/state-
|
|
160
|
+
<link rel="modulepreload" crossorigin href="./assets/RenderHTML-CUf8D25v.js">
|
|
161
|
+
<link rel="modulepreload" crossorigin href="./assets/datasource-CJC51Mx7.js">
|
|
162
|
+
<link rel="modulepreload" crossorigin href="./assets/state-B6DTPcu0.js">
|
|
163
163
|
<link rel="modulepreload" crossorigin href="./assets/package-Tv6ztuzw.js">
|
|
164
164
|
<link rel="modulepreload" crossorigin href="./assets/sparkles-lWUAsPhp.js">
|
|
165
|
-
<link rel="modulepreload" crossorigin href="./assets/MarimoErrorOutput-
|
|
165
|
+
<link rel="modulepreload" crossorigin href="./assets/MarimoErrorOutput-BOi3ico3.js">
|
|
166
166
|
<link rel="modulepreload" crossorigin href="./assets/spinner-Bhir8k53.js">
|
|
167
|
-
<link rel="modulepreload" crossorigin href="./assets/html-to-image-
|
|
168
|
-
<link rel="modulepreload" crossorigin href="./assets/focus-
|
|
167
|
+
<link rel="modulepreload" crossorigin href="./assets/html-to-image-CKzwKb6F.js">
|
|
168
|
+
<link rel="modulepreload" crossorigin href="./assets/focus-Cyz1EKwA.js">
|
|
169
169
|
<link rel="modulepreload" crossorigin href="./assets/useAsyncData-bgszE9F0.js">
|
|
170
170
|
<link rel="modulepreload" crossorigin href="./assets/LazyAnyLanguageCodeMirror-DFv3yT20.js">
|
|
171
171
|
<link rel="modulepreload" crossorigin href="./assets/micromark-factory-space-BygYYKhs.js">
|
|
172
172
|
<link rel="modulepreload" crossorigin href="./assets/chunk-5FQGJX7Z-BzXnQz5s.js">
|
|
173
|
-
<link rel="modulepreload" crossorigin href="./assets/markdown-renderer-
|
|
173
|
+
<link rel="modulepreload" crossorigin href="./assets/markdown-renderer-3mlKZ6Wa.js">
|
|
174
174
|
<link rel="modulepreload" crossorigin href="./assets/command-KARR7KMq.js">
|
|
175
175
|
<link rel="modulepreload" crossorigin href="./assets/field-zLmMOSA4.js">
|
|
176
176
|
<link rel="modulepreload" crossorigin href="./assets/popover-Bz_0Vkyf.js">
|
|
177
177
|
<link rel="modulepreload" crossorigin href="./assets/errors-vr57w7Ul.js">
|
|
178
|
-
<link rel="modulepreload" crossorigin href="./assets/download-
|
|
178
|
+
<link rel="modulepreload" crossorigin href="./assets/download-DucsG72p.js">
|
|
179
179
|
<link rel="modulepreload" crossorigin href="./assets/table-BGPSHfig.js">
|
|
180
180
|
<link rel="modulepreload" crossorigin href="./assets/useIframeCapabilities-CcI1zSdn.js">
|
|
181
181
|
<link rel="modulepreload" crossorigin href="./assets/error-banner-LdWZDbqd.js">
|
|
@@ -196,25 +196,25 @@
|
|
|
196
196
|
<link rel="modulepreload" crossorigin href="./assets/plus-BgB18UzY.js">
|
|
197
197
|
<link rel="modulepreload" crossorigin href="./assets/trash-2-rVklqqFF.js">
|
|
198
198
|
<link rel="modulepreload" crossorigin href="./assets/react-resizable-panels.browser.esm-CV8-hvjx.js">
|
|
199
|
-
<link rel="modulepreload" crossorigin href="./assets/JsonOutput-
|
|
199
|
+
<link rel="modulepreload" crossorigin href="./assets/JsonOutput-7sNcAFVa.js">
|
|
200
200
|
<link rel="modulepreload" crossorigin href="./assets/chart-no-axes-column-nqk474t8.js">
|
|
201
201
|
<link rel="modulepreload" crossorigin href="./assets/square-function-uY_yJr5g.js">
|
|
202
202
|
<link rel="modulepreload" crossorigin href="./assets/spec-CPQR_o92.js">
|
|
203
203
|
<link rel="modulepreload" crossorigin href="./assets/ellipsis-vertical-CkwWkOQL.js">
|
|
204
204
|
<link rel="modulepreload" crossorigin href="./assets/refresh-cw-DHwG4Mac.js">
|
|
205
205
|
<link rel="modulepreload" crossorigin href="./assets/tree-actions-BM_EJr3E.js">
|
|
206
|
-
<link rel="modulepreload" crossorigin href="./assets/components-
|
|
207
|
-
<link rel="modulepreload" crossorigin href="./assets/column-preview-
|
|
206
|
+
<link rel="modulepreload" crossorigin href="./assets/components-DXfCmhfc.js">
|
|
207
|
+
<link rel="modulepreload" crossorigin href="./assets/column-preview-D1138kdA.js">
|
|
208
208
|
<link rel="modulepreload" crossorigin href="./assets/icons-Ol38nIbL.js">
|
|
209
|
-
<link rel="modulepreload" crossorigin href="./assets/radio-group-
|
|
210
|
-
<link rel="modulepreload" crossorigin href="./assets/floating-outline-
|
|
209
|
+
<link rel="modulepreload" crossorigin href="./assets/radio-group-CdBpUP4h.js">
|
|
210
|
+
<link rel="modulepreload" crossorigin href="./assets/floating-outline-B_uWfcys.js">
|
|
211
211
|
<link rel="modulepreload" crossorigin href="./assets/objectWithoutPropertiesLoose-DfWeGRFv.js">
|
|
212
212
|
<link rel="modulepreload" crossorigin href="./assets/esm-CqWdmSnV.js">
|
|
213
|
-
<link rel="modulepreload" crossorigin href="./assets/readonly-python-code-
|
|
213
|
+
<link rel="modulepreload" crossorigin href="./assets/readonly-python-code-beWODwjc.js">
|
|
214
214
|
<link rel="modulepreload" crossorigin href="./assets/file-headphone-BrQspHac.js">
|
|
215
215
|
<link rel="modulepreload" crossorigin href="./assets/file-BrdxGLRX.js">
|
|
216
216
|
<link rel="modulepreload" crossorigin href="./assets/image-DQHXdEQn.js">
|
|
217
|
-
<link rel="modulepreload" crossorigin href="./assets/file-icons-
|
|
217
|
+
<link rel="modulepreload" crossorigin href="./assets/file-icons-CNUmy2Dl.js">
|
|
218
218
|
<link rel="modulepreload" crossorigin href="./assets/switch-BrbMxTq7.js">
|
|
219
219
|
<link rel="modulepreload" crossorigin href="./assets/events-CPoJAfgx.js">
|
|
220
220
|
<link rel="modulepreload" crossorigin href="./assets/globals-CDDbRvHE.js">
|
|
@@ -223,11 +223,11 @@
|
|
|
223
223
|
<link rel="modulepreload" crossorigin href="./assets/memoize-Tp7rARFe.js">
|
|
224
224
|
<link rel="modulepreload" crossorigin href="./assets/get-C-qh_et5.js">
|
|
225
225
|
<link rel="modulepreload" crossorigin href="./assets/_baseSet-CxV9N1bc.js">
|
|
226
|
-
<link rel="modulepreload" crossorigin href="./assets/state-
|
|
226
|
+
<link rel="modulepreload" crossorigin href="./assets/state-vUGPvNzI.js">
|
|
227
227
|
<link rel="modulepreload" crossorigin href="./assets/label-xHqFtfdz.js">
|
|
228
|
-
<link rel="modulepreload" crossorigin href="./assets/textarea-
|
|
228
|
+
<link rel="modulepreload" crossorigin href="./assets/textarea-DRcRrFRn.js">
|
|
229
229
|
<link rel="modulepreload" crossorigin href="./assets/refresh-ccw-C-n2VFP5.js">
|
|
230
|
-
<link rel="modulepreload" crossorigin href="./assets/form-
|
|
230
|
+
<link rel="modulepreload" crossorigin href="./assets/form-Dyt6aN_7.js">
|
|
231
231
|
<link rel="modulepreload" crossorigin href="./assets/renderShortcut-BrHz5aiT.js">
|
|
232
232
|
<link rel="modulepreload" crossorigin href="./assets/useBoolean-CtP63TnG.js">
|
|
233
233
|
<link rel="modulepreload" crossorigin href="./assets/useDeepCompareMemoize-zUHU--0D.js">
|
package/package.json
CHANGED
|
@@ -313,6 +313,26 @@ describe("sqlCode", () => {
|
|
|
313
313
|
);
|
|
314
314
|
});
|
|
315
315
|
|
|
316
|
+
it("should preserve dots inside quoted schema names", () => {
|
|
317
|
+
const sqlTableContext: SQLTableContext = {
|
|
318
|
+
engine: "postgres",
|
|
319
|
+
schema: "analytics.events",
|
|
320
|
+
defaultSchema: "public",
|
|
321
|
+
defaultDatabase: "mydb",
|
|
322
|
+
database: "remote",
|
|
323
|
+
dialect: "postgres",
|
|
324
|
+
};
|
|
325
|
+
|
|
326
|
+
const result = sqlCode({
|
|
327
|
+
table: mockTable,
|
|
328
|
+
columnName: mockColumn.name,
|
|
329
|
+
sqlTableContext,
|
|
330
|
+
});
|
|
331
|
+
expect(result).toBe(
|
|
332
|
+
'_df = mo.sql(f"""\nSELECT "email" FROM "remote"."analytics.events"."users" LIMIT 100\n""", engine=postgres)',
|
|
333
|
+
);
|
|
334
|
+
});
|
|
335
|
+
|
|
316
336
|
it("should not quote * column name", () => {
|
|
317
337
|
const sqlTableContext: SQLTableContext = {
|
|
318
338
|
engine: "postgres",
|
|
@@ -334,6 +354,68 @@ describe("sqlCode", () => {
|
|
|
334
354
|
});
|
|
335
355
|
});
|
|
336
356
|
|
|
357
|
+
describe("Dremio dialect", () => {
|
|
358
|
+
it("should quote reserved column names and table path parts", () => {
|
|
359
|
+
const sqlTableContext: SQLTableContext = {
|
|
360
|
+
engine: "dremio_conn",
|
|
361
|
+
schema: "operations",
|
|
362
|
+
defaultSchema: "",
|
|
363
|
+
defaultDatabase: "",
|
|
364
|
+
database: "lakehouse",
|
|
365
|
+
dialect: "dremio",
|
|
366
|
+
};
|
|
367
|
+
|
|
368
|
+
const result = sqlCode({
|
|
369
|
+
table: { ...mockTable, name: "shipments" as const },
|
|
370
|
+
columnName: "order",
|
|
371
|
+
sqlTableContext,
|
|
372
|
+
});
|
|
373
|
+
expect(result).toBe(
|
|
374
|
+
'_df = mo.sql(f"""\nSELECT "order" FROM "lakehouse"."operations"."shipments" LIMIT 100\n""", engine=dremio_conn)',
|
|
375
|
+
);
|
|
376
|
+
});
|
|
377
|
+
|
|
378
|
+
it("should not quote * column name", () => {
|
|
379
|
+
const sqlTableContext: SQLTableContext = {
|
|
380
|
+
engine: "dremio_conn",
|
|
381
|
+
schema: "operations",
|
|
382
|
+
defaultSchema: "",
|
|
383
|
+
defaultDatabase: "",
|
|
384
|
+
database: "lakehouse",
|
|
385
|
+
dialect: "dremio",
|
|
386
|
+
};
|
|
387
|
+
|
|
388
|
+
const result = sqlCode({
|
|
389
|
+
table: { ...mockTable, name: "customers" as const },
|
|
390
|
+
columnName: "*",
|
|
391
|
+
sqlTableContext,
|
|
392
|
+
});
|
|
393
|
+
expect(result).toBe(
|
|
394
|
+
'_df = mo.sql(f"""\nSELECT * FROM "lakehouse"."operations"."customers" LIMIT 100\n""", engine=dremio_conn)',
|
|
395
|
+
);
|
|
396
|
+
});
|
|
397
|
+
|
|
398
|
+
it("should preserve dots inside quoted schema names", () => {
|
|
399
|
+
const sqlTableContext: SQLTableContext = {
|
|
400
|
+
engine: "dremio_conn",
|
|
401
|
+
schema: "samples.dremio.com",
|
|
402
|
+
defaultSchema: "",
|
|
403
|
+
defaultDatabase: "",
|
|
404
|
+
database: "Samples",
|
|
405
|
+
dialect: "dremio",
|
|
406
|
+
};
|
|
407
|
+
|
|
408
|
+
const result = sqlCode({
|
|
409
|
+
table: { ...mockTable, name: "airlines" as const },
|
|
410
|
+
columnName: "*",
|
|
411
|
+
sqlTableContext,
|
|
412
|
+
});
|
|
413
|
+
expect(result).toBe(
|
|
414
|
+
'_df = mo.sql(f"""\nSELECT * FROM "Samples"."samples.dremio.com"."airlines" LIMIT 100\n""", engine=dremio_conn)',
|
|
415
|
+
);
|
|
416
|
+
});
|
|
417
|
+
});
|
|
418
|
+
|
|
337
419
|
describe("fallback behavior", () => {
|
|
338
420
|
it("should use default formatter for unknown dialect", () => {
|
|
339
421
|
const sqlTableContext: SQLTableContext = {
|
|
@@ -15,9 +15,9 @@ export function isSchemaless(schemaName: string) {
|
|
|
15
15
|
|
|
16
16
|
interface SqlCodeFormatter {
|
|
17
17
|
/**
|
|
18
|
-
* Format the table
|
|
18
|
+
* Format the table path based on dialect-specific rules
|
|
19
19
|
*/
|
|
20
|
-
|
|
20
|
+
formatTablePath: (tablePath: string[]) => string;
|
|
21
21
|
/**
|
|
22
22
|
* Format the SELECT clause
|
|
23
23
|
*/
|
|
@@ -25,7 +25,7 @@ interface SqlCodeFormatter {
|
|
|
25
25
|
}
|
|
26
26
|
|
|
27
27
|
const defaultFormatter: SqlCodeFormatter = {
|
|
28
|
-
|
|
28
|
+
formatTablePath: (tablePath: string[]) => tablePath.join("."),
|
|
29
29
|
formatSelectClause: (columnName: string, tableName: string) =>
|
|
30
30
|
`SELECT ${columnName} FROM ${tableName} LIMIT 100`,
|
|
31
31
|
};
|
|
@@ -41,7 +41,8 @@ function getFormatter(dialect: string): SqlCodeFormatter {
|
|
|
41
41
|
const quote = BigQueryDialect.spec.identifierQuotes;
|
|
42
42
|
return {
|
|
43
43
|
// BigQuery uses backticks for identifiers
|
|
44
|
-
|
|
44
|
+
formatTablePath: (tablePath: string[]) =>
|
|
45
|
+
`${quote}${tablePath.join(".")}${quote}`,
|
|
45
46
|
formatSelectClause: defaultFormatter.formatSelectClause,
|
|
46
47
|
};
|
|
47
48
|
}
|
|
@@ -49,7 +50,7 @@ function getFormatter(dialect: string): SqlCodeFormatter {
|
|
|
49
50
|
case "sqlserver":
|
|
50
51
|
case "microsoft sql server":
|
|
51
52
|
return {
|
|
52
|
-
|
|
53
|
+
formatTablePath: defaultFormatter.formatTablePath,
|
|
53
54
|
formatSelectClause: (columnName: string, tableName: string) =>
|
|
54
55
|
`SELECT TOP 100 ${columnName} FROM ${tableName}`,
|
|
55
56
|
};
|
|
@@ -57,12 +58,11 @@ function getFormatter(dialect: string): SqlCodeFormatter {
|
|
|
57
58
|
case "postgres":
|
|
58
59
|
case "postgresql":
|
|
59
60
|
case "duckdb":
|
|
61
|
+
case "dremio":
|
|
60
62
|
// Quote column and table names to avoid raising errors on weird characters
|
|
61
63
|
return {
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
return parts.map((part) => `"${part}"`).join(".");
|
|
65
|
-
},
|
|
64
|
+
formatTablePath: (tablePath: string[]) =>
|
|
65
|
+
tablePath.map((part) => `"${part}"`).join("."),
|
|
66
66
|
formatSelectClause: (columnName: string, tableName: string) =>
|
|
67
67
|
`SELECT ${columnName === "*" ? "*" : `"${columnName}"`} FROM ${tableName} LIMIT 100`,
|
|
68
68
|
};
|
|
@@ -114,26 +114,27 @@ export function sqlCode({
|
|
|
114
114
|
database,
|
|
115
115
|
dialect,
|
|
116
116
|
} = sqlTableContext;
|
|
117
|
-
|
|
117
|
+
const tablePath = [table.name];
|
|
118
118
|
|
|
119
119
|
// Set the fully qualified table name based on schema and database
|
|
120
120
|
if (isSchemaless(schema)) {
|
|
121
|
-
|
|
122
|
-
database
|
|
121
|
+
if (database !== defaultDatabase) {
|
|
122
|
+
tablePath.unshift(database);
|
|
123
|
+
}
|
|
123
124
|
} else {
|
|
124
125
|
// Include schema if it's not the default schema
|
|
125
126
|
if (schema !== defaultSchema) {
|
|
126
|
-
|
|
127
|
+
tablePath.unshift(schema);
|
|
127
128
|
}
|
|
128
129
|
|
|
129
130
|
// Include database if it's not the default database
|
|
130
131
|
if (database !== defaultDatabase) {
|
|
131
|
-
|
|
132
|
+
tablePath.unshift(database);
|
|
132
133
|
}
|
|
133
134
|
}
|
|
134
135
|
|
|
135
136
|
const formatter = getFormatter(dialect);
|
|
136
|
-
const formattedTableName = formatter.
|
|
137
|
+
const formattedTableName = formatter.formatTablePath(tablePath);
|
|
137
138
|
const selectClause = formatter.formatSelectClause(
|
|
138
139
|
columnName,
|
|
139
140
|
formattedTableName,
|
|
@@ -52,6 +52,7 @@ const KNOWN_DIALECTS_ARRAY = [
|
|
|
52
52
|
"databricks",
|
|
53
53
|
"datafusion",
|
|
54
54
|
"microsoft sql server",
|
|
55
|
+
"dremio",
|
|
55
56
|
] as const;
|
|
56
57
|
const KNOWN_DIALECTS: ReadonlySet<string> = new Set(KNOWN_DIALECTS_ARRAY);
|
|
57
58
|
type KnownDialect = (typeof KNOWN_DIALECTS_ARRAY)[number];
|
|
@@ -115,6 +116,7 @@ export function guessDialect(
|
|
|
115
116
|
case "spark":
|
|
116
117
|
case "databricks":
|
|
117
118
|
case "datafusion":
|
|
119
|
+
case "dremio":
|
|
118
120
|
Logger.debug("Unsupported dialect", { dialect });
|
|
119
121
|
return ModifiedStandardSQL;
|
|
120
122
|
default:
|