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