實(shí)現(xiàn)一個(gè)tcp文件下載器

服務(wù)器端的實(shí)現(xiàn)

from socket import *
import sys


def get_file_content(file_name):
    """獲取文件的內(nèi)容"""
    try:
        with open(file_name, "rb") as f:
            content = f.read()
        return content
    except:
        print("沒(méi)有下載的文件:%s" % file_name)


def main():

    if len(sys.argv) != 2:
        print("請(qǐng)按照如下方式運(yùn)行:python3 xxx.py 7890")
        return
    else:
        # 運(yùn)行方式為python3 xxx.py 7890
        port = int(sys.argv[1])


    # 創(chuàng)建socket
    tcp_server_socket = socket(AF_INET, SOCK_STREAM)
    # 本地信息
    address = ('', port)
    # 綁定本地信息
    tcp_server_socket.bind(address)
    # 將主動(dòng)套接字變?yōu)楸粍?dòng)套接字
    tcp_server_socket.listen(128)

    while True:
        # 等待客戶端的鏈接,即為這個(gè)客戶端發(fā)送文件
        client_socket, clientAddr = tcp_server_socket.accept()
        # 接收對(duì)方發(fā)送過(guò)來(lái)的數(shù)據(jù)
        recv_data = client_socket.recv(1024)  # 接收1024個(gè)字節(jié)
        file_name = recv_data.decode("utf-8")
        print("對(duì)方請(qǐng)求下載的文件名為:%s" % file_name)
        file_content = get_file_content(file_name)
        # 發(fā)送文件的數(shù)據(jù)給客戶端
        # 因?yàn)楂@取打開(kāi)文件時(shí)是以rb方式打開(kāi),所以file_content中的數(shù)據(jù)已經(jīng)是二進(jìn)制的格式,因此不需要encode編碼
        if file_content:
            client_socket.send(file_content)
        # 關(guān)閉這個(gè)套接字
        client_socket.close()

    # 關(guān)閉監(jiān)聽(tīng)套接字
    tcp_server_socket.close()


if __name__ == "__main__":
    main()

在終端開(kāi)始服務(wù)器:


55.jpg

客戶端的實(shí)現(xiàn)

from socket import *


def main():

    # 創(chuàng)建socket
    tcp_client_socket = socket(AF_INET, SOCK_STREAM)

    # 目的信息
    server_ip = input("請(qǐng)輸入服務(wù)器ip:")
    server_port = int(input("請(qǐng)輸入服務(wù)器port:"))

    # 鏈接服務(wù)器
    tcp_client_socket.connect((server_ip, server_port))

    # 輸入需要下載的文件名
    file_name = input("請(qǐng)輸入要下載的文件名:")

    # 發(fā)送文件下載請(qǐng)求
    tcp_client_socket.send(file_name.encode("utf-8"))

    # 接收對(duì)方發(fā)送過(guò)來(lái)的數(shù)據(jù),最大接收1024個(gè)字節(jié)(1K)
    recv_data = tcp_client_socket.recv(1024)
    # print('接收到的數(shù)據(jù)為:', recv_data.decode('utf-8'))
    # 如果接收到數(shù)據(jù)再創(chuàng)建文件,否則不創(chuàng)建
    if recv_data:
        with open("[接收]"+file_name, "wb") as f:
            f.write(recv_data)

    # 關(guān)閉套接字
    tcp_client_socket.close()


if __name__ == "__main__":
    main()

運(yùn)行結(jié)果:


66.jpg
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容