實現(xiàn)的效果如下的Gif所示，就是網(wǎng)絡(luò)通信Socket模塊實現(xiàn)文件下載傳輸。

服務(wù)端

首先需要獲取本機(jī)ip，這里服務(wù)端采用多線程的方法，就是定義一個函數(shù)，然后用threading創(chuàng)建任務(wù)。客戶端連接成功，接收客戶端的請求信息，就是下載的文件名。所以需要判斷，有輸出文件字節(jié)數(shù)。然后在問用戶是不是要下載，得到信息就使用 while True: 讀文件的內(nèi)容，再一個send。看代碼是不是就是這么回事。

import socket 
import os 
import threading 
 
# 獲取本機(jī)ip 
def get_host_ip(): 
    try: 
        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) 
        s.connect(('8.8.8.8', 80)) 
        ip = s.getsockname()[0] 
    finally: 
        s.close() 
 
    return ip 
 
# 處理客戶端請求下載文件的操作（從主線程提出來的代碼） 
def deal_client_request(ip_port, service_client_socket): 
    # 連接成功后，輸出“客戶端連接成功”和客戶端的ip和端口 
    print("客戶端連接成功", ip_port) 
    # 接收客戶端的請求信息【recv】 
    file_name = service_client_socket.recv(1024) 
    # 解碼 
    file_name_data = file_name.decode("utf-8") 
    # 判斷文件是否存在 
    if os.path.exists(file_name_data): 
        #輸出文件字節(jié)數(shù) 
        fsize = os.path.getsize(file_name_data) 
        #轉(zhuǎn)化為兆單位 
        fmb = fsize/float(1024*1024) 
        #要傳輸?shù)奈募畔?nbsp;
        senddata = "文件名：%s  文件大小：%.2fMB"%(file_name_data,fmb) 
        #發(fā)送和打印文件信息【send】 
        service_client_socket.send(senddata.encode("utf-8")) 
        print("請求文件名：%s  文件大小：%.2f MB"%(file_name_data,fmb)) 
        #接受客戶是否需要下載【recv】 
        options = service_client_socket.recv(1024) 
        if options.decode("utf-8") == "y": 
            # 打開文件 
            with open(file_name_data, "rb") as f: 
                # 計算總數(shù)據(jù)包數(shù)目 
                nums = fsize/1024 
                # 當(dāng)前傳輸?shù)臄?shù)據(jù)包數(shù)目 
                cnum = 0 
 
                while True: 
                    file_data = f.read(1024) 
                    cnum = cnum + 1 
                    #progress = cnum/nums*100 
 
                    #print("當(dāng)前已下載：%.2f%%"%progress,end = "\r") 
                    if file_data: 
                        # 只要讀取到數(shù)據(jù)，就向客戶端進(jìn)行發(fā)送【send】 
                        service_client_socket.send(file_data) 
                    # 數(shù)據(jù)讀完，退出循環(huán) 
                    else: 
                        print("請求的文件數(shù)據(jù)發(fā)送完成") 
                        break 
        else: 
            print("下載取消！") 
    else: 
        print("下載的文件不存在！") 
    # 關(guān)閉服務(wù)當(dāng)前客戶端的套接字【close】 
    service_client_socket.close() 
 
 
if __name__ == '__main__': 
    # 獲取本機(jī)ip 
    print("TCP文件傳輸服務(wù)器，本機(jī)IP:" + get_host_ip()) 
     
    # 把工作目錄切換到data目錄下 
    os.chdir("./data") 
    # 創(chuàng)建套接字【socket】 
    tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
    # 綁定端口號【bind】 
    tcp_server_socket.bind(("", 3356)) 
    # 設(shè)置監(jiān)聽，將主動套接字變?yōu)楸粍犹捉幼帧緇isten】 
    tcp_server_socket.listen(128) 
 
    # 循環(huán)調(diào)用【accept】，可以支持多個客戶端同時連接，和多個客戶端同時下載文件 
    while True: 
        service_client_socket, ip_port = tcp_server_socket.accept() 
        # 連接成功后打印套接字號 
        #print(id(service_client_socket)) 
 
        # 創(chuàng)建子線程 
        sub_thread = threading.Thread(target=deal_client_request, args=(ip_port, service_client_socket)) 
        # 啟動子線程 
        sub_thread.start() 

客戶端

客戶端更簡單，連接服務(wù)端，發(fā)送下載文件的請求，定義一個寫入的文件夾，就是小兒科東西。不寫了，看代碼。

# -*- coding:utf-8 -*- 
# 多任務(wù)文件下載器客戶端 
import socket 
import os 
 
if __name__ == '__main__': 
    # 創(chuàng)建套接字【socket】 
    tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
    # 和服務(wù)端連接【connect】 
    server_ip = input("輸入服務(wù)器IP：") 
    tcp_client_socket.connect((server_ip, 3356)) 
    # 發(fā)送下載文件的請求 
    file_name = input("請輸入要下載的文件名：") 
    # 編碼 
    file_name_data = file_name.encode("utf-8") 
    # 發(fā)送文件下載請求數(shù)據(jù)【send】 
    tcp_client_socket.send(file_name_data) 
    # 接收要下載的文件信息【recv】 
    file_info = tcp_client_socket.recv(1024) 
    # 文件信息解碼 
    info_decode = file_info.decode("utf-8") 
    print(info_decode) 
    #獲取文件大小 
    fileszie = float(info_decode.split('：')[2].split('MB')[0]) 
    fileszie2 = fileszie*1024 
    # 是否下載？輸入ｙ 確認(rèn) 輸入ｑ 取消 
    opts = input("是否下載？(y 確認(rèn) q 取消)") 
    if opts == 'q': 
        print("下載取消！程序退出") 
    else: 
        print("正在下載 >>>>>>") 
        #向服務(wù)器確認(rèn)正在下載【send】 
        tcp_client_socket.send(b'y') 
 
        recvpath = "./receive/" 
        if not os.path.exists(recvpath): 
            os.mkdir(recvpath)  
         
        # 把數(shù)據(jù)寫入到文件里 
        with open(recvpath + file_name, "wb") as file: 
            #目前接收到的數(shù)據(jù)包數(shù)目 
            cnum = 0 
 
            while True: 
                # 循環(huán)接收文件數(shù)據(jù)【recv】 
                file_data = tcp_client_socket.recv(1024) 
                # 接收到數(shù)據(jù) 
                if file_data: 
                    # 寫入數(shù)據(jù) 
                    file.write(file_data) 
                    cnum = cnum+1 
                    #progress =cnum/fileszie2*100 
                    #print("當(dāng)前已下載：%.2f%%"%progress,end = "\r") 
                # 接收完成 
                else: 
                    print("下載結(jié)束！") 
                    break 
    # 關(guān)閉套接字【close】 
    tcp_client_socket.close() 

擴(kuò)展

如果手機(jī)想要下載電腦上的文件(電腦和手機(jī)之間)，需要確保電腦和手機(jī)連接的網(wǎng)絡(luò)在同一個路由器下

通過ipconfig查看ip

命令行輸入python -m http.server 8888來搭建局域網(wǎng)

手機(jī)訪問192.168.0.101:8888即可下載文件

本文已收錄 GitHub，傳送門~[1] ，里面更有大廠面試完整考點，歡迎 Star。

Reference

[1]傳送門~:

https://github.com/MaoliRUNsen/runsenlearnpy100