fix22

dialogs.py

@@ -0,0 +1,445 @@
+# dialogs.py
+import os
+from PySide6.QtWidgets import (QDialog, QVBoxLayout, QHBoxLayout, QLabel,
+                               QLineEdit, QComboBox, QDoubleSpinBox,
+                               QPushButton, QFormLayout, QFileDialog,
+                               QMessageBox, QCheckBox, QListWidget, QListWidgetItem)
+from PySide6.QtCore import Qt
+
+class CreatePartitionDialog(QDialog):
+    def __init__(self, parent=None, disk_path="", total_disk_mib=0.0, max_available_mib=0.0):
+        super().__init__(parent)
+        self.setWindowTitle("创建分区")
+        self.setMinimumWidth(300)
+
+        self.disk_path = disk_path
+        self.total_disk_mib = total_disk_mib # 磁盘总大小 (MiB)
+        self.max_available_mib = max_available_mib # 可用空间 (MiB)
+
+        self.partition_table_type_combo = QComboBox()
+        self.partition_table_type_combo.addItems(["gpt", "msdos"])
+
+        self.size_spinbox = QDoubleSpinBox()
+        self.size_spinbox.setMinimum(0.1) # 最小分区大小 0.1 GB
+        self.size_spinbox.setMaximum(self.max_available_mib / 1024.0) # 将 MiB 转换为 GB 显示
+        self.size_spinbox.setSuffix(" GB")
+        self.size_spinbox.setDecimals(2)
+
+        self.use_max_space_checkbox = QCheckBox("使用最大可用空间")
+        self.use_max_space_checkbox.setChecked(True) # 默认选中
+
+        self._setup_ui()
+        self._connect_signals()
+        self._initialize_state()
+
+    def _setup_ui(self):
+        layout = QVBoxLayout(self)
+
+        form_layout = QFormLayout()
+        form_layout.addRow("磁盘路径:", QLabel(self.disk_path))
+        form_layout.addRow("分区表类型:", self.partition_table_type_combo)
+        form_layout.addRow("分区大小:", self.size_spinbox)
+        form_layout.addRow("", self.use_max_space_checkbox) # 添加复选框
+
+        layout.addLayout(form_layout)
+
+        button_box = QHBoxLayout()
+        self.confirm_button = QPushButton("确定")
+        self.cancel_button = QPushButton("取消")
+        button_box.addWidget(self.confirm_button)
+        button_box.addWidget(self.cancel_button)
+        layout.addLayout(button_box)
+
+    def _connect_signals(self):
+        self.confirm_button.clicked.connect(self.accept)
+        self.cancel_button.clicked.connect(self.reject)
+        self.use_max_space_checkbox.stateChanged.connect(self._toggle_size_input)
+
+    def _initialize_state(self):
+        # 根据默认选中状态设置 spinbox
+        self._toggle_size_input(self.use_max_space_checkbox.checkState())
+
+    def _toggle_size_input(self, state):
+        if state == Qt.Checked:
+            self.size_spinbox.setDisabled(True)
+            self.size_spinbox.setValue(self.size_spinbox.maximum()) # 设置为最大可用 GB
+        else:
+            self.size_spinbox.setDisabled(False)
+            # 如果之前是最大值，取消勾选后，恢复到最小值或一个合理值
+            if self.size_spinbox.value() == self.size_spinbox.maximum():
+                self.size_spinbox.setValue(self.size_spinbox.minimum())
+
+    def get_partition_info(self):
+        size_gb = self.size_spinbox.value()
+        use_max_space = self.use_max_space_checkbox.isChecked()
+
+        if not use_max_space and size_gb <= 0:
+            QMessageBox.warning(self, "输入错误", "分区大小必须大于0。")
+            return None
+        if not use_max_space and size_gb > self.max_available_mib / 1024.0:
+            QMessageBox.warning(self, "输入错误", "分区大小不能超过最大可用空间。")
+            return None
+
+        return {
+            'disk_path': self.disk_path,
+            'partition_table_type': self.partition_table_type_combo.currentText(),
+            'size_gb': size_gb,
+            'total_disk_mib': self.total_disk_mib, # 传递磁盘总大小 (MiB)
+            'use_max_space': use_max_space # 返回此标志
+        }
+
+class MountDialog(QDialog):
+    def __init__(self, parent=None, device_path=""):
+        super().__init__(parent)
+        self.setWindowTitle("挂载分区")
+        self.setMinimumWidth(300)
+
+        self.device_path = device_path
+        self.mount_point_input = QLineEdit()
+        self.add_to_fstab_checkbox = QCheckBox("开机自动挂载 (添加到 /etc/fstab)") # 新增
+
+        self._setup_ui()
+        self._connect_signals()
+
+    def _setup_ui(self):
+        layout = QVBoxLayout(self)
+
+        form_layout = QFormLayout()
+        form_layout.addRow("设备路径:", QLabel(self.device_path))
+        form_layout.addRow("挂载点:", self.mount_point_input)
+        form_layout.addRow("", self.add_to_fstab_checkbox) # 添加复选框
+
+        layout.addLayout(form_layout)
+
+        button_box = QHBoxLayout()
+        self.confirm_button = QPushButton("确定")
+        self.cancel_button = QPushButton("取消")
+        button_box.addWidget(self.confirm_button)
+        button_box.addWidget(self.cancel_button)
+        layout.addLayout(button_box)
+
+        # 默认挂载点，可以根据设备名生成
+        default_mount_point = f"/mnt/{os.path.basename(self.device_path)}"
+        self.mount_point_input.setText(default_mount_point)
+
+    def _connect_signals(self):
+        self.confirm_button.clicked.connect(self.accept)
+        self.cancel_button.clicked.connect(self.reject)
+
+    def get_mount_info(self):
+        mount_point = self.mount_point_input.text().strip()
+        add_to_fstab = self.add_to_fstab_checkbox.isChecked()
+
+        if not mount_point:
+            QMessageBox.warning(self, "输入错误", "挂载点不能为空。")
+            return None
+        if not mount_point.startswith('/'):
+            QMessageBox.warning(self, "输入错误", "挂载点必须是绝对路径 (以 '/' 开头)。")
+            return None
+
+        return {
+            'mount_point': mount_point,
+            'add_to_fstab': add_to_fstab
+        }
+
+class CreateRaidDialog(QDialog):
+    def __init__(self, parent=None, available_devices=None):
+        super().__init__(parent)
+        self.setWindowTitle("创建 RAID 阵列")
+        self.setMinimumWidth(350)
+
+        self.available_devices = available_devices if available_devices is not None else []
+        self.selected_devices = []
+
+        self.raid_level_combo = QComboBox()
+        self.raid_level_combo.addItems(["raid0", "raid1", "raid5", "raid6", "raid10"])
+
+        self.chunk_size_spinbox = QDoubleSpinBox()
+        self.chunk_size_spinbox.setMinimum(4)
+        self.chunk_size_spinbox.setMaximum(1024)
+        self.chunk_size_spinbox.setSingleStep(4)
+        self.chunk_size_spinbox.setValue(512) # 默认值
+        self.chunk_size_spinbox.setSuffix(" KB")
+
+        self.device_list_widget = QListWidget()
+        for dev in self.available_devices:
+            item = QListWidgetItem(dev)
+            item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
+            item.setCheckState(Qt.Unchecked)
+            self.device_list_widget.addItem(item)
+
+        self._setup_ui()
+        self._connect_signals()
+
+    def _setup_ui(self):
+        layout = QVBoxLayout(self)
+
+        form_layout = QFormLayout()
+        form_layout.addRow("RAID 级别:", self.raid_level_combo)
+        form_layout.addRow("Chunk 大小:", self.chunk_size_spinbox)
+        form_layout.addRow("选择设备:", self.device_list_widget)
+
+        layout.addLayout(form_layout)
+
+        button_box = QHBoxLayout()
+        self.confirm_button = QPushButton("确定")
+        self.cancel_button = QPushButton("取消")
+        button_box.addWidget(self.confirm_button)
+        button_box.addWidget(self.cancel_button)
+        layout.addLayout(button_box)
+
+    def _connect_signals(self):
+        self.confirm_button.clicked.connect(self.accept)
+        self.cancel_button.clicked.connect(self.reject)
+
+    def get_raid_info(self):
+        self.selected_devices = []
+        for i in range(self.device_list_widget.count()):
+            item = self.device_list_widget.item(i)
+            if item.checkState() == Qt.Checked:
+                self.selected_devices.append(item.text())
+
+        if not self.selected_devices:
+            QMessageBox.warning(self, "输入错误", "请选择至少一个设备来创建 RAID 阵列。")
+            return None
+
+        raid_level = self.raid_level_combo.currentText()
+        num_devices = len(self.selected_devices)
+
+        # RAID level specific checks for minimum devices
+        if raid_level == "raid0" and num_devices < 1: # RAID0 can be 1 device in some contexts, but usually >1
+             QMessageBox.warning(self, "输入错误", "RAID0 至少需要一个设备。")
+             return None
+        elif raid_level == "raid1" and num_devices < 2:
+            QMessageBox.warning(self, "输入错误", "RAID1 至少需要两个设备。")
+            return None
+        elif raid_level == "raid5" and num_devices < 3:
+            QMessageBox.warning(self, "输入错误", "RAID5 至少需要三个设备。")
+            return None
+        elif raid_level == "raid6" and num_devices < 4:
+            QMessageBox.warning(self, "输入错误", "RAID6 至少需要四个设备。")
+            return None
+        elif raid_level == "raid10" and num_devices < 2: # RAID10 needs at least 2 (for 1+0)
+            QMessageBox.warning(self, "输入错误", "RAID10 至少需要两个设备。")
+            return None
+
+        return {
+            'devices': self.selected_devices,
+            'level': self.raid_level_combo.currentText().replace('raid', ''), # 移除 'raid' 前缀
+            'chunk_size': int(self.chunk_size_spinbox.value()) # KB
+        }
+
+class CreatePvDialog(QDialog):
+    def __init__(self, parent=None, available_partitions=None):
+        super().__init__(parent)
+        self.setWindowTitle("创建物理卷 (PV)")
+        self.setMinimumWidth(300)
+
+        self.available_partitions = available_partitions if available_partitions is not None else []
+
+        self.device_combo_box = QComboBox()
+        self.device_combo_box.addItems(self.available_partitions)
+
+        self._setup_ui()
+        self._connect_signals()
+
+    def _setup_ui(self):
+        layout = QVBoxLayout(self)
+
+        form_layout = QFormLayout()
+        form_layout.addRow("选择设备:", self.device_combo_box)
+
+        layout.addLayout(form_layout)
+
+        button_box = QHBoxLayout()
+        self.confirm_button = QPushButton("确定")
+        self.cancel_button = QPushButton("取消")
+        button_box.addWidget(self.confirm_button)
+        button_box.addWidget(self.cancel_button)
+        layout.addLayout(button_box)
+
+    def _connect_signals(self):
+        self.confirm_button.clicked.connect(self.accept)
+        self.cancel_button.clicked.connect(self.reject)
+
+    def get_pv_info(self):
+        device_path = self.device_combo_box.currentText()
+        if not device_path:
+            QMessageBox.warning(self, "输入错误", "请选择一个设备来创建物理卷。")
+            return None
+        return {'device_path': device_path}
+
+class CreateVgDialog(QDialog):
+    def __init__(self, parent=None, available_pvs=None):
+        super().__init__(parent)
+        self.setWindowTitle("创建卷组 (VG)")
+        self.setMinimumWidth(300)
+
+        self.available_pvs = available_pvs if available_pvs is not None else []
+        self.selected_pvs = []
+
+        self.vg_name_input = QLineEdit()
+        self.pv_list_widget = QListWidget()
+        for pv in self.available_pvs:
+            item = QListWidgetItem(pv)
+            item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
+            item.setCheckState(Qt.Unchecked)
+            self.pv_list_widget.addItem(item)
+
+        self._setup_ui()
+        self._connect_signals()
+
+    def _setup_ui(self):
+        layout = QVBoxLayout(self)
+
+        form_layout = QFormLayout()
+        form_layout.addRow("卷组名称:", self.vg_name_input)
+        form_layout.addRow("选择物理卷:", self.pv_list_widget)
+
+        layout.addLayout(form_layout)
+
+        button_box = QHBoxLayout()
+        self.confirm_button = QPushButton("确定")
+        self.cancel_button = QPushButton("取消")
+        button_box.addWidget(self.confirm_button)
+        button_box.addWidget(self.cancel_button)
+        layout.addLayout(button_box)
+
+    def _connect_signals(self):
+        self.confirm_button.clicked.connect(self.accept)
+        self.cancel_button.clicked.connect(self.reject)
+
+    def get_vg_info(self):
+        vg_name = self.vg_name_input.text().strip()
+        self.selected_pvs = []
+        for i in range(self.pv_list_widget.count()):
+            item = self.pv_list_widget.item(i)
+            if item.checkState() == Qt.Checked:
+                self.selected_pvs.append(item.text())
+
+        if not vg_name:
+            QMessageBox.warning(self, "输入错误", "卷组名称不能为空。")
+            return None
+        if not self.selected_pvs:
+            QMessageBox.warning(self, "输入错误", "请选择至少一个物理卷来创建卷组。")
+            return None
+
+        return {
+            'vg_name': vg_name,
+            'pvs': self.selected_pvs
+        }
+
+class CreateLvDialog(QDialog):
+    def __init__(self, parent=None, available_vgs=None, vg_sizes=None):
+        super().__init__(parent)
+        self.setWindowTitle("创建逻辑卷")
+        self.setMinimumWidth(300)
+
+        self.available_vgs = available_vgs if available_vgs is not None else []
+        self.vg_sizes = vg_sizes if vg_sizes is not None else {} # 存储每个 VG 的可用大小 (GB)
+
+        self.lv_name_input = QLineEdit()
+        self.vg_combo_box = QComboBox()
+        self.lv_size_spinbox = QDoubleSpinBox()
+        self.use_max_space_checkbox = QCheckBox("使用最大可用空间") # 新增复选框
+
+        self._setup_ui()
+        self._connect_signals()
+        self._initialize_state() # 初始化状态
+
+    def _setup_ui(self):
+        layout = QVBoxLayout(self)
+
+        form_layout = QFormLayout()
+        form_layout.addRow("逻辑卷名称:", self.lv_name_input)
+        form_layout.addRow("选择卷组:", self.vg_combo_box)
+        form_layout.addRow("逻辑卷大小 (GB):", self.lv_size_spinbox)
+        form_layout.addRow("", self.use_max_space_checkbox) # 添加复选框
+
+        layout.addLayout(form_layout)
+
+        button_box = QHBoxLayout()
+        self.confirm_button = QPushButton("确定")
+        self.cancel_button = QPushButton("取消")
+        button_box.addWidget(self.confirm_button)
+        button_box.addWidget(self.cancel_button)
+        layout.addLayout(button_box)
+
+        self.lv_size_spinbox.setMinimum(0.1) # 逻辑卷最小大小，例如 0.1 GB
+        self.lv_size_spinbox.setSuffix(" GB")
+        self.lv_size_spinbox.setDecimals(2)
+
+    def _connect_signals(self):
+        self.confirm_button.clicked.connect(self.accept)
+        self.cancel_button.clicked.connect(self.reject)
+        self.vg_combo_box.currentIndexChanged.connect(self._update_size_options) # 卷组选择改变时更新大小选项
+        self.use_max_space_checkbox.stateChanged.connect(self._toggle_size_input) # 复选框状态改变时切换输入
+
+    def _initialize_state(self):
+        self.vg_combo_box.addItems(self.available_vgs)
+        self._update_size_options() # 首次调用以设置初始状态
+        self.use_max_space_checkbox.setChecked(True) # 默认选中“使用最大可用空间”
+
+    def _update_size_options(self):
+        """根据选中的卷组更新逻辑卷大小的选项。"""
+        selected_vg = self.vg_combo_box.currentText()
+        max_size_gb = self.vg_sizes.get(selected_vg, 0.0)
+
+        # 确保 max_size_gb 至少是 spinbox 的最小值，以防卷组可用空间过小导致 UI 问题
+        if max_size_gb < self.lv_size_spinbox.minimum():
+            self.lv_size_spinbox.setMinimum(max_size_gb) # 临时将最小值设为实际最大值
+        else:
+            self.lv_size_spinbox.setMinimum(0.1) # 恢复正常最小值
+
+        self.lv_size_spinbox.setMaximum(max_size_gb) # 设置最大值
+
+        # 如果选中了“使用最大可用空间”，则将 spinbox 值设置为最大值
+        if self.use_max_space_checkbox.isChecked():
+            self.lv_size_spinbox.setValue(max_size_gb)
+        else:
+            # 如果当前值超过了新的最大值，则调整为新的最大值
+            if self.lv_size_spinbox.value() > max_size_gb:
+                self.lv_size_spinbox.setValue(max_size_gb)
+            # 如果当前值小于新的最小值 (例如，VG可用空间变为0，最小值被调整为0)，则调整
+            elif self.lv_size_spinbox.value() < self.lv_size_spinbox.minimum():
+                self.lv_size_spinbox.setValue(self.lv_size_spinbox.minimum())
+
+
+    def _toggle_size_input(self, state):
+        """根据“使用最大可用空间”复选框的状态切换大小输入框的启用/禁用状态。"""
+        if state == Qt.Checked:
+            self.lv_size_spinbox.setDisabled(True)
+            self.lv_size_spinbox.setValue(self.lv_size_spinbox.maximum()) # 设置为最大值
+        else:
+            self.lv_size_spinbox.setDisabled(False)
+            # 如果之前是最大值，取消勾选后，恢复到最小值或一个合理值
+            if self.lv_size_spinbox.value() == self.lv_size_spinbox.maximum():
+                self.lv_size_spinbox.setValue(self.lv_size_spinbox.minimum())
+
+
+    def get_lv_info(self):
+        lv_name = self.lv_name_input.text().strip()
+        vg_name = self.vg_combo_box.currentText()
+        size_gb = self.lv_size_spinbox.value()
+        use_max_space = self.use_max_space_checkbox.isChecked() # 获取复选框状态
+
+        if not lv_name:
+            QMessageBox.warning(self, "输入错误", "逻辑卷名称不能为空。")
+            return None
+        if not vg_name:
+            QMessageBox.warning(self, "输入错误", "请选择一个卷组。")
+            return None
+        if not use_max_space and size_gb <= 0:
+            QMessageBox.warning(self, "输入错误", "逻辑卷大小必须大于0。")
+            return None
+        if not use_max_space and size_gb > self.vg_sizes.get(vg_name, 0.0):
+            QMessageBox.warning(self, "输入错误", "逻辑卷大小不能超过卷组的可用空间。")
+            return None
+
+        return {
+            'lv_name': lv_name,
+            'vg_name': vg_name,
+            'size_gb': size_gb,
+            'use_max_space': use_max_space # 返回此标志
+        }
+