fix22
This commit is contained in:
zj
@@ -1,237 +1,536 @@
|
||||
# disk_operations.py
|
||||
import os
|
||||
import subprocess
|
||||
import logging
|
||||
import os
|
||||
import re
|
||||
from PySide6.QtWidgets import QMessageBox, QInputDialog
|
||||
from system_info import SystemInfoManager # 引入 SystemInfoManager 来复用 _run_command
|
||||
|
||||
logger = logging.getLogger(__name__) # 获取当前模块的 logger 实例
|
||||
# 导入我们自己编写的系统信息管理模块
|
||||
from system_info import SystemInfoManager
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class DiskOperations:
|
||||
def __init__(self):
|
||||
self.system_manager = SystemInfoManager() # 复用 SystemInfoManager 的命令执行器
|
||||
self.system_manager = SystemInfoManager() # 实例化 SystemInfoManager
|
||||
|
||||
def _get_device_path(self, device_name):
|
||||
def _execute_shell_command(self, command_list, error_message, root_privilege=True,
|
||||
suppress_critical_dialog_on_stderr_match=None, input_data=None):
|
||||
"""
|
||||
辅助函数：将设备名（如 'sda1'）转换为完整路径（如 '/dev/sda1'）。
|
||||
通用地运行一个 shell 命令，并处理错误。
|
||||
:param command_list: 命令及其参数的列表。
|
||||
:param error_message: 命令失败时显示给用户的错误消息。
|
||||
:param root_privilege: 如果为 True，则使用 sudo 执行命令。
|
||||
:param suppress_critical_dialog_on_stderr_match: 如果 stderr 包含此字符串，则不显示关键错误对话框。
|
||||
:param input_data: 传递给命令stdin的数据 (str)。
|
||||
:return: (True/False, stdout_str, stderr_str)
|
||||
"""
|
||||
if not device_name.startswith('/dev/'):
|
||||
return f'/dev/{device_name}'
|
||||
return device_name
|
||||
# 确保 command_list 中的所有项都是字符串
|
||||
if not all(isinstance(arg, str) for arg in command_list):
|
||||
logger.error(f"命令列表包含非字符串元素: {command_list}")
|
||||
QMessageBox.critical(None, "错误", f"内部错误：尝试执行的命令包含无效参数。\n命令详情: {command_list}")
|
||||
return False, "", "内部错误：命令参数类型不正确。"
|
||||
|
||||
def mount_partition(self, device_name, mount_point=None):
|
||||
"""
|
||||
挂载指定的分区。
|
||||
如果未提供挂载点，会尝试查找现有挂载点或提示用户输入。
|
||||
"""
|
||||
dev_path = self._get_device_path(device_name)
|
||||
logger.info(f"尝试挂载设备: {dev_path}")
|
||||
if root_privilege:
|
||||
command_list = ["sudo"] + command_list
|
||||
|
||||
if not mount_point:
|
||||
# 尝试从系统信息中获取当前挂载点
|
||||
devices = self.system_manager.get_block_devices()
|
||||
found_mount_point = None
|
||||
for dev in devices:
|
||||
if dev.get('name') == device_name:
|
||||
found_mount_point = dev.get('mountpoint')
|
||||
break
|
||||
if 'children' in dev:
|
||||
for child in dev['children']:
|
||||
if child.get('name') == device_name:
|
||||
found_mount_point = child.get('mountpoint')
|
||||
break
|
||||
if found_mount_point:
|
||||
break
|
||||
full_cmd_str = ' '.join(command_list)
|
||||
logger.debug(f"执行命令: {full_cmd_str}")
|
||||
|
||||
if found_mount_point and found_mount_point != '[SWAP]' and found_mount_point != '':
|
||||
# 如果设备已经有挂载点，并且不是SWAP，则直接使用
|
||||
mount_point = found_mount_point
|
||||
logger.info(f"设备 {dev_path} 已经挂载到 {mount_point}。")
|
||||
QMessageBox.information(None, "信息", f"设备 {dev_path} 已经挂载到 {mount_point}。")
|
||||
return True # 已经挂载，视为成功
|
||||
try:
|
||||
result = subprocess.run(
|
||||
command_list,
|
||||
capture_output=True,
|
||||
text=True,
|
||||
check=True,
|
||||
encoding='utf-8',
|
||||
input=input_data
|
||||
)
|
||||
logger.info(f"命令成功: {full_cmd_str}")
|
||||
return True, result.stdout.strip(), result.stderr.strip()
|
||||
except subprocess.CalledProcessError as e:
|
||||
stderr_output = e.stderr.strip()
|
||||
logger.error(f"命令失败: {full_cmd_str}")
|
||||
logger.error(f"退出码: {e.returncode}")
|
||||
logger.error(f"标准输出: {e.stdout.strip()}")
|
||||
logger.error(f"标准错误: {stderr_output}")
|
||||
|
||||
if suppress_critical_dialog_on_stderr_match and \
|
||||
suppress_critical_dialog_on_stderr_match in stderr_output:
|
||||
logger.info(f"错误信息 '{stderr_output}' 匹配抑制条件，不显示关键错误对话框。")
|
||||
else:
|
||||
# 如果没有挂载点，或者挂载点是SWAP，则提示用户输入
|
||||
mount_point, ok = QInputDialog.getText(None, "挂载分区",
|
||||
f"请输入 {dev_path} 的挂载点 (例如: /mnt/data):",
|
||||
text=f"/mnt/{device_name}")
|
||||
if not ok or not mount_point:
|
||||
logger.info("用户取消了挂载操作或未提供挂载点。")
|
||||
return False
|
||||
|
||||
# 确保挂载点目录存在
|
||||
if not os.path.exists(mount_point):
|
||||
try:
|
||||
os.makedirs(mount_point, exist_ok=True)
|
||||
logger.info(f"创建挂载点目录: {mount_point}")
|
||||
except OSError as e:
|
||||
logger.error(f"创建挂载点目录失败 {mount_point}: {e}")
|
||||
QMessageBox.critical(None, "错误", f"创建挂载点目录失败: {e}")
|
||||
return False
|
||||
|
||||
try:
|
||||
stdout, stderr = self.system_manager._run_command(["mount", dev_path, mount_point], root_privilege=True)
|
||||
logger.info(f"成功挂载 {dev_path} 到 {mount_point}")
|
||||
QMessageBox.information(None, "成功", f"成功挂载 {dev_path} 到 {mount_point}")
|
||||
return True
|
||||
QMessageBox.critical(None, "错误", f"{error_message}\n错误详情: {stderr_output}")
|
||||
return False, e.stdout.strip(), stderr_output
|
||||
except FileNotFoundError:
|
||||
QMessageBox.critical(None, "错误", f"命令 '{command_list[0]}' 未找到。请确保已安装相关工具。")
|
||||
logger.error(f"命令 '{command_list[0]}' 未找到。")
|
||||
return False, "", f"命令 '{command_list[0]}' 未找到。"
|
||||
except Exception as e:
|
||||
logger.error(f"挂载 {dev_path} 失败: {e}")
|
||||
QMessageBox.critical(None, "错误", f"挂载 {dev_path} 失败: {e}")
|
||||
return False
|
||||
QMessageBox.critical(None, "错误", f"执行命令时发生未知错误: {e}")
|
||||
logger.error(f"执行命令 {full_cmd_str} 时发生未知错误: {e}")
|
||||
return False, "", str(e)
|
||||
|
||||
def unmount_partition(self, device_name):
|
||||
"""
|
||||
卸载指定的分区。
|
||||
"""
|
||||
dev_path = self._get_device_path(device_name)
|
||||
logger.info(f"尝试卸载设备: {dev_path}")
|
||||
def _get_fstab_path(self):
|
||||
return "/etc/fstab"
|
||||
|
||||
# 尝试从系统信息中获取当前挂载点
|
||||
current_mount_point = None
|
||||
devices = self.system_manager.get_block_devices()
|
||||
for dev in devices:
|
||||
if dev.get('name') == device_name:
|
||||
current_mount_point = dev.get('mountpoint')
|
||||
break
|
||||
if 'children' in dev:
|
||||
for child in dev['children']:
|
||||
if child.get('name') == device_name:
|
||||
current_mount_point = child.get('mountpoint')
|
||||
def create_partition(self, disk_path, partition_table_type, size_gb, total_disk_mib, use_max_space):
|
||||
"""
|
||||
在指定磁盘上创建分区。
|
||||
:param disk_path: 磁盘设备路径，例如 /dev/sdb。
|
||||
:param partition_table_type: 分区表类型，'gpt' 或 'msdos'。
|
||||
:param size_gb: 分区大小（GB）。
|
||||
:param total_disk_mib: 磁盘总大小 (MiB)。
|
||||
:param use_max_space: 是否使用最大可用空间创建分区。
|
||||
:return: 新创建的分区路径，如果失败则为 None。
|
||||
"""
|
||||
if not isinstance(disk_path, str) or not isinstance(partition_table_type, str):
|
||||
logger.error(f"尝试创建分区时传入无效的磁盘路径或分区表类型。磁盘: {disk_path} (类型: {type(disk_path)}), 类型: {partition_table_type} (类型: {type(partition_table_type)})")
|
||||
QMessageBox.critical(None, "错误", "无效的磁盘路径或分区表类型。")
|
||||
return None
|
||||
|
||||
logger.info(f"尝试在 {disk_path} 上创建 {size_gb}GB 的分区，分区表类型为 {partition_table_type}。")
|
||||
|
||||
# 1. 检查磁盘是否已经有分区表，如果没有则创建
|
||||
# parted -s /dev/sdb print
|
||||
try:
|
||||
stdout, _ = self.system_manager._run_command(["parted", "-s", disk_path, "print"], root_privilege=True)
|
||||
if "unrecognised disk label" in stdout:
|
||||
logger.info(f"磁盘 {disk_path} 没有分区表，将创建 {partition_table_type} 分区表。")
|
||||
success, _, _ = self._execute_shell_command(
|
||||
["parted", "-s", disk_path, "mklabel", partition_table_type],
|
||||
f"创建分区表 {partition_table_type} 失败"
|
||||
)
|
||||
if not success:
|
||||
return None
|
||||
else:
|
||||
logger.info(f"磁盘 {disk_path} 已有分区表。")
|
||||
except Exception as e:
|
||||
logger.error(f"检查或创建分区表失败: {e}")
|
||||
QMessageBox.critical(None, "错误", f"检查或创建分区表失败: {e}")
|
||||
return None
|
||||
|
||||
# 2. 获取下一个分区的起始扇区
|
||||
start_sector = self.get_disk_next_partition_start_sector(disk_path)
|
||||
if start_sector is None:
|
||||
QMessageBox.critical(None, "错误", f"无法确定磁盘 {disk_path} 的分区起始位置。")
|
||||
return None
|
||||
|
||||
# 3. 构建 parted 命令
|
||||
parted_cmd = ["parted", "-s", disk_path, "mkpart", "primary"]
|
||||
|
||||
# 对于 GPT，默认文件系统是 ext2。对于 msdos，默认是 ext2。
|
||||
# parted mkpart primary [fstype] start end
|
||||
# 我们可以省略 fstype，让用户稍后格式化。
|
||||
|
||||
# 计算结束位置
|
||||
start_mib = start_sector * 512 / (1024 * 1024) # 转换为 MiB
|
||||
|
||||
if use_max_space:
|
||||
end_mib = total_disk_mib # 使用磁盘总大小作为结束位置
|
||||
parted_cmd.extend([f"{start_mib}MiB", f"{end_mib}MiB"])
|
||||
else:
|
||||
end_mib = start_mib + size_gb * 1024 # 将 GB 转换为 MiB
|
||||
parted_cmd.extend([f"{start_mib}MiB", f"{end_mib}MiB"])
|
||||
|
||||
# 执行创建分区命令
|
||||
success, stdout, stderr = self._execute_shell_command(
|
||||
parted_cmd,
|
||||
f"在 {disk_path} 上创建分区失败"
|
||||
)
|
||||
if not success:
|
||||
return None
|
||||
|
||||
# 4. 刷新内核分区表
|
||||
self._execute_shell_command(["partprobe", disk_path], f"刷新 {disk_path} 分区表失败", root_privilege=True)
|
||||
|
||||
# 5. 获取新创建的分区路径
|
||||
# 通常新分区是磁盘的最后一个分区，例如 /dev/sdb1, /dev/sdb2 等
|
||||
# 我们可以通过 lsblk 再次获取信息，找到最新的分区
|
||||
try:
|
||||
# 重新获取设备信息，找到最新的分区
|
||||
devices = self.system_manager.get_block_devices()
|
||||
new_partition_path = None
|
||||
for dev in devices:
|
||||
if dev.get('path') == disk_path and dev.get('children'):
|
||||
# 找到最新的子分区
|
||||
latest_partition = None
|
||||
for child in dev['children']:
|
||||
if child.get('type') == 'part':
|
||||
if latest_partition is None or int(child.get('maj:min').split(':')[1]) > int(latest_partition.get('maj:min').split(':')[1]):
|
||||
latest_partition = child
|
||||
if latest_partition:
|
||||
new_partition_path = latest_partition.get('path')
|
||||
break
|
||||
if current_mount_point:
|
||||
break
|
||||
|
||||
if not current_mount_point or current_mount_point == '[SWAP]' or current_mount_point == '':
|
||||
logger.warning(f"设备 {dev_path} 未挂载或无法确定挂载点，无法卸载。")
|
||||
QMessageBox.warning(None, "警告", f"设备 {dev_path} 未挂载或无法确定挂载点，无法卸载。")
|
||||
return False
|
||||
if new_partition_path:
|
||||
logger.info(f"成功在 {disk_path} 上创建分区: {new_partition_path}")
|
||||
return new_partition_path
|
||||
else:
|
||||
logger.error(f"在 {disk_path} 上创建分区成功，但未能确定新分区的设备路径。")
|
||||
QMessageBox.warning(None, "警告", f"分区创建成功，但未能确定新分区的设备路径。请手动检查。")
|
||||
return None
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"获取新创建分区路径失败: {e}")
|
||||
QMessageBox.critical(None, "错误", f"获取新创建分区路径失败: {e}")
|
||||
return None
|
||||
|
||||
def get_disk_next_partition_start_sector(self, disk_path):
|
||||
"""
|
||||
获取磁盘上下一个可用的分区起始扇区。
|
||||
:param disk_path: 磁盘设备路径，例如 /dev/sdb。
|
||||
:return: 下一个分区起始扇区（整数），如果失败则为 None。
|
||||
"""
|
||||
if not isinstance(disk_path, str):
|
||||
logger.error(f"传入 get_disk_next_partition_start_sector 的 disk_path 不是字符串: {disk_path} (类型: {type(disk_path)})")
|
||||
return None
|
||||
|
||||
try:
|
||||
# 尝试通过挂载点卸载，通常更可靠
|
||||
stdout, stderr = self.system_manager._run_command(["umount", current_mount_point], root_privilege=True)
|
||||
logger.info(f"成功卸载 {current_mount_point} ({dev_path})")
|
||||
QMessageBox.information(None, "成功", f"成功卸载 {current_mount_point} ({dev_path})")
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error(f"卸载 {current_mount_point} ({dev_path}) 失败: {e}")
|
||||
QMessageBox.critical(None, "错误", f"卸载 {current_mount_point} ({dev_path}) 失败: {e}")
|
||||
return False
|
||||
stdout, _ = self.system_manager._run_command(["parted", "-s", disk_path, "unit", "s", "print", "free"], root_privilege=True)
|
||||
# 查找最大的空闲空间
|
||||
free_space_pattern = re.compile(r'^\s*\d+\s+([\d.]+s)\s+([\d.]+s)\s+([\d.]+s)\s+Free Space', re.MULTILINE)
|
||||
matches = free_space_pattern.findall(stdout)
|
||||
|
||||
def delete_partition(self, device_name):
|
||||
if matches:
|
||||
# 找到所有空闲空间的起始扇区，并取最小的作为下一个分区的起始
|
||||
# 或者，更准确地说，找到最后一个分区的结束扇区 + 1
|
||||
# 简化处理：如果没有分区，从 2048s 开始 (常见对齐)
|
||||
# 如果有分区，从最后一个分区的结束扇区 + 1 开始
|
||||
|
||||
# 先尝试获取所有分区信息
|
||||
stdout_partitions, _ = self.system_manager._run_command(["parted", "-s", disk_path, "unit", "s", "print"], root_privilege=True)
|
||||
partition_end_pattern = re.compile(r'^\s*\d+\s+[\d.]+s\s+([\d.]+s)', re.MULTILINE)
|
||||
partition_end_sectors = [int(float(s.replace('s', ''))) for s in partition_end_pattern.findall(stdout_partitions)]
|
||||
|
||||
if partition_end_sectors:
|
||||
# 最后一个分区的结束扇区 + 1
|
||||
last_end_sector = max(partition_end_sectors)
|
||||
# 确保对齐，通常是 1MiB (2048扇区) 或 4MiB (8192扇区)
|
||||
# 这里我们简单地取下一个 1MiB 对齐的扇区
|
||||
start_sector = (last_end_sector // 2048 + 1) * 2048
|
||||
logger.debug(f"磁盘 {disk_path} 最后一个分区结束于 {last_end_sector}s，下一个分区起始扇区建议为 {start_sector}s。")
|
||||
return start_sector
|
||||
else:
|
||||
# 没有分区，从 2048s 开始
|
||||
logger.debug(f"磁盘 {disk_path} 没有现有分区，建议起始扇区为 2048s。")
|
||||
return 2048 # 2048扇区是 1MiB，常见的起始对齐位置
|
||||
else:
|
||||
logger.warning(f"无法在磁盘 {disk_path} 上找到空闲空间信息。")
|
||||
return None
|
||||
except Exception as e:
|
||||
logger.error(f"获取磁盘 {disk_path} 下一个分区起始扇区失败: {e}")
|
||||
return None
|
||||
|
||||
def get_disk_next_partition_start_mib(self, disk_path):
|
||||
"""
|
||||
获取磁盘上下一个可用的分区起始位置 (MiB)。
|
||||
:param disk_path: 磁盘设备路径，例如 /dev/sdb。
|
||||
:return: 下一个分区起始位置 (MiB)，如果失败则为 None。
|
||||
"""
|
||||
start_sector = self.get_disk_next_partition_start_sector(disk_path)
|
||||
if start_sector is not None:
|
||||
return start_sector * 512 / (1024 * 1024) # 转换为 MiB
|
||||
return None
|
||||
|
||||
def delete_partition(self, device_path):
|
||||
"""
|
||||
删除指定的分区。
|
||||
此操作不可逆，会丢失数据。
|
||||
:param device_path: 要删除的分区路径，例如 /dev/sdb1。
|
||||
:return: True 如果删除成功，否则 False。
|
||||
"""
|
||||
dev_path = self._get_device_path(device_name)
|
||||
logger.info(f"尝试删除分区: {dev_path}")
|
||||
|
||||
# 安全检查: 确保是分区，而不是整个磁盘
|
||||
# 简单的检查方法是看设备名是否包含数字 (如 sda1, nvme0n1p1)
|
||||
if not any(char.isdigit() for char in device_name):
|
||||
QMessageBox.warning(None, "警告", f"{dev_path} 看起来不是一个分区。为安全起见，不执行删除操作。")
|
||||
logger.warning(f"尝试删除整个磁盘 {dev_path}，已阻止。")
|
||||
return False
|
||||
if not isinstance(device_path, str):
|
||||
logger.error(f"尝试删除非字符串设备路径: {device_path} (类型: {type(device_path)})")
|
||||
QMessageBox.critical(None, "错误", "无效的设备路径。")
|
||||
return False
|
||||
|
||||
reply = QMessageBox.question(None, "确认删除分区",
|
||||
f"你确定要删除分区 {dev_path} 吗？\n"
|
||||
"此操作不可逆，将导致数据丢失！",
|
||||
QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
|
||||
f"您确定要删除分区 {device_path} 吗？此操作不可逆！",
|
||||
QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
|
||||
if reply == QMessageBox.No:
|
||||
logger.info(f"用户取消了删除分区 {dev_path} 的操作。")
|
||||
logger.info(f"用户取消了删除分区 {device_path} 的操作。")
|
||||
return False
|
||||
|
||||
# 首先尝试卸载分区（如果已挂载）
|
||||
try:
|
||||
# umount 如果设备未挂载会返回非零退出码，所以这里 check_output=False
|
||||
self.system_manager._run_command(["umount", dev_path], root_privilege=True, check_output=False)
|
||||
logger.info(f"尝试卸载 {dev_path} (如果已挂载)。")
|
||||
except Exception as e:
|
||||
logger.warning(f"卸载 {dev_path} 失败或未挂载: {e}")
|
||||
# 继续执行，因为即使卸载失败也可能能删除
|
||||
logger.info(f"尝试删除分区: {device_path}")
|
||||
|
||||
# 使用 parted 来删除分区
|
||||
# parted 需要父磁盘设备名和分区号
|
||||
# 例如，对于 /dev/sda1，需要 /dev/sda 和分区号 1
|
||||
partition_number_str = ''.join(filter(str.isdigit, device_name)) # 从 sda1 提取 "1"
|
||||
if not partition_number_str:
|
||||
QMessageBox.critical(None, "错误", f"无法从 {device_name} 解析分区号。")
|
||||
logger.error(f"无法从 {device_name} 解析分区号。")
|
||||
# 1. 尝试卸载分区 (静默处理，如果未挂载则不报错)
|
||||
self.unmount_partition(device_path, show_dialog_on_error=False)
|
||||
|
||||
# 2. 从 fstab 中移除条目
|
||||
self._remove_fstab_entry(device_path)
|
||||
|
||||
# 3. 获取父磁盘和分区号
|
||||
match = re.match(r'(/dev/\w+)(\d+)', device_path)
|
||||
if not match:
|
||||
QMessageBox.critical(None, "错误", f"无法解析分区路径 {device_path}。")
|
||||
logger.error(f"无法解析分区路径 {device_path}。")
|
||||
return False
|
||||
|
||||
parent_disk_name = device_name.rstrip(partition_number_str) # 从 sda1 提取 "sda"
|
||||
parent_disk_path = self._get_device_path(parent_disk_name)
|
||||
disk_path = match.group(1)
|
||||
partition_number = match.group(2)
|
||||
|
||||
try:
|
||||
# parted -s /dev/sda rm 1
|
||||
stdout, stderr = self.system_manager._run_command(
|
||||
["parted", "-s", parent_disk_path, "rm", partition_number_str],
|
||||
root_privilege=True
|
||||
)
|
||||
logger.info(f"成功删除分区 {dev_path}")
|
||||
QMessageBox.information(None, "成功", f"成功删除分区 {dev_path}")
|
||||
# 4. 执行 parted 命令删除分区
|
||||
success, _, stderr = self._execute_shell_command(
|
||||
["parted", "-s", disk_path, "rm", partition_number],
|
||||
f"删除分区 {device_path} 失败"
|
||||
)
|
||||
if success:
|
||||
QMessageBox.information(None, "成功", f"分区 {device_path} 已成功删除。")
|
||||
# 刷新内核分区表
|
||||
self._execute_shell_command(["partprobe", disk_path], f"刷新 {disk_path} 分区表失败", root_privilege=True)
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error(f"删除分区 {dev_path} 失败: {e}")
|
||||
QMessageBox.critical(None, "错误", f"删除分区 {dev_path} 失败: {e}")
|
||||
else:
|
||||
return False
|
||||
|
||||
def format_partition(self, device_name, fstype=None):
|
||||
def format_partition(self, device_path, fstype=None):
|
||||
"""
|
||||
格式化指定的分区。
|
||||
此操作不可逆，会丢失数据。
|
||||
格式化指定分区。
|
||||
:param device_path: 要格式化的设备路径，例如 /dev/sdb1。
|
||||
:param fstype: 文件系统类型，例如 'ext4', 'xfs'。如果为 None，则弹出对话框让用户选择。
|
||||
:return: True 如果格式化成功，否则 False。
|
||||
"""
|
||||
dev_path = self._get_device_path(device_name)
|
||||
logger.info(f"尝试格式化分区: {dev_path}")
|
||||
if not isinstance(device_path, str):
|
||||
logger.error(f"尝试格式化非字符串设备路径: {device_path} (类型: {type(device_path)})")
|
||||
QMessageBox.critical(None, "错误", "无效的设备路径。")
|
||||
return False
|
||||
|
||||
# 安全检查: 确保是分区
|
||||
if not any(char.isdigit() for char in device_name):
|
||||
QMessageBox.warning(None, "警告", f"{dev_path} 看起来不是一个分区。为安全起见，不执行格式化操作。")
|
||||
logger.warning(f"尝试格式化整个磁盘 {dev_path}，已阻止。")
|
||||
return False
|
||||
# 1. 尝试卸载分区 (静默处理，如果未挂载则不报错)
|
||||
self.unmount_partition(device_path, show_dialog_on_error=False)
|
||||
|
||||
if not fstype:
|
||||
# 提示用户选择文件系统类型
|
||||
fstypes = ["ext4", "xfs", "fat32", "ntfs"] # 常用文件系统
|
||||
fstype, ok = QInputDialog.getItem(None, "格式化分区",
|
||||
f"请选择 {dev_path} 的文件系统类型:",
|
||||
fstypes, 0, False) # 默认选择 ext4
|
||||
if not ok or not fstype:
|
||||
logger.info("用户取消了格式化操作或未选择文件系统。")
|
||||
# 2. 从 fstab 中移除旧条目 (因为格式化后 UUID 会变)
|
||||
self._remove_fstab_entry(device_path)
|
||||
|
||||
if fstype is None:
|
||||
# 弹出对话框让用户选择文件系统类型
|
||||
items = ["ext4", "xfs", "fat32", "ntfs"] # 常用文件系统
|
||||
selected_fstype, ok = QInputDialog.getItem(None, "选择文件系统",
|
||||
f"请为 {device_path} 选择文件系统类型:",
|
||||
items, 0, False)
|
||||
if not ok or not selected_fstype:
|
||||
logger.info(f"用户取消了格式化 {device_path} 的操作。")
|
||||
return False
|
||||
fstype = selected_fstype
|
||||
|
||||
reply = QMessageBox.question(None, "确认格式化分区",
|
||||
f"你确定要格式化分区 {dev_path} 为 {fstype} 吗？\n"
|
||||
"此操作不可逆，将导致数据丢失！",
|
||||
QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
|
||||
f"您确定要格式化分区 {device_path} 为 {fstype} 吗？此操作将擦除所有数据！",
|
||||
QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
|
||||
if reply == QMessageBox.No:
|
||||
logger.info(f"用户取消了格式化分区 {dev_path} 的操作。")
|
||||
logger.info(f"用户取消了格式化分区 {device_path} 的操作。")
|
||||
return False
|
||||
|
||||
# 首先尝试卸载分区（如果已挂载）
|
||||
try:
|
||||
self.system_manager._run_command(["umount", dev_path], root_privilege=True, check_output=False)
|
||||
logger.info(f"尝试卸载 {dev_path} (如果已挂载)。")
|
||||
except Exception as e:
|
||||
logger.warning(f"卸载 {dev_path} 失败或未挂载: {e}")
|
||||
logger.info(f"尝试格式化设备 {device_path} 为 {fstype}。")
|
||||
|
||||
# 执行 mkfs 命令
|
||||
format_cmd = []
|
||||
if fstype == "ext4":
|
||||
format_cmd = ["mkfs.ext4", "-F", device_path]
|
||||
elif fstype == "xfs":
|
||||
format_cmd = ["mkfs.xfs", "-f", device_path]
|
||||
elif fstype == "fat32":
|
||||
format_cmd = ["mkfs.vfat", "-F", "32", device_path]
|
||||
elif fstype == "ntfs":
|
||||
format_cmd = ["mkfs.ntfs", "-f", device_path]
|
||||
else:
|
||||
QMessageBox.critical(None, "错误", f"不支持的文件系统类型: {fstype}")
|
||||
logger.error(f"不支持的文件系统类型: {fstype}")
|
||||
return False
|
||||
|
||||
success, _, stderr = self._execute_shell_command(
|
||||
format_cmd,
|
||||
f"格式化设备 {device_path} 为 {fstype} 失败"
|
||||
)
|
||||
if success:
|
||||
QMessageBox.information(None, "成功", f"设备 {device_path} 已成功格式化为 {fstype}。")
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
def mount_partition(self, device_path, mount_point, add_to_fstab=False):
|
||||
"""
|
||||
挂载一个分区。
|
||||
:param device_path: 要挂载的设备路径，例如 /dev/sdb1。
|
||||
:param mount_point: 挂载点，例如 /mnt/data。
|
||||
:param add_to_fstab: 是否将挂载信息添加到 /etc/fstab。
|
||||
:return: True 如果挂载成功，否则 False。
|
||||
"""
|
||||
if not isinstance(device_path, str) or not isinstance(mount_point, str):
|
||||
logger.error(f"尝试挂载时传入无效的设备路径或挂载点。设备: {device_path} (类型: {type(device_path)}), 挂载点: {mount_point} (类型: {type(mount_point)})")
|
||||
QMessageBox.critical(None, "错误", "无效的设备路径或挂载点。")
|
||||
return False
|
||||
|
||||
logger.info(f"尝试挂载设备 {device_path} 到 {mount_point}。")
|
||||
try:
|
||||
mkfs_cmd = []
|
||||
if fstype == "ext4":
|
||||
mkfs_cmd = ["mkfs.ext4", "-F", dev_path] # -F 强制执行
|
||||
elif fstype == "xfs":
|
||||
mkfs_cmd = ["mkfs.xfs", "-f", dev_path] # -f 强制执行
|
||||
elif fstype == "fat32":
|
||||
mkfs_cmd = ["mkfs.fat", "-F", "32", dev_path]
|
||||
elif fstype == "ntfs":
|
||||
mkfs_cmd = ["mkfs.ntfs", "-f", dev_path] # -f 强制执行
|
||||
# 确保挂载点存在
|
||||
if not os.path.exists(mount_point):
|
||||
logger.debug(f"创建挂载点目录: {mount_point}")
|
||||
os.makedirs(mount_point, exist_ok=True)
|
||||
|
||||
# 执行挂载命令
|
||||
success, _, stderr = self._execute_shell_command(
|
||||
["mount", device_path, mount_point],
|
||||
f"挂载设备 {device_path} 到 {mount_point} 失败"
|
||||
)
|
||||
|
||||
if success:
|
||||
QMessageBox.information(None, "成功", f"设备 {device_path} 已成功挂载到 {mount_point}。")
|
||||
if add_to_fstab:
|
||||
self._add_to_fstab(device_path, mount_point)
|
||||
return True
|
||||
else:
|
||||
raise ValueError(f"不支持的文件系统类型: {fstype}")
|
||||
return False
|
||||
except Exception as e:
|
||||
logger.error(f"挂载设备 {device_path} 时发生错误: {e}")
|
||||
QMessageBox.critical(None, "错误", f"挂载设备 {device_path} 失败。\n错误详情: {e}")
|
||||
return False
|
||||
|
||||
stdout, stderr = self.system_manager._run_command(mkfs_cmd, root_privilege=True)
|
||||
logger.info(f"成功格式化分区 {dev_path} 为 {fstype}")
|
||||
QMessageBox.information(None, "成功", f"成功格式化分区 {dev_path} 为 {fstype}")
|
||||
def unmount_partition(self, device_path, show_dialog_on_error=True):
|
||||
"""
|
||||
卸载一个分区。
|
||||
:param device_path: 要卸载的设备路径，例如 /dev/sdb1 或 /dev/md0。
|
||||
:param show_dialog_on_error: 是否在卸载失败时显示错误对话框。
|
||||
:return: True 如果卸载成功，否则 False。
|
||||
"""
|
||||
if not isinstance(device_path, str):
|
||||
logger.error(f"尝试卸载非字符串设备路径: {device_path} (类型: {type(device_path)})")
|
||||
if show_dialog_on_error:
|
||||
QMessageBox.critical(None, "错误", f"无效的设备路径: {device_path}")
|
||||
return False
|
||||
|
||||
logger.info(f"尝试卸载设备: {device_path}")
|
||||
try:
|
||||
# 尝试从 fstab 中移除条目，如果存在的话
|
||||
# 注意：这里先移除 fstab 条目，是为了防止卸载失败后 fstab 中仍然有无效条目
|
||||
# 但如果卸载成功，fstab 条目也需要移除。
|
||||
# 如果卸载失败，且 fstab 移除成功，用户需要手动处理。
|
||||
# 更好的做法可能是：卸载成功后，再移除 fstab 条目。
|
||||
# 但为了简化逻辑，这里先移除，并假设卸载会成功。
|
||||
# 实际上，_remove_fstab_entry 应该在卸载成功后才调用，或者在删除分区/格式化时调用。
|
||||
# 对于单纯的卸载操作，不应该自动移除 fstab 条目，除非用户明确要求。
|
||||
# 考虑到当前场景是“卸载后刷新”，通常用户只是想临时卸载，fstab条目不应被删除。
|
||||
# 所以这里暂时注释掉，或者只在删除分区/格式化时调用。
|
||||
# self._remove_fstab_entry(device_path)
|
||||
|
||||
# 执行卸载命令
|
||||
success, _, stderr = self._execute_shell_command(
|
||||
["umount", device_path],
|
||||
f"卸载设备 {device_path} 失败",
|
||||
suppress_critical_dialog_on_stderr_match="not mounted" if not show_dialog_on_error else None
|
||||
)
|
||||
if success:
|
||||
QMessageBox.information(None, "成功", f"设备 {device_path} 已成功卸载。")
|
||||
return success
|
||||
except Exception as e:
|
||||
logger.error(f"卸载设备 {device_path} 时发生错误: {e}")
|
||||
if show_dialog_on_error:
|
||||
QMessageBox.critical(None, "错误", f"卸载设备 {device_path} 失败。\n错误详情: {e}")
|
||||
return False
|
||||
|
||||
def _add_to_fstab(self, device_path, mount_point):
|
||||
"""
|
||||
将设备的挂载信息添加到 /etc/fstab。
|
||||
使用 UUID 方式添加。
|
||||
"""
|
||||
if not isinstance(device_path, str) or not isinstance(mount_point, str):
|
||||
logger.error(f"尝试添加到 fstab 时传入无效的设备路径或挂载点。设备: {device_path} (类型: {type(device_path)}), 挂载点: {mount_point} (类型: {type(mount_point)})")
|
||||
QMessageBox.critical(None, "错误", "无效的设备路径或挂载点，无法添加到 fstab。")
|
||||
return False
|
||||
|
||||
logger.info(f"尝试将设备 {device_path} (挂载点: {mount_point}) 添加到 /etc/fstab。")
|
||||
device_details = self.system_manager.get_device_details_by_path(device_path)
|
||||
uuid = device_details.get('uuid') if device_details else None
|
||||
fstype = device_details.get('fstype') if device_details else None
|
||||
|
||||
if not uuid or not fstype:
|
||||
logger.error(f"无法获取设备 {device_path} 的 UUID 或文件系统类型，无法添加到 fstab。UUID: {uuid}, FSTYPE: {fstype}")
|
||||
QMessageBox.critical(None, "错误", f"无法获取设备 {device_path} 的 UUID 或文件系统类型，无法添加到 /etc/fstab。")
|
||||
return False
|
||||
|
||||
fstab_entry = f"UUID={uuid} {mount_point} {fstype} defaults 0 2"
|
||||
fstab_path = self._get_fstab_path()
|
||||
|
||||
try:
|
||||
# 检查是否已存在相同的挂载点或 UUID 条目
|
||||
with open(fstab_path, 'r') as f:
|
||||
lines = f.readlines()
|
||||
|
||||
for line in lines:
|
||||
if f"UUID={uuid}" in line or mount_point in line.split():
|
||||
logger.warning(f"fstab 中已存在设备 {device_path} (UUID: {uuid}) 或挂载点 {mount_point} 的条目，跳过添加。")
|
||||
return True # 已经存在，认为成功
|
||||
|
||||
# 添加新条目
|
||||
with open(fstab_path, 'a') as f:
|
||||
f.write(fstab_entry + '\n')
|
||||
logger.info(f"成功将 '{fstab_entry}' 添加到 {fstab_path}。")
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error(f"格式化分区 {dev_path} 失败: {e}")
|
||||
QMessageBox.critical(None, "错误", f"格式化分区 {dev_path} 失败: {e}")
|
||||
logger.error(f"将设备 {device_path} 添加到 /etc/fstab 失败: {e}")
|
||||
QMessageBox.critical(None, "错误", f"将设备 {device_path} 添加到 /etc/fstab 失败。\n错误详情: {e}")
|
||||
return False
|
||||
|
||||
def _remove_fstab_entry(self, device_path):
|
||||
"""
|
||||
从 /etc/fstab 中移除指定设备的条目。
|
||||
:param device_path: 设备路径，例如 /dev/sdb1 或 /dev/md0。
|
||||
:return: True 如果成功移除或没有找到条目，否则 False。
|
||||
"""
|
||||
if not isinstance(device_path, str):
|
||||
logger.warning(f"尝试移除 fstab 条目时传入非字符串设备路径: {device_path} (类型: {type(device_path)})")
|
||||
return False
|
||||
|
||||
logger.info(f"尝试从 /etc/fstab 移除与 {device_path} 相关的条目。")
|
||||
fstab_path = self._get_fstab_path()
|
||||
try:
|
||||
device_details = self.system_manager.get_device_details_by_path(device_path)
|
||||
uuid = device_details.get('uuid') if device_details else None
|
||||
|
||||
logger.debug(f"尝试移除 fstab 条目，获取到的设备 {device_path} 的 UUID: {uuid}")
|
||||
|
||||
# 如果没有 UUID 且不是 /dev/ 路径，则无法可靠地从 fstab 中移除
|
||||
if not uuid and not device_path.startswith('/dev/'):
|
||||
logger.warning(f"无法获取设备 {device_path} 的 UUID，也无法识别为 /dev/ 路径，跳过 fstab 移除。")
|
||||
return False
|
||||
|
||||
with open(fstab_path, 'r') as f:
|
||||
lines = f.readlines()
|
||||
|
||||
new_lines = []
|
||||
removed_count = 0
|
||||
for line in lines:
|
||||
# 检查是否是注释或空行
|
||||
if line.strip().startswith('#') or not line.strip():
|
||||
new_lines.append(line)
|
||||
continue
|
||||
|
||||
is_match = False
|
||||
# 尝试匹配 UUID
|
||||
if uuid:
|
||||
if f"UUID={uuid}" in line:
|
||||
is_match = True
|
||||
# 尝试匹配设备路径 (如果 UUID 不存在或不匹配)
|
||||
if not is_match and device_path:
|
||||
# 分割行以避免部分匹配，例如 /dev/sda 匹配 /dev/sda1
|
||||
parts = line.split()
|
||||
if len(parts) > 0 and parts[0] == device_path:
|
||||
is_match = True
|
||||
|
||||
if is_match:
|
||||
logger.info(f"从 {fstab_path} 移除了条目: {line.strip()}")
|
||||
removed_count += 1
|
||||
else:
|
||||
new_lines.append(line)
|
||||
|
||||
if removed_count > 0:
|
||||
with open(fstab_path, 'w') as f:
|
||||
f.writelines(new_lines)
|
||||
logger.info(f"成功从 {fstab_path} 移除了 {removed_count} 个条目。")
|
||||
return True
|
||||
else:
|
||||
logger.info(f"在 {fstab_path} 中未找到与 {device_path} 相关的条目。")
|
||||
return True # 没有找到也算成功，因为目标是确保没有该条目
|
||||
except FileNotFoundError:
|
||||
logger.warning(f"fstab 文件 {fstab_path} 不存在。")
|
||||
return True
|
||||
except Exception as e:
|
||||
logger.error(f"从 /etc/fstab 移除条目时发生错误: {e}")
|
||||
QMessageBox.critical(None, "错误", f"从 /etc/fstab 移除条目失败。\n错误详情: {e}")
|
||||
return False
|
||||
|
||||
|
||||
Reference in New Issue
Block a user