多文件下载

# server.py 服务端
import os
import socketserver

class Server(socketserver.BaseRequestHandler):

    server_file_path = "F:/patch"

    def get_server_files(self):
        for root, dirs, files in os.walk(self.server_file_path):
            return files

    def get_file_context(self, file_name):
        try:
            with open(self.server_file_path + "/" + file_name, "rb") as file:
                data = file.read()
            return data
        except:
            return "请求文件不存在"

    def handle(self):
        print("Connection: ", self.client_address)
        file_string = ",".join(self.get_server_files())

        while True:
            message = self.request.recv(1024)
            if not message:
                break
            # 动作判断
            if message.decode() == "__file_list__":
                # 发送文件列表
                self.request.send(bytes(file_string, "UTF-8"))
            else:
                # 发送文件数据
                data = self.get_file_context(message.decode())
                self.request.send(data)


if __name__ == '__main__':
    host, port = "127.0.0.1", 8080
    server = socketserver.ThreadingTCPServer((host, port), Server)
    server.serve_forever()

# SocketDownload.py 下载线程
import socket
import threading

class SocketDownload(threading.Thread):

    server_ip = "127.0.0.1"
    server_port = 8080

    def __init__(self, file_name, file_save_path):
        threading.Thread.__init__(self)
        self.file_name = file_name
        self.file_save_path = file_save_path

    def run(self):
        address = (self.server_ip, self.server_port)
        sk = socket.socket()
        sk.connect(address)
        while True:
            sk.send(bytes(self.file_name, "utf-8"))
            msg = sk.recv(1024)
            if msg:
                with open(self.file_save_path + "/" + self.file_name, "wb") as f:
                    f.write(msg)
            print(self.file_name + ": 下载成功，保存路径：" + self.file_save_path)
            break
        sk.close()

# DownloadSurface 客户端界面
import socket
import tkinter as tk
from tkinter.filedialog import *

from MutilFile.client.SocketDownload import SocketDownload


class DownloadSurface(tk.Frame):
    def __init__(self, master):
        tk.Frame.__init__(self, master=None)
        self.grid()
        self.create_widgets()

    def create_widgets(self):
        # row 0
        self.lbIpPort = tk.Label(self, text="IP & 端口")
        self.lbIpPort.grid(row=0, column=0)

        self.defaultIpStringVar = StringVar()
        self.defaultIpStringVar.set("127.0.0.1")
        self.entryIp = tk.Entry(self, textvariable=self.defaultIpStringVar)
        self.entryIp.grid(row=0, column=1)

        self.defaultPortStringVar = StringVar()
        self.defaultPortStringVar.set("8080")
        self.entryPort = tk.Entry(self, textvariable=self.defaultPortStringVar)
        self.entryPort.grid(row=0, column=2)

        self.btnConn = tk.Button(self, text="连接服务器", command=self.connect_server)
        self.btnConn.grid(row=0, column=3)

        # row 1
        self.lbSavePath = tk.Label(self, text="保存路径")
        self.lbSavePath.grid(row=1, column=0)

        self.savePathStringVar = StringVar()
        self.savePathStringVar.set("F:/patch")
        self.entrySavePath = tk.Entry(self, textvariable=self.savePathStringVar, width=40)
        self.entrySavePath.grid(row=1, column=1, columnspan=2)

        self.btnSelSavePath = tk.Button(self, text="选择文件夹", command=self.select_save_path)
        self.btnSelSavePath.grid(row=1, column=3)

        # row 2
        self.lbServerFile = tk.Label(self, text="服务端文件列表")
        self.lbServerFile.grid(row=2, column=1)

        self.listboxServerFile = tk.Listbox(self,  width=10, height=6)
        self.listboxServerFile.grid(row=3, column=1, rowspan=5)

        self.btnToRight = tk.Button(self, text="   >   ", command=self.add_to_download)
        self.btnToRight.grid(row=8, column=1)

        self.btnToLeft = tk.Button(self, text="   <   ", command=self.cancel_download)
        self.btnToLeft.grid(row=8, column=2)

        self.listboxDownloadFile = tk.Listbox(self, width=10, height=6)
        self.listboxDownloadFile.grid(row=3, column=2, rowspan=5)

        self.lbDownloadFile = tk.Label(self, text="文件下载列表")
        self.lbDownloadFile.grid(row=2, column=2)

        # row 3
        self.btnConn = tk.Button(self, text="开始下载", command=self.download)
        self.btnConn.grid(row=2, column=3)

    def connect_server(self):
        self.listboxDownloadFile.delete(0, END)
        self.listboxServerFile.delete(0, END)

        ip_port = (self.entryIp.get(), int(self.entryPort.get()))
        print(ip_port)
        sk = socket.socket()
        sk.connect(ip_port)
        sk.send(bytes("__file_list__", "utf-8"))
        msg = str(sk.recv(1024), "UTF-8").split(",")
        for item in msg:
            self.listboxServerFile.insert(tk.END, item)
        sk.close()

    def select_save_path(self):
        file = askdirectory(title="选择下载文件保存路径")
        self.savePathStringVar.set(file)

    def add_to_download(self):
        for item in self.listboxServerFile.curselection():
            self.listboxDownloadFile.insert(tk.END, self.listboxServerFile.get(item))
            self.listboxServerFile.delete(item)

    def cancel_download(self):
        for item in self.listboxDownloadFile.curselection():
            self.listboxServerFile.insert(tk.END, self.listboxDownloadFile.get(item))
            self.listboxDownloadFile.delete(item)

    def download(self):
        print("开始下载")
        fileCount = self.listboxDownloadFile.size()
        thread = []
        for i in range(0, fileCount):
            print(self.listboxDownloadFile.get(i))
            print(self.entrySavePath.get())
            thread.append(SocketDownload(self.listboxDownloadFile.get(i), self.entrySavePath.get()))
        # 开启多线程
        for i in range(0, len(thread)):
            thread[i].start()


def init_window(root):
    sw = root.winfo_screenwidth()
    # 得到屏幕宽度
    sh = root.winfo_screenheight()
    # 得到屏幕高度
    ww = 480
    wh = 450
    # 窗口宽高为100
    x = (sw - ww) / 2
    y = (sh - wh) / 2
    root.geometry("%dx%d+%d+%d" % (ww, wh, x - 200, y))

root = tk.Tk()
root.title("多文件下载器")
init_window(root)
app = DownloadSurface(master=root)
app.mainloop()