下载oss2上面的屏幕截图文件的代码

import os
import re
import oss2
import tkinter as tk
from tkinter import filedialog, messagebox
from tkinter.scrolledtext import ScrolledText
import pickle
from tkinter import filedialog
import subprocess


def on_entry_focus_in(event):
    # 当Entry获得焦点时，检查是否包含默认文本并删除它
    if entry_date.get() == "例:2024-01-01":
        entry_date.delete(0, tk.END)
        # 如果需要，也可以在这里将前景色改回黑色或其他颜色


def on_entry_focus_out(event):
    # 当Entry失去焦点时，如果它是空的，则添加默认文本
    if not entry_date.get():
        entry_date.insert(0, "2024-01-01")


def create_dir_if_not_exists(path):
    if not os.path.exists(path):
        os.makedirs(path)


def mac_to_name(file_name, mac_to_name_dict):
    mac_address_pattern = re.compile(r'([0-9a-fA-F]{2}[_-]){5}[0-9a-fA-F]{2}')
    match = mac_address_pattern.search(file_name)
    if match:
        mac_address = match.group(0)
        if mac_address in mac_to_name_dict:
            return file_name.replace(mac_address, mac_to_name_dict[mac_address])
    return file_name


def download_matching_files(bucket, file_name_part, download_path, mac_to_name_dict, log_text):
    file_count = 0
    for obj in oss2.ObjectIterator(bucket):
        if file_name_part in obj.key:
            file_name = mac_to_name(obj.key, mac_to_name_dict)
            file_path = os.path.join(download_path, file_name)
            create_dir_if_not_exists(os.path.dirname(file_path))  # 确保目录存在
            try:
                bucket.get_object_to_file(obj.key, file_path)
                log_text.insert(tk.END, f'下载 {obj.key} 到 {file_path}\n')
                log_text.see(tk.END)  # 自动滚动到最后一行
                log_text.update()
                file_count += 1
            except Exception as e:
                log_text.insert(tk.END, f'下载 {obj.key} 失败: {str(e)}\n')
                log_text.see(tk.END)  # 自动滚动到最后一行
                log_text.update()
    return file_count


def save_mac_to_name_dict(mac_to_name_dict, filename='mac_to_name_dict.pkl'):
    with open(filename, 'wb') as f:
        pickle.dump(mac_to_name_dict, f)


def load_mac_to_name_dict(filename='mac_to_name_dict.pkl'):
    if os.path.exists(filename):
        with open(filename, 'rb') as f:
            return pickle.load(f)
    return {}


def start_download():
    file_name_part = entry_date.get()
    if not file_name_part:
        messagebox.showerror("错误", "请输入文件名中包含的日期")
        return

    download_path = entry_download_path.get()
    if not download_path:
        messagebox.showerror("错误", "请选择保存下载文件的路径")
        return

    mac_to_name_dict = load_mac_to_name_dict()

    auth = oss2.Auth('123', '456')
    bucket = oss2.Bucket(auth, 'http://oss-cn-hangzhou.aliyuncs.com', 'singsongpic')

    log_text.insert(tk.END, "开始下载文件...\n")
    file_count = download_matching_files(bucket, file_name_part, download_path, mac_to_name_dict, log_text)
    messagebox.showinfo("完成", f"总共下载的文件数：{file_count}")
    log_text.insert(tk.END, f"下载完成，总共下载的文件数：{file_count}\n")


# def add_mac_to_name():
#     mac = entry_mac.get()
#     name = entry_name.get()
#     if not mac or not name:
#         messagebox.showerror("错误", "请输入MAC地址和对应的姓名")
#         return
#
#     mac_to_name_dict = load_mac_to_name_dict()
#     mac_to_name_dict[mac] = name
#     save_mac_to_name_dict(mac_to_name_dict)
#     messagebox.showinfo("完成", "MAC地址和姓名已添加")
def open_download_folder():
    """打开下载文件夹"""
    # 确保entry_download_path中有有效的路径
    if entry_download_path.get():
        path = entry_download_path.get()
        if os.path.isdir(path):
            # 使用subprocess调用系统默认的文件夹打开程序
            if os.name == 'nt':  # Windows系统
                subprocess.Popen(['explorer', path])
            else:  # Linux或macOS系统
                subprocess.Popen(['xdg-open', path])
        else:
            print("路径不是有效的文件夹")
    else:
        print("请先选择下载路径")

def choose_directory():
    download_path = filedialog.askdirectory()
    if download_path:
        entry_download_path.delete(0, tk.END)
        entry_download_path.insert(0, download_path)


# GUI部分
root = tk.Tk()
root.title("OSS屏幕文件下载器-SINGSONG专用")

tk.Label(root, text="文件名中包含的日期:").grid(row=0, column=0)
entry_date = tk.Entry(root, foreground='dim gray')  # 使用浅灰色作为前景色
entry_date.insert(0, "例:2024-01-01")  # 插入默认文本
entry_date.grid(row=0, column=1)

# 绑定FocusIn事件以便在用户点击时删除默认文本
entry_date.bind("<FocusIn>", on_entry_focus_in)

tk.Label(root, text="下载文件保存路径:").grid(row=1, column=0)
entry_download_path = tk.Entry(root)
entry_download_path.grid(row=1, column=1)
tk.Button(root, text="选择下载路径", command=choose_directory).grid(row=1, column=2)

tk.Button(root, text="下载文件", command=start_download).grid(row=2, column=0, columnspan=3)
# # 创建打开下载文件夹按钮
# tk.Button(root, text="打开下载文件夹", command=open_download_folder).grid(row=2, column=2)
# #
# tk.Label(root, text="MAC地址:").grid(row=3, column=0)
# entry_mac = tk.Entry(root)
# entry_mac.grid(row=3, column=1)

# tk.Label(root, text="对应的姓名:").grid(row=4, column=0)
# entry_name = tk.Entry(root)
# entry_name.grid(row=4, column=1)

# tk.Button(root, text="添加MAC地址和姓名", command=add_mac_to_name).grid(row=5, column=0, columnspan=3)

# 添加用于显示下载信息的文本框
log_text = ScrolledText(root, wrap=tk.WORD, width=50, height=10)
log_text.grid(row=6, column=0, columnspan=3)

root.mainloop()