基于Python+PyQt5开发的Windows磁盘分析与清理工具

class ScanWorker(threading.Thread):
    """扫描工作线程类"""
    def __init__(self, task_queue, result_queue, size_threshold, stop_event):
        super().__init__()
        self.task_queue = task_queue
        self.result_queue = result_queue
        self.size_threshold = size_threshold
        self.stop_event = stop_event
​
    def run(self):
        while not self.stop_event.is_set():
            try:
                dirpath, dirnames, filenames = self.task_queue.get(timeout=1)
            except:
                break
            dir_size = 0
            big_files = []
            for f in filenames:
                try:
                    fp = os.path.join(dirpath, f)
                    if os.path.isfile(fp):
                        st = os.stat(fp)
                        size = st.st_size
                        dir_size += size
                        if size >= self.size_threshold:
                            big_files.append((fp, size, ...))
                except (PermissionError, OSError):
                    continue
            self.result_queue.put((dirpath, dir_size, big_files))
            self.task_queue.task_done()

def setup_flow_progress_style(self):
    """设置流光进度条样式"""
    self.flow_gradient_pos = 0
    self.flow_timer = QTimer(self)
    self.flow_timer.timeout.connect(self.update_flow_animation)
    base_style = """
    QProgressBar {
        border: 1px solid #666666;
        border-radius: 4px;
        background: qlineargradient(x1:0, y1:0, x2:0, y2:1, 
            stop:0 #4a4a4a, stop:0.5 #3d3d3d, stop:1 #333333);
        text-align: center;
        color: #00ffff;
        font-weight: bold;
        font-size: 12px;
    }
    QProgressBar::chunk {
        border-radius: 3px;
        margin: 1px;
    }
    """
    self.progress_bar.setStyleSheet(base_style)
​
def update_flow_animation(self):
    """更新流光动画"""
    self.flow_gradient_pos = (self.flow_gradient_pos + 1) % 100
    p = self.flow_gradient_pos / 100.0
    style = f"""
    QProgressBar::chunk {{
        background: qlineargradient(x1:0, y1:0, x2:1, y2:0, 
            stop:{(p - 0.3) % 1} transparent, 
            stop:{(p - 0.2) % 1} rgba(0, 200, 255, 0.3), 
            stop:{(p - 0.1) % 1} rgba(0, 220, 255, 0.6), 
            stop:{p % 1} rgba(0, 255, 255, 1), 
            stop:{(p + 0.1) % 1} rgba(0, 220, 255, 0.6), 
            stop:{(p + 0.2) % 1} rgba(0, 200, 255, 0.3), 
            stop:{(p + 0.3) % 1} transparent);
    }}
    """
    self.progress_bar.setStyleSheet(style)

from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure
​
class MainWindow(QMainWindow):
    def __init__(self):
        ...
        self.fig = Figure(figsize=(5, 5))
        self.canvas = FigureCanvas(self.fig)
        self.left_layout.addWidget(self.canvas)
        self.canvas.mpl_connect('button_press_event', self.on_pie_click)
    def on_pie_click(self, event):
        """处理饼图双击事件"""
        if event.dblclick and hasattr(self, 'wedges'):
            for i, wedge in enumerate(self.wedges):
                if wedge.contains(event)[0]:
                    label = self.pie_labels[i]
                    if label == '其他目录':
                        return
                    new_path = os.path.join(self.current_path, label)
                    if os.path.isdir(new_path):
                        self.path_history.append(self.current_path)
                        self.current_path = new_path
                        self.update_pie_chart()
                    break

import ctypes
import sys
​
def is_admin():
    """检查当前进程是否以管理员身份运行"""
    try:
        return ctypes.windll.shell32.IsUserAnAdmin()
    except:
        return False
​
def run_as_admin():
    """以管理员身份重新启动程序"""
    if sys.platform == 'win32':
        script = os.path.abspath(sys.argv[0])
        params = ' '.join(['"' + arg + '"' for arg in sys.argv[1:]])
        ctypes.windll.shell32.ShellExecuteW(
            None, "runas", sys.executable, 
            f'"{script}" {params}', None, 1
        )
​
if __name__ == "__main__":
    if not is_admin():
        run_as_admin()
        sys.exit(0)
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())

import matplotlib.font_manager as fm
​
def setup_matplotlib_font():
    """设置matplotlib中文字体"""
    available_fonts = [f.name for f in fm.fontManager.ttflist]
    preferred_fonts = [
        'SimHei', 'Microsoft YaHei', 'SimSun', 'KaiTi',
        'Heiti SC', 'STHeiti'
    ]
    selected_font = None
    for font in preferred_fonts:
        if font in available_fonts:
            selected_font = font
            break
    if selected_font:
        matplotlib.rcParams['font.sans-serif'] = [selected_font]
        matplotlib.rcParams['font.family'] = 'sans-serif'
    matplotlib.rcParams['axes.unicode_minus'] = False
​
setup_matplotlib_font()

if sys.platform == 'win32':
    ctypes.windll.kernel32.SetConsoleCP(65001)
    ctypes.windll.kernel32.SetConsoleOutputCP(65001)
os.environ['PYTHONUTF8'] = '1'

cd src
python main.py