用python的socket写一个局域网传输文件的程序

局域网传输文件是最最常用的功能,我参考https://www.jb51.net/python/345837qrz.htm这篇文章,复制粘贴,开发了一个。但发现进度条没有用,也没有显示传输用时和传输速度的功能,于是我改写了代码,使它实现这个功能。

python 复制代码
import socket
import os

def start_server(host='192.168.1.145', port=8888):
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.bind((host, port))
    server_socket.listen(1)
    print(f"服务器启动,监听 {host}:{port}")

    while True:
        client_socket, addr = server_socket.accept()
        print(f"连接来自 {addr}")

        # 接收文件名和大小
        file_info = client_socket.recv(1024).decode()
        file_name, file_size = file_info.split('|')
        file_size = int(file_size)

        # 创建文件并写入数据
        with open('d:/socketfile/'+file_name, 'wb') as file:
            received = 0
            while received < file_size:
                data = client_socket.recv(4096)
                file.write(data)
                received += len(data)
                print(f"接收进度: {received}/{file_size} 字节")

        client_socket.close()
        print(f"文件 {file_name} 接收完成")

if __name__ == "__main__":
    start_server()

上面是服务器端,没有怎么改,增加了'd:/socketfile/'+file_name,把文件放在我们想要放的文件夹。

python 复制代码
# -*- coding: utf-8 -*-
"""
Created on Mon Sep  8 14:31:28 2025

@author: YBK
"""

import socket
import os
import tkinter as tk
from tkinter import filedialog, messagebox, scrolledtext
import threading
from tkinter import ttk
import time

class FileTransferGUI:
    def __init__(self, master):
        self.master = master
        master.title("局域网文件传输工具")
        master.geometry("500x400")

        # 输入服务器IP
        tk.Label(master, text="服务器IP:").pack(pady=5)
        self.ip_entry = tk.Entry(master, width=30)
        self.ip_entry.pack(pady=5)
        self.ip_entry.insert(0, "192.168.1.145")  # 默认IP,根据实际修改
        
        #发生使用时间、传输速度
        self.usetime = 0
        self.sudu = 0

        # 文件选择按钮
        tk.Button(master, text="选择文件", command=self.select_file).pack(pady=10)
        self.file_path = ""

        # 发送按钮
        tk.Button(master, text="发送文件", command=self.start_send).pack(pady=10)

        # 进度显示
        self.progress_var = tk.DoubleVar()
        self.progress_bar = ttk.Progressbar(master, variable=self.progress_var, maximum=100)
        self.progress_bar.pack(fill=tk.X, padx=20, pady=10)

        # 日志区域
        self.log_area = scrolledtext.ScrolledText(master, height=10)
        self.log_area.pack(fill=tk.BOTH, expand=True, padx=10, pady=10)
        self.log_area.config(state=tk.DISABLED)
        
    def send_file(self,server_ip, file_path, port=8888):        
        client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        try:
            client_socket.connect((server_ip, port))
            file_name = os.path.basename(file_path)
            file_size = os.path.getsize(file_path)

            # 发送文件信息
            client_socket.send(f"{file_name}|{file_size}".encode())

            # 分块发送文件数据
            with open(file_path, 'rb') as file:
                sent = 0
                start = time.time()
                while sent < file_size:
                    data = file.read(4096)
                    client_socket.send(data)
                    sent += len(data)
                    # print(f"发送进度: {sent}/{file_size} 字节")
                    self.progress_var.set(int(sent/file_size*100))
            self.usetime = time.time() - start
            self.sudu = file_size / 1024 / 1024 / self.usetime
            print(f"文件发送成功,耗时{self.usetime:.2f}秒,速度:{self.sudu:.2f}m/s")
        except Exception as e:
            print(f"错误: {e}")
        finally:
            client_socket.close()

    def select_file(self):
        self.file_path = filedialog.askopenfilename()
        if self.file_path:
            self.log(f"已选择文件: {self.file_path}")

    def start_send(self):
        server_ip = self.ip_entry.get()
        if not server_ip or not self.file_path:
            messagebox.showerror("错误", "请输入IP并选择文件")
            return

        # 在新线程中发送文件,避免GUI冻结
        threading.Thread(target=self.send_file_thread, args=(server_ip, self.file_path)).start()

    def send_file_thread(self, server_ip, file_path):
        try:
            self.log("开始发送文件...")
            self.send_file(server_ip, file_path)  # 调用步骤2的发送函数
            self.progress_var.set(100)
            self.log(f"文件发送完成!耗时{self.usetime:.4f}秒,速度:{self.sudu:.2f}m/s")
        except Exception as e:
            self.log(f"错误: {e}")

    def log(self, message):
        self.log_area.config(state=tk.NORMAL)
        self.log_area.insert(tk.END, message + "\n")
        self.log_area.config(state=tk.DISABLED)
        self.log_area.yview(tk.END)

if __name__ == "__main__":
    root = tk.Tk()
    app = FileTransferGUI(root)
    root.mainloop()

以上就改动比较大,也让我慢慢熟悉面向对象编程,用self.可以为公用变量,增加时间和速度的显示,原来代码出现tk.Progressbar报错,进度条无法正常显示等问题。

速度还是可以的。

用一段时间,感觉还是不爽,要开2个程序,一个是服务器端,一个是客户端,才能在2台电脑中互传文件,为此,我把它一体化,只要用一个程序就实现了服务器端和客户端的功能,又能接受文件,又能发送文件。同时,加上我一直喜欢用的拖放功能,不能每次去选择文件传输。

python 复制代码
# -*- coding: utf-8 -*-
"""
Created on Mon Sep  8 14:31:28 2025

@author: YBK
"""

import socket
import os
import tkinter as tk
from tkinter import filedialog, messagebox, scrolledtext
import threading
from tkinter import ttk
import time
import windnd
from tkinter import LabelFrame
import chardet

def start_server(host='192.168.1.140', port=8888):
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.bind((host, port))
    server_socket.listen(1)
    print(f"服务器启动,监听 {host}:{port}")

    while True:
        client_socket, addr = server_socket.accept()
        print(f"连接来自 {addr}")

        # 接收文件名和大小
        raw_data = client_socket.recv(1024)
        encoding = chardet.detect(raw_data)['encoding']
        file_info = raw_data.decode(encoding)
        file_name, file_size = file_info.split('|')
        file_size = int(file_size)

        # 创建文件并写入数据
        with open('e:/socketfile/'+file_name, 'wb') as file:
            received = 0
            while received < file_size:
                data = client_socket.recv(4096)
                file.write(data)
                received += len(data)
                print(f"接收进度: {received}/{file_size} 字节")

        client_socket.close()
        print(f"文件 {file_name} 接收完成")

class FileTransferGUI:
    def __init__(self, master):
        self.master = master
        master.title("局域网文件传输工具")
        master.geometry("500x600")

        # 输入服务器IP
        tk.Label(master, text="服务器IP:").pack(pady=5)
        self.ip_entry = tk.Entry(master, width=30)
        self.ip_entry.pack(pady=5)
        self.ip_entry.insert(0, "192.168.1.145")  # 默认IP,根据实际修改
        
        tk.Label(master, text="本机IP(作为服务器):").pack(pady=5)
        self.bip_entry = tk.Entry(master, width=30)
        self.bip_entry.pack(pady=5)
        self.bip_entry.insert(0, "192.168.1.140")  # 默认IP,根据实际修改
        
        # 启动服务器按钮
        self.bt_fwq = tk.Button(master, text="启动服务器", command=self.start_fwq)
        self.bt_fwq.pack(pady=10)
        
        #发生使用时间
        self.usetime = 0
        self.sudu = 0

        # 文件选择按钮
        tk.Button(master, text="选择文件", command=self.select_file).pack(pady=10)
        self.file_path = ""
        
        # 文件拖放区域
        self.f_frame = LabelFrame(master, text="也可以拖放文件到此", width=250)
        self.f_frame.pack(expand=True, padx=10)
        self.label1 = tk.Label(self.f_frame,
                          wraplength=580,
                          text="")
        self.label1.pack(anchor="w", pady=5)
        windnd.hook_dropfiles(self.f_frame, func=self.handle_file)

        # 发送按钮
        tk.Button(master, text="发送文件", command=self.start_send).pack(pady=10)

        # 进度显示
        self.progress_var = tk.DoubleVar()
        self.progress_bar = ttk.Progressbar(master, variable=self.progress_var, maximum=100)
        self.progress_bar.pack(fill=tk.X, padx=20, pady=10)

        # 日志区域
        self.log_area = scrolledtext.ScrolledText(master, height=10)
        self.log_area.pack(fill=tk.BOTH, expand=True, padx=10, pady=10)
        self.log_area.config(state=tk.DISABLED)
        
    def start_server(self,host='192.168.1.140', port=8888):
        server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        server_socket.bind((host, port))
        server_socket.listen(1)
        print(f"服务器启动,监听 {host}:{port}")

        while True:
            client_socket, addr = server_socket.accept()
            print(f"连接来自 {addr}")

            # 接收文件名和大小
            raw_data = client_socket.recv(1024)
            encoding = chardet.detect(raw_data)['encoding']
            file_info = raw_data.decode(encoding)
            file_name, file_size = file_info.split('|')
            file_size = int(file_size)

            # 创建文件并写入数据
            with open('e:/socketfile/'+file_name, 'wb') as file:
                received = 0
                while received < file_size:
                    data = client_socket.recv(4096)
                    file.write(data)
                    received += len(data)
                    print(f"接收进度: {received}/{file_size} 字节")

            client_socket.close()
            print(f"文件 {file_name} 接收完成")
            self.log(f"文件 {file_name} 接收完成")
        
    def handle_file(self,files):
        if len(files) > 0:
            path = files[0].decode('gbk').replace('\\', '/')
            self.file_path = path
        if self.file_path:
            self.log(f"已选择文件: {self.file_path}")
    def start_fwq(self):
        fwq_ip = self.bip_entry.get()
        if not fwq_ip:
            messagebox.showerror("错误", "请输入IP并选择文件")
            return
        # print(fwq_ip)
        # 在新线程中发送文件,避免GUI冻结
        threading.Thread(target=self.start_server, args=(fwq_ip, 8888)).start()
        self.bt_fwq.config(state="disabled", bg="lightgray")
        
        
        
    def send_file(self,server_ip, file_path, port=8888):        
        client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        try:
            client_socket.connect((server_ip, port))
            file_name = os.path.basename(file_path)
            file_size = os.path.getsize(file_path)

            # 发送文件信息
            client_socket.send(f"{file_name}|{file_size}".encode())

            # 分块发送文件数据
            with open(file_path, 'rb') as file:
                sent = 0
                start = time.time()
                while sent < file_size:
                    data = file.read(4096)
                    client_socket.send(data)
                    sent += len(data)
                    # print(f"发送进度: {sent}/{file_size} 字节")
                    self.progress_var.set(int(sent/file_size*100))
            self.usetime = time.time() - start
            self.sudu = file_size / 1024 / 1024 / self.usetime
            print(f"文件发送成功,耗时{self.usetime:.2f}秒,速度:{self.sudu:.2f}m/s")
        except Exception as e:
            print(f"错误: {e}")
        finally:
            client_socket.close()

    def select_file(self):
        self.file_path = filedialog.askopenfilename()
        if self.file_path:
            self.log(f"已选择文件: {self.file_path}")

    def start_send(self):
        server_ip = self.ip_entry.get()
        if not server_ip or not self.file_path:
            messagebox.showerror("错误", "请输入IP并选择文件")
            return

        # 在新线程中发送文件,避免GUI冻结
        threading.Thread(target=self.send_file_thread, args=(server_ip, self.file_path)).start()

    def send_file_thread(self, server_ip, file_path):
        try:
            self.log("开始发送文件...")
            self.send_file(server_ip, file_path)  # 调用步骤2的发送函数
            self.progress_var.set(100)
            self.log(f"文件发送完成!耗时{self.usetime:.4f}秒,速度:{self.sudu:.2f}m/s")
        except Exception as e:
            self.log(f"错误: {e}")

    def log(self, message):
        self.log_area.config(state=tk.NORMAL)
        self.log_area.insert(tk.END, message + "\n")
        self.log_area.config(state=tk.DISABLED)
        self.log_area.yview(tk.END)

if __name__ == "__main__":
    root = tk.Tk()
    app = FileTransferGUI(root)
    root.mainloop()

填好ip,点击启动服务器就可以接收文件。下面还能显示接收到的文件。

相关推荐
土星云SaturnCloud4 小时前
边缘侧大模型部署的新利器——国科环宇GK 300I大模型一体机深度评测与架构解析
服务器·人工智能·ai·边缘计算
_Jimmy_4 小时前
Agent 溯源精度提升方案
人工智能·python·langchain
晚风吹长发5 小时前
Docker基础——Docker Network(网络详解)
linux·运维·网络·ubuntu·docker·容器
Cx330_FCQ5 小时前
Tmux使用
服务器·git·算法
hz567895 小时前
音视频SDK核心能力分析:低延迟传输与高并发接入方案
网络·安全·音视频·实时音视频·信息与通信
q567315235 小时前
企业级 HTTP 代理采购选型:技术评估清单 15 项
开发语言·网络·爬虫·网络协议·http·隧道ip·代理ip
白嫖一茶5 小时前
HTTP 状态码详解:五大类与常见状态码解析
网络·网络协议·http
SamChan905 小时前
在Web应用中集成PDF多语言翻译功能:PDFTranslator API实战指南
前端·python·ai·pdf·yapi·机器翻译
天天进步20156 小时前
Python全栈项目--智能办公自动化系统
开发语言·python
高磊20056 小时前
LVS(Linux virual server)
linux·服务器·lvs