Python实现TCP文件传输

废话少说，直接上代码

# 发送端
import socket,os
from struct import pack

def send_file(file_name,file_socket:socket.socket):
    try:
        f = open(file_name,'rb')
        size = os.path.getsize(file_name)
        if size < 1024:
            read_size = 500
        elif size < 1024*1024 and size >= 1024:
            read_size = 500*1024
        else:
            read_size = 500*1024*1024
        file_socket.send(pack('q',size))
        del size
        file_socket.recv(1024)
        while True:
            data = f.read(read_size)
            if not data:
                break
            file_socket.send(data)
        f.close()
    except FileNotFoundError:
        print(f'没有找到{file_name}')

if __name__ == '__main__':
    try:
        file_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
        file_socket.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,True)
        ip = str(input('请输入接收端IP：'))
        port = int(input('请输入接收端端口号：'))
        file_socket.connect((ip,port))
        print('连接成功，准备开始传输。。。')
        path = r'%s' % (input('请输入文件所在文件夹：') + '/')
        print(file_socket.recv(1024).decode('gbk'))
        file_name = str(input())
        file_socket.send(file_name.encode('gbk'))
        send_file(path + file_name,file_socket)
    except ConnectionResetError:
        print('接收端断开连接')

# 接收端
import socket,os
from time import time
from struct import unpack
from tqdm import tqdm

DOWNLOAD_PATH = 'D:/Python文件传输/' # 传输目录

def download(file_name,file_socket:socket.socket):
    file_size = unpack('q',file_socket.recv(1024))[0]
    if file_size < 1024:
        print(f'文件大小：{file_size} B')
    elif file_size < 1024*1024 and file_size >= 1024:
        print(f'文件大小：{file_size/1024} KB')
    else:
        print(f'文件大小：{file_size/(1024*1024)} MB')
    f = open(file_name,'wb')
    print('开始传输...')
    download_size = 2048
    file_socket.send('开始传输'.encode('gbk'))
    start = time()
    for i in tqdm(range(int(file_size/download_size) + 1)):
        data = file_socket.recv(download_size)
        f.write(data)
    end = time()
    f.close()
    print('传输完成！')
    print(f'大约耗时{end - start}秒')

if __name__ == '__main__':
    os.chdir(DOWNLOAD_PATH)
    try:
        file_socket = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
        file_socket.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,True)
        file_socket_port = int(input('请输入端口号：'))
        file_socket.bind(('',file_socket_port))
        print('成功启动，等待连接。。。')
        file_socket.listen(1)
        f_socket, f_addr = file_socket.accept()
        print(f'建立连接{f_addr}')
        f_socket.send('请输入文件名'.encode('gbk'))
        file_name = f_socket.recv(1024)
        download(file_name.decode('gbk'),f_socket)
        f_socket.close()
        file_socket.close()
    except ConnectionResetError:
        print('发送端已断开连接')
    except UnicodeDecodeError:
        print('文件编码错误，请检查文件格式是否正确')