文件传输-一个大文件传输

server

import socket
import struct
import hashlib
import os

host = '127.0.0.1'
port = 8080
buffer_size = 1024
head_strut = '128sq32s' #格式 file_name, file_size, md5 128+8+32
path = '/Users/apple/Desktop/test.mp4'
info_size = struct.calcsize(head_strut)

#md5的编译
def compile_md5(path):
    file_size = os.path.getsize(path)
    md5 = hashlib.md5()
    readed_size = 0
    with open(path, 'rb') as f:
        while readed_size < file_size:
            remain_size = file_size - readed_size
            read_size = buffer_size if buffer_size < remain_size else remain_size
            readed_size += read_size
            md5.update(f.read(read_size))
    md = md5.hexdigest()
    return md

#将struct翻译回来
def unpack_recv(recv_info):
    file_name, file_size, md5 = struct.unpack(head_strut, recv_info)
    return file_name, file_size, md5

def main():
    sk = socket.socket()
    sever_address = (host, port)
    sk.bind(sever_address)
    sk.listen()
    conn, addr = sk.accept()
    print('端口{}客户端连接成功'.format(addr[1]))

    recv_info = conn.recv(info_size)  # 接收长度信息
    file_name, file_size, md5_info = unpack_recv(recv_info)

    recved_size = 0
    with open(path, 'wb') as f:
        while recved_size < file_size:
            remain_size = file_size - recved_size
            recv_size = buffer_size if buffer_size < remain_size else remain_size
            recved_size += recv_size
            info = conn.recv(recv_size)   #接收相应长度的数据
            f.write(info)   #将接受的数据写入指定位置
    md5 = compile_md5(path)
    if md5 == md5_info.decode('utf-8'):
        print('文件校验正确，确认传输完成！')
    else:
        print('文件校验失败！', md5)

    conn.close()
    sk.close()

if __name__ == '__main__':
    main()
------------------------------------------------
------------------------------------------------
#client
import socket
import struct
import os
import hashlib


host = '127.0.0.1'
port = 8080
head_struct = '128sq32s'
buffer_size = 1024
path = '/Users/apple/Desktop/vido/day33/03 python fullstack s9day33 黏包现象.mp4'

#过去md5
def get_md5(path):
    file_size = os.path.getsize(path)
    md5 = hashlib.md5()
    readed_size = 0
    with open(path, 'rb') as f:
        while readed_size < file_size:
            remain_size = file_size - readed_size
            read_size = buffer_size if buffer_size < remain_size else remain_size
            readed_size += read_size
            md5.update(f.read(read_size))
    md = md5.hexdigest()
    return md

#获取file_name, file_size, md5值
def get_value(path):
    file_name = os.path.basename(path)
    file_size = os.path.getsize(path)
    md = get_md5(path)
    return file_name, file_size, md

def main():
    sk = socket.socket()
    sk.connect((host, port))

    file_name, file_size, md = get_value(path)
    print(type(file_name), type(file_size), type(md))
    struct_info = struct.pack(head_struct, bytes(file_name, encoding='utf-8'), file_size, bytes(md, encoding='utf-8'))
    sk.send(struct_info)

    sended_size = 0
    with open(path, 'rb') as f:
        while sended_size < file_size:
            remain_size = file_size - sended_size
            send_size = buffer_size if buffer_size < remain_size else buffer_size
            sended_size += send_size
            send_info = f.read(send_size)
            sk.send(send_info)

        print('文件传输结束', md)

    sk.close()

if __name__ == '__main__':
    main()