Python--网络编程-----文件传输优化版本

一、服务端代码为：

import socket
import subprocess
import struct
import json
import os

share_dir = r'C:Usersxu516PycharmProjectsPython全栈开发第三模块网络编程6_文件传输优化版本_写函数servershare'


def get(conn, cmds):
    filename = cmds[1]
    # 3、以读的方式打开文件，读取文件内容发送到客户端

    # 第一步：制作固定长度的报头
    header_dic = {
        'filename': filename,
        'md5': 'xxdxx',
        'file_size': os.path.getsize('%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(r'%s\%s' % (share_dir, filename), 'rb') as f:
        # conn.send(f.read())
        for line in f:
            conn.send(line)


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', 8081))  # 0-65535:0-1024给操作系统使用
    phone.listen(5)

    print('starting......')
    while True:  # 链接循环
        conn, client_addr = phone.accept()
        print(client_addr)

        while True:  # 通信循环
            try:
                # 1、收命令
                res = conn.recv(8096)  # b'get a.txt'
                if not res:  # 适用于linux操作系统
                    break

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

            except ConnectionResetError:  # 适用于Windows操作系统
                break

        conn.close()

    phone.close()


if __name__ == '__main__':
    run()

二、客户端代码为：

import socket
import struct
import json

download_dir = r'C:Usersxu516PycharmProjectsPython全栈开发第三模块网络编程6_文件传输优化版本_写函数clientdownload'


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', 8081))

    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()

先启动服务端，在启动客户端，在客户端输入get test.py 下载成功：

客户端运行结果为：

>>:get test.py
{'filename': 'test.py', 'md5': 'xxdxx', 'file_size': 60}
总大小：60  已下载大小：60
>>: