TCP的文件上传主要考虑到如下几个问题:
- TCP的粘包问题--可以通过struct模块解决
--利用struck两个主要内置函数pack,unpack可以将发送的字符长度加工为指定的四个字节,这样接受的时候只接收四个字节长度的数据,就可以避免TCP的粘包
- 如何获取上传文件的大小及文件名--os模块的引用
- recv()每次接收的数据大小多是有限制的,如何传输较大的文件--通过while循环实现
server端
import json
import socket
import struct
# 接收
sk = socket.socket()
sk.bind(('127.0.0.1', 9002))
sk.listen()
conn, adrss = sk.accept()
msg_len = conn.recv(4) # 接收指定的四个字节的数据,表示的json串的长度,返回的是列表
dic_len = struct.unpack('i', msg_len)[0] # 列表的第一个元素为接收到数据
msg = conn.recv(dic_len).decode('utf-8')
msg = json.loads(msg)
with open(msg['filename'], 'wb') as f: # 以字节的形式写入文件
while msg['filesize'] > 0:
content = conn.recv(1024)
msg['filesize'] -= len(content) # 在当前长度的基础上减去收到的数据长度
f.write(content)
conn.close()
sk.close()
client端
import json
import socket
import os
import struct
# 发送
sk = socket.socket()
sk.connect(('127.0.0.1', 9002))
# 文件名\文件大小
abs_path = r'上传文件的绝对路径'
filename = os.path.basename(abs_path) # 获取上传文件的文件名
filesize = os.path.getsize(abs_path) # 获取上传文件的文件大小
dic = {'filename': filename, 'filesize': filesize}
str_dic = json.dumps(dic)
b_dic = str_dic.encode('utf-8')
mlen = struct.pack('i', len(b_dic)) # 通过struct模块将表示json长度的数值加工成四个字节
sk.send(mlen) # 发送四个字节表示的长度
sk.send(b_dic) # 发送字典的内容
with open(abs_path, 'rb') as f:
while filesize > 0: # 通过循环每次读取1024并发送
content = f.read(1024)
filesize -= len(content)
sk.send(content)
sk.close()