zj/diskmanager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add raid

Browse Source
This commit is contained in:
zj
2026-02-02 04:20:00 +08:00
parent a22ea24274
commit be5a0bc2c7
12 changed files with 1162 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

210
mainwindow.py
View File

@@ -1,18 +1,14 @@
# mainwindow.py
import sys
import logging # 导入 logging 模块
import logging
from PySide6.QtWidgets import (QApplication, QMainWindow, QTreeWidgetItem,
QMessageBox, QHeaderView, QMenu, QInputDialog)
from PySide6.QtCore import Qt, QPoint
# 导入自动生成的 UI 文件
from ui_form import Ui_MainWindow
# 导入我们自己编写的系统信息管理模块
from system_info import SystemInfoManager
# 导入日志配置
from logger_config import setup_logging, logger # 导入日志配置函数和 logger 实例
# 导入磁盘操作模块
from logger_config import setup_logging, logger
from disk_operations import DiskOperations
class MainWindow(QMainWindow):
@@ -21,35 +17,39 @@ class MainWindow(QMainWindow):
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
# 设置日志输出到 QTextEdit
setup_logging(self.ui.logOutputTextEdit)
logger.info("应用程序启动。")
# 初始化系统信息管理器和磁盘操作管理器
self.system_manager = SystemInfoManager()
self.disk_ops = DiskOperations()
# 连接刷新按钮的信号到槽函数
if hasattr(self.ui, 'refreshButton'):
self.ui.refreshButton.clicked.connect(self.refresh_block_devices_info)
self.ui.refreshButton.clicked.connect(self.refresh_all_info)
else:
logger.warning("Warning: refreshButton not found in UI. Please add it in form.ui and regenerate ui_form.py.")
# 启用 treeWidget 的自定义上下文菜单
self.ui.treeWidget_block_devices.setContextMenuPolicy(Qt.CustomContextMenu)
self.ui.treeWidget_block_devices.customContextMenuRequested.connect(self.show_block_device_context_menu)
# 初始化时刷新一次数据
self.refresh_all_info()
logger.info("所有设备信息已初始化加载。")
def refresh_all_info(self):
"""
刷新所有设备信息：块设备、RAID和LVM。
"""
logger.info("开始刷新所有设备信息...")
self.refresh_block_devices_info()
logger.info("块设备信息已初始化加载。")
self.refresh_raid_info()
self.refresh_lvm_info()
logger.info("所有设备信息刷新完成。")
def refresh_block_devices_info(self):
"""
刷新块设备信息并显示在 QTreeWidget 中。
"""
self.ui.treeWidget_block_devices.clear() # 清空现有内容
self.ui.treeWidget_block_devices.clear()
# 定义所有要显示的列头和对应的 lsblk 字段名
columns = [
("设备名", 'name'),
("类型", 'type'),
@@ -67,12 +67,11 @@ class MainWindow(QMainWindow):
]
headers = [col[0] for col in columns]
self.field_keys = [col[1] for col in columns] # 保存字段键，供后续使用
self.field_keys = [col[1] for col in columns]
self.ui.treeWidget_block_devices.setColumnCount(len(headers)) # 确保列数正确
self.ui.treeWidget_block_devices.setColumnCount(len(headers))
self.ui.treeWidget_block_devices.setHeaderLabels(headers)
# 调整列宽以适应内容
for i in range(len(headers)):
self.ui.treeWidget_block_devices.header().setSectionResizeMode(i, QHeaderView.ResizeToContents)
@@ -81,7 +80,6 @@ class MainWindow(QMainWindow):
for dev in devices:
self._add_device_to_tree(self.ui.treeWidget_block_devices, dev)
# 自动调整列宽
for i in range(len(headers)):
self.ui.treeWidget_block_devices.resizeColumnToContents(i)
logger.info("块设备信息刷新成功。")
@@ -98,21 +96,164 @@ class MainWindow(QMainWindow):
item = QTreeWidgetItem(parent_item)
for i, key in enumerate(self.field_keys):
value = dev_data.get(key)
if key == 'ro': # 特殊处理布尔值
if key == 'ro':
item.setText(i, "是" if value else "否")
elif value is None: # None 值显示为空字符串
elif value is None:
item.setText(i, "")
else:
item.setText(i, str(value))
# 将原始设备数据存储在 item 的 data 属性中，方便后续操作时获取
item.setData(0, Qt.UserRole, dev_data)
# 如果有子设备（分区），也显示出来
if 'children' in dev_data:
for child in dev_data['children']:
self._add_device_to_tree(item, child)
item.setExpanded(True) # 默认展开父节点，以便看到分区
item.setExpanded(True)
def refresh_raid_info(self):
"""
刷新RAID阵列信息并显示在 QTreeWidget 中。
"""
self.ui.treeWidget_raid.clear()
# 定义RAID显示列头
raid_headers = [
"阵列设备", "级别", "状态", "大小", "活动设备", "失败设备", "备用设备",
"总设备数", "UUID", "名称", "Chunk Size"
]
self.ui.treeWidget_raid.setColumnCount(len(raid_headers))
self.ui.treeWidget_raid.setHeaderLabels(raid_headers)
for i in range(len(raid_headers)):
self.ui.treeWidget_raid.header().setSectionResizeMode(i, QHeaderView.ResizeToContents)
try:
raid_arrays = self.system_manager.get_mdadm_arrays()
if not raid_arrays:
item = QTreeWidgetItem(self.ui.treeWidget_raid)
item.setText(0, "未找到RAID阵列。")
logger.info("未找到RAID阵列。")
return
for array in raid_arrays:
array_item = QTreeWidgetItem(self.ui.treeWidget_raid)
array_item.setText(0, array.get('device', 'N/A'))
array_item.setText(1, array.get('level', 'N/A'))
array_item.setText(2, array.get('state', 'N/A'))
array_item.setText(3, array.get('array_size', 'N/A'))
array_item.setText(4, array.get('active_devices', 'N/A'))
array_item.setText(5, array.get('failed_devices', 'N/A'))
array_item.setText(6, array.get('spare_devices', 'N/A'))
array_item.setText(7, array.get('total_devices', 'N/A'))
array_item.setText(8, array.get('uuid', 'N/A'))
array_item.setText(9, array.get('name', 'N/A'))
array_item.setText(10, array.get('chunk_size', 'N/A'))
array_item.setExpanded(True)
# 添加成员设备作为子节点
for member in array.get('member_devices', []):
member_item = QTreeWidgetItem(array_item)
member_item.setText(0, f" {member.get('device_path', 'N/A')}") # 缩进显示
member_item.setText(1, f"成员")
member_item.setText(2, member.get('state', 'N/A'))
# 其他列留空或填充N/A，因为是成员设备的特有信息
member_item.setText(3, f"RaidDevice: {member.get('raid_device', 'N/A')}")
for i in range(len(raid_headers)):
self.ui.treeWidget_raid.resizeColumnToContents(i)
logger.info("RAID阵列信息刷新成功。")
except Exception as e:
QMessageBox.critical(self, "错误", f"刷新RAID阵列信息失败: {e}")
logger.error(f"刷新RAID阵列信息失败: {e}")
def refresh_lvm_info(self):
"""
刷新LVM信息（PVs, VGs, LVs）并显示在 QTreeWidget 中。
"""
self.ui.treeWidget_lvm.clear()
# 定义LVM显示列头
lvm_headers = ["名称", "大小", "属性", "UUID", "关联", "空闲/已用", "路径"]
self.ui.treeWidget_lvm.setColumnCount(len(lvm_headers))
self.ui.treeWidget_lvm.setHeaderLabels(lvm_headers)
for i in range(len(lvm_headers)):
self.ui.treeWidget_lvm.header().setSectionResizeMode(i, QHeaderView.ResizeToContents)
try:
lvm_data = self.system_manager.get_lvm_info()
if not lvm_data.get('pvs') and not lvm_data.get('vgs') and not lvm_data.get('lvs'):
item = QTreeWidgetItem(self.ui.treeWidget_lvm)
item.setText(0, "未找到LVM信息。")
logger.info("未找到LVM信息。")
return
# 物理卷 (PVs)
pv_root_item = QTreeWidgetItem(self.ui.treeWidget_lvm)
pv_root_item.setText(0, "物理卷 (PVs)")
pv_root_item.setExpanded(True)
if lvm_data.get('pvs'):
for pv in lvm_data['pvs']:
pv_item = QTreeWidgetItem(pv_root_item)
pv_item.setText(0, pv.get('pv_name', 'N/A'))
pv_item.setText(1, pv.get('pv_size', 'N/A'))
pv_item.setText(2, pv.get('pv_attr', 'N/A'))
pv_item.setText(3, pv.get('pv_uuid', 'N/A'))
pv_item.setText(4, f"VG: {pv.get('vg_name', 'N/A')}")
pv_item.setText(5, f"空闲: {pv.get('pv_free', 'N/A')}")
pv_item.setText(6, pv.get('pv_fmt', 'N/A')) # PV格式
else:
item = QTreeWidgetItem(pv_root_item)
item.setText(0, "未找到物理卷。")
# 卷组 (VGs)
vg_root_item = QTreeWidgetItem(self.ui.treeWidget_lvm)
vg_root_item.setText(0, "卷组 (VGs)")
vg_root_item.setExpanded(True)
if lvm_data.get('vgs'):
for vg in lvm_data['vgs']:
vg_item = QTreeWidgetItem(vg_root_item)
vg_item.setText(0, vg.get('vg_name', 'N/A'))
vg_item.setText(1, vg.get('vg_size', 'N/A'))
vg_item.setText(2, vg.get('vg_attr', 'N/A'))
vg_item.setText(3, vg.get('vg_uuid', 'N/A'))
vg_item.setText(4, f"PVs: {vg.get('pv_count', 'N/A')}, LVs: {vg.get('lv_count', 'N/A')}")
vg_item.setText(5, f"空闲: {vg.get('vg_free', 'N/A')}, 已分配: {vg.get('vg_alloc_percent', 'N/A')}%")
vg_item.setText(6, vg.get('vg_fmt', 'N/A')) # VG格式
else:
item = QTreeWidgetItem(vg_root_item)
item.setText(0, "未找到卷组。")
# 逻辑卷 (LVs)
lv_root_item = QTreeWidgetItem(self.ui.treeWidget_lvm)
lv_root_item.setText(0, "逻辑卷 (LVs)")
lv_root_item.setExpanded(True)
if lvm_data.get('lvs'):
for lv in lvm_data['lvs']:
lv_item = QTreeWidgetItem(lv_root_item)
lv_item.setText(0, lv.get('lv_name', 'N/A'))
lv_item.setText(1, lv.get('lv_size', 'N/A'))
lv_item.setText(2, lv.get('lv_attr', 'N/A'))
lv_item.setText(3, lv.get('lv_uuid', 'N/A'))
lv_item.setText(4, f"VG: {lv.get('vg_name', 'N/A')}, Origin: {lv.get('origin', 'N/A')}")
lv_item.setText(5, f"快照: {lv.get('snap_percent', 'N/A')}%")
lv_item.setText(6, lv.get('lv_path', 'N/A'))
else:
item = QTreeWidgetItem(lv_root_item)
item.setText(0, "未找到逻辑卷。")
for i in range(len(lvm_headers)):
self.ui.treeWidget_lvm.resizeColumnToContents(i)
logger.info("LVM信息刷新成功。")
except Exception as e:
QMessageBox.critical(self, "错误", f"刷新LVM信息失败: {e}")
logger.error(f"刷新LVM信息失败: {e}")
def show_block_device_context_menu(self, pos: QPoint):
"""
@@ -120,7 +261,7 @@ class MainWindow(QMainWindow):
"""
item = self.ui.treeWidget_block_devices.itemAt(pos)
if item:
dev_data = item.data(0, Qt.UserRole) # 获取存储的原始设备数据
dev_data = item.data(0, Qt.UserRole)
if not dev_data:
logger.warning(f"无法获取设备 {item.text(0)} 的详细数据。")
return
@@ -131,22 +272,17 @@ class MainWindow(QMainWindow):
menu = QMenu(self)
# 挂载/卸载操作
# 只有 'part' (分区) 和 'disk' (整个磁盘，但通常只挂载分区) 可以被挂载/卸载
if device_type in ['part', 'disk']:
if not mount_point or mount_point == '' or mount_point == 'N/A': # 未挂载
if not mount_point or mount_point == '' or mount_point == 'N/A':
mount_action = menu.addAction(f"挂载 {device_name}...")
mount_action.triggered.connect(lambda: self._handle_mount(device_name))
elif mount_point != '[SWAP]': # 已挂载且不是SWAP
elif mount_point != '[SWAP]':
unmount_action = menu.addAction(f"卸载 {device_name}")
unmount_action.triggered.connect(lambda: self._handle_unmount(device_name))
# 分隔符，用于区分操作
if menu.actions():
menu.addSeparator()
# 删除分区和格式化操作
# 这些操作通常只针对 'part' (分区)
if device_type == 'part':
delete_action = menu.addAction(f"删除分区 {device_name}")
delete_action.triggered.connect(lambda: self._handle_delete_partition(device_name))
@@ -154,7 +290,7 @@ class MainWindow(QMainWindow):
format_action = menu.addAction(f"格式化分区 {device_name}...")
format_action.triggered.connect(lambda: self._handle_format_partition(device_name))
if menu.actions(): # 只有当菜单中有动作时才显示
if menu.actions():
menu.exec(self.ui.treeWidget_block_devices.mapToGlobal(pos))
else:
logger.info(f"设备 {device_name} 没有可用的操作。")
@@ -164,22 +300,22 @@ class MainWindow(QMainWindow):
def _handle_mount(self, device_name):
"""处理挂载操作，并刷新UI。"""
if self.disk_ops.mount_partition(device_name):
self.refresh_block_devices_info() # 操作成功后刷新UI
self.refresh_all_info()
def _handle_unmount(self, device_name):
"""处理卸载操作，并刷新UI。"""
if self.disk_ops.unmount_partition(device_name):
self.refresh_block_devices_info() # 操作成功后刷新UI
self.refresh_all_info()
def _handle_delete_partition(self, device_name):
"""处理删除分区操作，并刷新UI。"""
if self.disk_ops.delete_partition(device_name):
self.refresh_block_devices_info() # 操作成功后刷新UI
self.refresh_all_info()
def _handle_format_partition(self, device_name):
"""处理格式化分区操作，并刷新UI。"""
if self.disk_ops.format_partition(device_name):
self.refresh_block_devices_info() # 操作成功后刷新UI
self.refresh_all_info()
if __name__ == "__main__":