上传下载优化版本

标签：NPU   ade   /usr   bind   ror   pac   close   odi   try   

server------------

#!/usr/bin/env python
# encoding: utf-8  
# Date: 2018/6/7


import socket
import os
import json
import struct

shar_dir = r‘E:\python\PycharmProjects\python.base.demo\model3\socket1\上传下载\优化版本\server\share‘


def put(conn, cmds):
    obj = conn.recv(4)
    header_size = struct.unpack(‘i‘, obj)[0]
    # 2.2收报头
    header_bytes = conn.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‘]
    # 2.4,接收真实的数据
    with open(‘%s/%s‘ % (shar_dir, filename), ‘wb‘) as f:  # 服务端打开读方式，这里这种wb方式容易卡主
        recv_size = 0
        while recv_size < total_size:
            line = conn.recv(1024)
            f.write(line)
            recv_size += len(line)
            print(‘总大小:%s 已上传大小： %s‘ % (total_size, recv_size))


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

    # 3,易读的方式打开文件，读取文件内容发送给客户端

    header_dic = {
        ‘filename‘: filename,
        ‘md5‘: ‘xxxdxxx‘,
        ‘file_size‘: os.path.getsize(r‘%s/%s‘ % (shar_dir, filename))
    }

    header_json = json.dumps(header_dic)

    header_bytes = header_json.encode(‘utf-8‘)

    # 2,先发送报头的长度
    conn.send(struct.pack(‘i‘, len(header_bytes)))
    # 3,再发报头
    conn.send(header_bytes)

    # 4,再发送真实的数据
    with open(‘%s/%s‘ % (shar_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.bind((‘127.0.0.1‘, 9909))
    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:break

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

            except ConnectionResetError:
                break
        conn.close()
    phone.close()


if __name__ == ‘__main__‘:
    run()


client--------------------

#!/usr/bin/env python
# encoding: utf-8  
# Date: 2018/6/7


import socket
import struct
import json
import os

dow_dir = r‘E:\python\PycharmProjects\python.base.demo\model3\socket1\上传下载\优化版本\client\download‘


def put(phone, cmds):
    filename = cmds[1]

    # 3,易读的方式打开文件，读取文件内容发送给客户端

    header_dic = {
        ‘filename‘: filename,
        ‘md5‘: ‘xxxdxxx‘,
        ‘file_size‘: os.path.getsize(r‘%s/%s‘ % (dow_dir, filename))
    }

    header_json = json.dumps(header_dic)

    header_bytes = header_json.encode(‘utf-8‘)

    # 2,先发送报头的长度
    phone.send(struct.pack(‘i‘, len(header_bytes)))
    # 3,再发报头
    phone.send(header_bytes)

    # 4,再发送真实的数据
    with open(‘%s/%s‘ % (dow_dir, filename), ‘rb‘) as f:
        # conn.send(f.read())
        for line in f:
            phone.send(line)


def get(phone, cmds):
    # 2,以写的方式打开一个新文件
    # 2.1收报头长度
    obj = phone.recv(4)
    header_size = struct.unpack(‘i‘, obj)[0]
    # 2.2收报头
    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‘]
    # 2.4,接收真实的数据
    with open(‘%s/%s‘ % (dow_dir, filename), ‘wb‘) as f:  # 服务端打开读方式，这里这种wb方式容易卡主
        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 run():

    phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    phone.connect((‘127.0.0.1‘, 9909))

    while True:
        # 1,发命令
        cmd = input(‘>>:‘).strip()
        if not cmd:continue
        phone.send(cmd.encode(‘utf-8‘))

        cmds = cmd.split()
        if cmds[0] == ‘get‘:
            get(phone, cmds)
        elif cmds[0] == ‘put‘:
            put(phone, cmds)

    phone.close()


if __name__ == ‘__main__‘:
    run()

上传下载优化版本

标签：NPU   ade   /usr   bind   ror   pac   close   odi   try   

原文地址：https://www.cnblogs.com/fmgao-technology/p/9189438.html