Python网络编程(三)-基于tcp协议实现文件传输(解决传输大文件完成后,文件不完整情况)

server.py

#服务端接收
#解决传输大问题可能出现的传输完成后，文件不完整现象

import json
import socket

sk = socket.socket()
sk.bind(('127.0.0.1',9000))
sk.listen()

conn,addr = sk.accept()
msg = conn.recv(1024).decode('utf-8')
msg = json.loads(msg)
print(msg['filename'],msg['filesize'])

with open(msg['filename'],mode='wb') as f:
    #针对大文件，每次传1024字节，真实情况下，tcp socket 会自动优化，可能将1024个字节切分为几个小文件，
    #真实大小则不一定为1024，默认接受方还是会按照1024进行减1024字节，因此发送大文件可能会出问题
    #需要减去的是，每次接受到的字节大小
    while msg['filesize'] >0:
        content = conn.recv(1024)
        # msg['filesize'] -= 1024  #减1024，在传输数据时候会出问题
        # 应减去每次具体收到的字节大小，才会准确
        msg['filesize'] -= len(content)
        # 写入内容到指定文件
        f.write(content)
conn.close()
sk.close()

 

client.py

#客户端发送
#
import os
import json
import socket

sk = socket.socket()
sk.connect(('127.0.0.1',9000))

#需要发送文件名，文件大小

# abs_path = r'E:\project\testfile'
abs_path = r'F:\web\day43\day43-002 内容回顾.mp4'
# os.path模块获取文件名
filename = os.path.basename(abs_path)
# os.path模块获取文件大小
filesize = os.path.getsize(abs_path)
dic = {'filename':filename,'filesize':filesize}
#  网编中传输数据，常用的是json类型
str_dic = json.dumps(dic)
sk.send(str_dic.encode('utf-8'))

with open(abs_path,mode='rb') as f:
    #针对大文件  每次发送1024字节，总文件大小减去每次发送的字节大小进行while循环
    while filesize > 0:
        content = f.read( 1024)
        # filesize -= 1024
        filesize -= len(content)
        sk.send(content)

sk.close()