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('文件編碼錯誤，請檢查文件格式是否正確')