[ENH] Added dropdown to select drives and to select files from the tree view while selecting location of new project

datashuttle/tui/screens/modal_dialogs.py

@@ -12,11 +12,20 @@
     from datashuttle.tui.app import TuiApp
     from datashuttle.utils.custom_types import InterfaceOutput
 
+import platform
 from pathlib import Path
 
+import psutil
 from textual.containers import Container, Horizontal
 from textual.screen import ModalScreen
-from textual.widgets import Button, Input, Label, LoadingIndicator, Static
+from textual.widgets import (
+    Button,
+    Input,
+    Label,
+    LoadingIndicator,
+    Select,
+    Static,
+)
 
 from datashuttle.tui.custom_widgets import CustomDirectoryTree
 from datashuttle.tui.utils.tui_decorators import require_double_click
@@ -165,6 +174,10 @@ def __init__(
             path_ = Path().home()
         self.path_ = path_
 
+        if platform.system() == "Windows":
+            self.selected_drive = self.path_.drive + "\\"
+        else:
+            self.selected_drive = "/"
         self.prev_click_time = 0
 
     def compose(self) -> ComposeResult:
@@ -176,6 +189,12 @@ def compose(self) -> ComposeResult:
 
         yield Container(
             Static(label_message, id="select_directory_tree_screen_label"),
+            Select(  # Dropdown for drives
+                [(drive, drive) for drive in self.get_drives()],
+                value=self.selected_drive,
+                allow_blank=False,
+                id="select_directory_tree_drive_select",
+            ),
             CustomDirectoryTree(
                 self.mainwindow,
                 self.path_,
@@ -185,6 +204,24 @@ def compose(self) -> ComposeResult:
             id="select_directory_tree_container",
         )
 
+    @staticmethod
+    def get_drives():
+        if platform.system() == "Windows":
+            return [disk.device for disk in psutil.disk_partitions(all=False)]
+
+        elif platform.system() in ["Darwin", "Linux"]:
+            return ["/"] + [
+                f"/{dir.name}" for dir in Path("/").iterdir() if dir.is_dir()
+            ]
+        return ["/"]
+
+    def on_select_changed(self, event: Select.Changed) -> None:
+        """Updates the directory tree when the drive is changed."""
+        self.path_ = Path(event.value)
+        self.query_one("#select_directory_tree_directory_tree").path = (
+            self.path_
+        )
+
     @require_double_click
     def on_directory_tree_directory_selected(self, node) -> None:
         if node.path.is_file():