使用python实现简单的文件下载
client端
import socket
import os
import json
import struct
sk = socket.socket()
IP_ADDRESS = "127.0.0.1"
PORT = 9999 ##端口号需要和server一直
BUFFER = 1024 ## buffer size 可以自定义大小,不需要和server一致
sk.connect((IP_ADDRESS, PORT)) # 连接server
header = {
"filePath" : "", # 本地文件路径
"filename" : "", # 本地文件名字
"filesize" : "", # 后面更新文件大小,可以先设置为空
}
filename = os.path.join(header["filePath"], header["filename"]) # 组成文件路径
filesize = os.path.getsize(filename) # 获取文件大小,并更新至header中
header["filesize"] = filesize
json_header = json.dumps(header) # 将字典转换为字符串,以供后面发送
struct_length = struct.pack('i', len(json_header)) # 数据头的长度
sk.send(struct_length) # 发送header长度
sk.send(json_header.encode('utf-8')) # 发送header
with open(filename, 'rb') as f: # 从本地读取文件,并将文件发送至server
while filesize:
print(filesize)
if filesize >= BUFFER:
content = f.read(BUFFER)
sk.send(content)
filesize -= BUFFER
else:
content = f.read(filesize)
sk.send(content)
break
sk.close() # 关闭socket
复制代码
server端
import socket
import struct
import json
IP_ADDRESS = "127.0.0.1" # 本地ip地址
PORT = 9999 # 端口号,可以自定义
BUFFER = 1024 ## TCP 缓冲大小设置,可以设置成其他大小
sk = socket.socket() # 创建一个socket对象
sk.bind((IP_ADDRESS, PORT)) # 监听这个ip地址
sk.listen() #
conn, addr = sk.accept() # 等待连接
struct_length = conn.recv(4) # 先接收struct 打包好的4个字节长度
json_length = struct.unpack('i', struct_length)[0] # 解析二进制数据并获取长度
json_header = conn.recv(json_length).decode("utf-8") # 根据长度接收自定义header
header = json.loads(json_header) # 由于header是json 格式,通过json转换为字典
print(header)
filesize = header["filesize"] # 获取文件大小
filename = header["filename"] # 获取文件名称
with open(filename, 'wb+') as f: # 将文件写入本地
while filesize:
if filesize >= BUFFER:
content = conn.recv(BUFFER)
f.write(content)
filesize -= BUFFER
else:
content = conn.recv(filesize)
f.write(content)
filesize -= filesize
break
conn.close() # 关闭连接
sk.close() # 关闭socket
复制代码