ahdoawhfo/SpaceSniffer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request '新增了重新选择目录扫描功能，新增了进度条显示' (#2) from dev into main

Browse Source 
Reviewed-on: #2
This commit is contained in:
ahdoawhfo 2024-06-23 23:23:23 +08:00
commit cad17f9bdf
1 changed files with 88 additions and 142 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

228
SpaceSniffer.py
View File

@ -1,100 +1,74 @@
import os
import threading
import tkinter as tk
from tkinter import ttk, messagebox
import tkinter.filedialog
from tkinter import ttk, filedialog, messagebox
from queue import Queue
from send2trash import send2trash
import subprocess
import send2trash
class FolderScanner(threading.Thread):
def __init__(self, start_path, queue):
"""
初始化函数，创建一个线程并设置起始路径和队列。
Args:
start_path (str): 起始路径，表示要遍历的文件夹的根目录。
queue (Queue): 队列对象，用于线程间的通信，存放待处理或已处理过的文件或文件夹路径。
Returns:
None
"""
# 使用super()调用父类threading.Thread的__init__方法
super().__init__()
def __init__(self, start_path, queue, progress_queue):
threading.Thread.__init__(self)
self.start_path = start_path
self.queue = queue
self.progress_queue = progress_queue
self.total_items = 0
def run(self):
"""
运行扫描文件夹并获取文件信息的操作。
Args:
无参数。
Returns:
无返回值。
"""
self.total_items = sum(
[len(files) + len(dirs) for r, dirs, files in os.walk(self.start_path)]
)
self.scan_folder(self.start_path)
self.queue.put(None) # Indicate that scanning is done
def scan_folder(self, folder):
"""
遍历指定文件夹下的所有子文件夹和文件，并将它们的路径、大小、父文件夹路径和类型（文件夹或文件）放入队列中。
Args:
folder (str): 需要遍历的文件夹路径。
Returns:
None.
"""
scanned_items = 0
for dirpath, dirnames, filenames in os.walk(folder):
for dirname in dirnames:
folder_path = os.path.join(dirpath, dirname)
size = self.get_size(folder_path)
self.queue.put((folder_path, size, dirpath, "folder"))
scanned_items += 1
self.update_progress(scanned_items, folder_path)
for filename in filenames:
file_path = os.path.join(dirpath, filename)
size = os.path.getsize(file_path)
self.queue.put((file_path, size, dirpath, "file"))
scanned_items += 1
self.update_progress(scanned_items, file_path)
def get_size(self, path):
# 初始化总大小为0
total_size = 0
# 遍历指定路径下的所有文件和目录
for dirpath, dirnames, filenames in os.walk(path):
# 遍历当前目录下的所有文件
for f in filenames:
# 拼接文件的完整路径
fp = os.path.join(dirpath, f)
# 将文件大小累加到总大小中
total_size += os.path.getsize(fp)
return total_size
def update_progress(self, scanned_items, current_path):
progress = (scanned_items / self.total_items) * 100
self.progress_queue.put(
(progress, scanned_items, self.total_items, current_path)
)
class App(tk.Tk):
def __init__(self, start_path):
"""
初始化目录扫描器窗口
Args:
start_path (str): 开始扫描的目录路径
Returns:
None
"""
super().__init__()
self.withdraw()
self.title("目录扫描器 V1.0")
self.start_path = start_path
self.queue = Queue()
self.scanner = FolderScanner(start_path, self.queue)
self.progress_queue = Queue()
self.progress_window = ProgressWindow(self)
self.scanner = FolderScanner(start_path, self.queue, self.progress_queue)
self.scanner.start()
self.style = ttk.Style(self)
self.style.configure(
"Treeview", font=("Helvetica", 10), rowheight=25
) # 设置行高
self.style.configure("Treeview", font=("Helvetica", 10), rowheight=25)
self.style.configure("Treeview.Heading", font=("Helvetica", 12, "bold"))
self.tree = ttk.Treeview(self, style="Treeview")
@ -134,21 +108,12 @@ class App(tk.Tk):
self.populate_root()
self.update_tree()
self.update_progress()
self.protocol("WM_DELETE_WINDOW", self.on_close)
self.set_window_size()
def populate_root(self):
"""
在树形控件中填充根节点。
Args:
无参数。
Returns:
无返回值。
"""
size = self.scanner.get_size(self.start_path) / 1024 / 1024
self.tree.insert(
"",
@ -160,19 +125,15 @@ class App(tk.Tk):
)
def update_tree(self):
"""
更新树形结构，将磁盘文件信息更新到树形视图中。
Args:
无
Returns:
无
"""
if not self.is_sorting:
while not self.queue.empty():
path, size, parent, tag = self.queue.get()
item = self.queue.get()
if item is None: # Scanning is done
self.progress_window.destroy()
self.deiconify()
return
path, size, parent, tag = item
size_mb = size / 1024 / 1024
parent_iid = self.get_iid(parent)
if parent == self.start_path:
@ -188,29 +149,26 @@ class App(tk.Tk):
)
self.after(100, self.update_tree)
def update_progress(self):
if not self.progress_window.winfo_exists(): # Check if the window still exists
return
while not self.progress_queue.empty():
progress, scanned_items, total_items, current_path = (
self.progress_queue.get()
)
self.progress_window.progress_var.set(progress)
self.progress_window.progress_bar["value"] = progress
self.progress_window.progress_label.config(
text=f"扫描进度: {scanned_items} / {total_items}"
)
self.progress_window.path_label.config(text=f"正在扫描: {current_path}")
self.after(100, self.update_progress)
def get_iid(self, path):
"""
获取指定路径的iid值
Args:
path (str): 文件或目录的路径
Returns:
str: 路径本身作为iid值返回
"""
return path
def sort_by_size(self):
"""
根据文件或文件夹的大小对文件列表进行排序。
Args:
无参数。
Returns:
无返回值，直接修改当前实例的文件列表。
"""
self.sort_items(
lambda x: (
0 if x[1] == "folder" and self.folder_first_var.get() else 1,
@ -225,16 +183,6 @@ class App(tk.Tk):
self.size_sort_order = not self.size_sort_order
def sort_by_name(self):
"""
根据名称对列表进行排序
Args:
无
Returns:
无返回值，但会改变对象的内部状态
"""
self.sort_items(
lambda x: (
0 if x[1] == "folder" and self.folder_first_var.get() else 1,
@ -245,17 +193,6 @@ class App(tk.Tk):
self.name_sort_order = not self.name_sort_order
def sort_items(self, key, reverse):
"""
对树形控件中的指定目录进行排序。
Args:
key (callable): 用于排序的key函数。
reverse (bool): 是否降序排序。
Returns:
None
"""
if self.folder_first_var.get():
selected = self.tree.selection()
if not selected:
@ -316,16 +253,6 @@ class App(tk.Tk):
self.tree.move(item[2], parent, index)
def show_context_menu(self, event):
"""
显示上下文菜单
Args:
event (tkinter.Event): 鼠标事件对象
Returns:
None
"""
item = self.tree.identify_row(event.y)
if item:
self.tree.selection_set(item)
@ -363,13 +290,11 @@ class App(tk.Tk):
def delete_item(self, item):
path = self.tree.item(item, "text").split(" ")[0]
full_path = os.path.normpath(
os.path.join(self.start_path, path)
) # 使用os.path.normpath来规范化路径
full_path = os.path.normpath(os.path.join(self.start_path, path))
confirm = messagebox.askyesno("提示", f"你确定要删除 {full_path} 吗?")
if confirm:
try:
send2trash.send2trash(full_path)
send2trash(full_path)
self.tree.delete(item)
except Exception as e:
messagebox.showerror("错误", f"无法删除 {full_path}\n{e}")
@ -379,12 +304,18 @@ class App(tk.Tk):
for item in self.tree.get_children():
self.tree.delete(item)
self.populate_root()
self.scanner = FolderScanner(self.start_path, self.queue)
self.scanner = FolderScanner(self.start_path, self.queue, self.progress_queue)
self.progress_window = ProgressWindow(self) # Create a new progress window
self.withdraw() # Hide the main window
self.scanner.start()
self.update_tree() # Start updating the tree again
self.update_progress() # Start updating the progress again
def on_close(self):
if self.scanner.is_alive():
self.scanner.join()
self.destroy()
root.destroy()
def set_window_size(self):
screen_width = self.winfo_screenwidth()
@ -409,23 +340,38 @@ class App(tk.Tk):
font=(current_font.split()[0], new_font_size + 2, "bold"),
)
def sort_root_items(self, by="size"):
if by == "size":
self.sort_by_size()
elif by == "name":
self.sort_by_name()
def update_folder_first_var(self):
if self.folder_first_var.get():
self.folder_first_var.set(False)
else:
self.folder_first_var.set(True)
self.folder_first_var.set(not self.folder_first_var.get())
class ProgressWindow(tk.Toplevel):
def __init__(self, parent):
super().__init__(parent)
self.title("扫描进度")
self.geometry("500x200")
self.label = tk.Label(
self, text="扫描中，请稍等", fg="deepskyblue", font=("Helvetica", 16)
)
self.label.pack(pady=20)
self.progress_var = tk.DoubleVar()
self.progress_bar = ttk.Progressbar(
self, variable=self.progress_var, maximum=100
)
self.progress_bar.pack(fill=tk.X, padx=20, pady=10)
self.progress_label = tk.Label(self, text="扫描进度: 0 / 0")
self.progress_label.pack(pady=5)
self.path_label = tk.Label(self, text="正在扫描: ")
self.path_label.pack(pady=5)
self.protocol("WM_DELETE_WINDOW", self.on_close)
def on_close(self):
pass # Prevent the user from closing the progress window
if __name__ == "__main__":
root = tk.Tk()
root.withdraw()
start_path = tkinter.filedialog.askdirectory()
start_path = filedialog.askdirectory()
if start_path:
app = App(start_path)
app.mainloop()