104 lines
3.2 KiB
Python
104 lines
3.2 KiB
Python
|
import os
|
||
|
|
import tkinter as tk
|
||
|
|
from tkinter import filedialog
|
||
|
|
from tkinter import ttk
|
||
|
|
from zipfile import ZipFile
|
||
|
|
|
||
|
|
selected_items = []
|
||
|
|
|
||
|
|
def toggle_selection(event):
|
||
|
|
item = tree.item(tree.focus())
|
||
|
|
file_path = item['text']
|
||
|
|
if item['tags'] == ('file',):
|
||
|
|
if file_path in selected_items:
|
||
|
|
selected_items.remove(file_path)
|
||
|
|
else:
|
||
|
|
selected_items.append(file_path)
|
||
|
|
update_selection_label()
|
||
|
|
|
||
|
|
def select_items():
|
||
|
|
if selected_items:
|
||
|
|
zip_file = 'selected_items.zip'
|
||
|
|
with ZipFile(zip_file, 'w') as zf:
|
||
|
|
for item in selected_items:
|
||
|
|
arcname = os.path.relpath(item, target_path)
|
||
|
|
if os.path.isfile(item):
|
||
|
|
zf.write(item, arcname)
|
||
|
|
elif os.path.isdir(item):
|
||
|
|
for root, dirs, files in os.walk(item):
|
||
|
|
for file in files:
|
||
|
|
file_path = os.path.join(root, file)
|
||
|
|
arcname = os.path.relpath(file_path, target_path)
|
||
|
|
zf.write(file_path, arcname)
|
||
|
|
result_label.config(text=f'文件已打包为 {zip_file}')
|
||
|
|
selected_items.clear()
|
||
|
|
update_selection_label()
|
||
|
|
|
||
|
|
def update_selection_label():
|
||
|
|
selection_label.config(text=f'已选中 {len(selected_items)} 个文件/文件夹')
|
||
|
|
|
||
|
|
def build_tree(parent, directory):
|
||
|
|
for item in os.listdir(directory):
|
||
|
|
item_path = os.path.join(directory, item)
|
||
|
|
if os.path.isdir(item_path):
|
||
|
|
item_node = tree.insert(parent, 'end', text=item, open=False, tags=('folder',))
|
||
|
|
build_tree(item_node, item_path)
|
||
|
|
else:
|
||
|
|
tree.insert(parent, 'end', text=item, tags=('file',))
|
||
|
|
|
||
|
|
def toggle_checkbox(event):
|
||
|
|
item_id = tree.identify_row(event.y)
|
||
|
|
if item_id:
|
||
|
|
is_checked = tree.tag_has('checked', item_id)
|
||
|
|
if is_checked:
|
||
|
|
tree.tag_remove('checked', item_id)
|
||
|
|
else:
|
||
|
|
tree.tag_add('checked', item_id)
|
||
|
|
|
||
|
|
# 创建主窗口
|
||
|
|
root = tk.Tk()
|
||
|
|
root.title("文件选择")
|
||
|
|
root.geometry("600x400")
|
||
|
|
|
||
|
|
# 计算窗口居中的位置
|
||
|
|
screen_width = root.winfo_screenwidth()
|
||
|
|
screen_height = root.winfo_screenheight()
|
||
|
|
window_width = 600
|
||
|
|
window_height = 400
|
||
|
|
x = (screen_width - window_width) // 2
|
||
|
|
y = (screen_height - window_height) // 2
|
||
|
|
root.geometry(f"{window_width}x{window_height}+{x}+{y}")
|
||
|
|
|
||
|
|
# 选择目标文件夹
|
||
|
|
target_path = os.path.join(os.getcwd(), 'target/classes/')
|
||
|
|
|
||
|
|
# 创建选择按钮
|
||
|
|
select_button = tk.Button(root, text='打包选中的文件或文件夹', command=select_items)
|
||
|
|
select_button.pack(pady=10)
|
||
|
|
|
||
|
|
# 创建目录树
|
||
|
|
tree = ttk.Treeview(root, selectmode='none')
|
||
|
|
tree.pack(fill='both', expand=True)
|
||
|
|
|
||
|
|
# 创建滚动条
|
||
|
|
scrollbar = ttk.Scrollbar(root, orient='vertical', command=tree.yview)
|
||
|
|
scrollbar.pack(side='right', fill='y')
|
||
|
|
tree.configure(yscrollcommand=scrollbar.set)
|
||
|
|
|
||
|
|
# 添加当前目录下的文件和文件夹到目录树
|
||
|
|
build_tree('', target_path)
|
||
|
|
|
||
|
|
# 创建结果标签
|
||
|
|
result_label = tk.Label(root, text='')
|
||
|
|
result_label.pack()
|
||
|
|
|
||
|
|
# 创建已选中文件数标签
|
||
|
|
selection_label = tk.Label(root, text='')
|
||
|
|
selection_label.pack()
|
||
|
|
|
||
|
|
# 绑定鼠标点击事件
|
||
|
|
tree.bind('<Button-1>', toggle_checkbox)
|
||
|
|
|
||
|
|
# 启动主循环
|
||
|
|
root.mainloop()
|