Filter for only directories in sync destination quickpick (#596)

databricks · Mar 21, 2023 · 952879b · 952879b
1 parent c493446
commit 952879b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 9 deletions.
diff --git a/packages/databricks-vscode/resources/python/file.workflow-wrapper.py b/packages/databricks-vscode/resources/python/file.workflow-wrapper.py
@@ -8,7 +8,7 @@
 for i, arg in enumerate(sys.argv):
     if i == 0:
         continue
-    if sys.argv[i-1] == "--databricks-python-source":
+    if sys.argv[i-1] == "--databricks-source-file":
         python_file = arg
     elif sys.argv[i-1] == "--databricks-project-root":
         project_root = arg

diff --git a/packages/databricks-vscode/src/configuration/ConnectionCommands.ts b/packages/databricks-vscode/src/configuration/ConnectionCommands.ts
@@ -1,4 +1,8 @@
-import {Cluster, WorkspaceFsEntity} from "@databricks/databricks-sdk";
+import {
+    Cluster,
+    WorkspaceFsEntity,
+    WorkspaceFsUtils,
+} from "@databricks/databricks-sdk";
 import {homedir} from "node:os";
 import {
     Disposable,
@@ -241,13 +245,17 @@ export class ConnectionCommands implements Disposable {
             input.items = children;
             if (workspaceConfigs.enableFilesInWorkspace) {
                 children.push(
-                    ...((await rootDir?.children) ?? []).map((entity) => {
-                        return {
-                            label: entity.basename,
-                            detail: entity.path,
-                            path: entity.path,
-                        };
-                    })
+                    ...((await rootDir?.children) ?? [])
+                        .filter((entity) =>
+                            WorkspaceFsUtils.isDirectory(entity)
+                        )
+                        .map((entity) => {
+                            return {
+                                label: entity.basename,
+                                detail: entity.path,
+                                path: entity.path,
+                            };
+                        })
                 );
             } else {
                 const repos = (await rootDir?.children) ?? [];