函数封装文件传输功能

服务端完整代码

import socket
import subprocess
import struct
import json
import os

share_dir = r"E:\mydesktop\My_Code\后端\b站网络编程\05文件传输\优化版本\server\share"

def get(conn,cmds):
    filename = cmds[1]

    # 3、以读的方式打开文件，读取文件内容发送给客户端
    # 第一步：制作固定长度的报头。
    header_dic = {
        "filename": filename,
        "md5": "xxxdxxx",
        'file_size': os.path.getsize(r"%s\%s" % (share_dir, filename))
    }
    header_json = json.dumps(header_dic)
    header_bytes = header_json.encode('utf-8')

    # 第二步：先发送报头的长度
    conn.send(struct.pack("i", len(header_bytes)))

    # 第三步：再发报头
    conn.send(header_bytes)

    # 第四步：再发送真实的数据
    with open(("%s\%s" % (share_dir, filename)), "rb") as f:
        # conn.send(f.read())
        for line in f:
            conn.send(line)  # 逐行读取再发送，有效节约系统内存，但是会粘包

def put(conn,cmds):
    pass

def run():
    phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    phone.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
    phone.bind(("127.0.0.1",9910))
    phone.listen(5)
    print("starting...")

    while True:
        conn,client_addr = phone.accept()
        print(client_addr)
        while True:
            # 1、收命令
            res = conn.recv(8096)
            if not res:break

            # 2、解析命令，提取相应命令参数
            cmds = res.decode("utf-8").split() #得到['get','a.txt']
            if cmds[0] == "get":
                get(conn,cmds)
            elif cmds[0] == "put":
                put(conn,cmds)

        conn.close()
    phone.close()

if __name__ == '__main__':
    run()

客户端完整代码

import socket
import struct
import json

download_dir = r"E:\mydesktop\My_Code\后端\b站网络编程\05文件传输\优化版本\client\Download"

def get(phone,cmds):
    # 2、以写的方式打开一个新的文件，接收服务端发来的文件的内容写入客户的新文件
    # 第一步：先收报头的长度
    obj = phone.recv(4)
    header_size = struct.unpack('i', obj)[0]

    # 第二步：再收报头
    header_bytes = phone.recv(header_size)

    # 第三步：从数据头中解析出对真实数据的描述信息
    header_json = header_bytes.decode("utf-8")
    header_dic = json.loads(header_json)
    print(header_dic)
    # 查询数据头中包含的信息
    total_size = header_dic["file_size"]
    filename = header_dic["filename"]

    # 第四步：接收真实的数据
    with open("%s\%s" % (download_dir, filename), "wb") as f:
        recv_size = 0
        while recv_size < total_size:
            line = phone.recv(1024)
            # 接收到一行就写入一行
            f.write(line)
            recv_size += len(line)
            print("总大小：%s 已经下载大小：%s" % (total_size, recv_size))

def put(phone,cmds):
    pass

def run():
    phone = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    phone.connect(("127.0.0.1",9910))

    while True:
        # 1、发命令
        inp = input(">>:").strip() # get a.txt
        if not inp:continue
        phone.send(inp.encode("utf-8"))

        cmds = inp.split() # ['get','a.txt']
        if cmds[0] == "get":
            get(phone,cmds)
        # 文件上传功能
        elif cmds[0] == "put":
            put(phone,cmds)

    phone.close()

if __name__ == '__main__':
    run()

文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/122895.html