Merge pull request '新增了重新选择目录扫描功能，新增了进度条显示' (#2) from dev into main
Reviewed-on: #2
This commit is contained in:
commit
cad17f9bdf
230
SpaceSniffer.py
230
SpaceSniffer.py
|
|
@ -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):
|
||||
self.scanner.join()
|
||||
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()
|
||||
|
|
|
|||
Loading…
Reference in New Issue