利用python对读写文件以及异常控制来实现文件的下载
首先,创建client.py和server.py,实现客户端和服务端的socket通信,先运行server.py,在运行client.py。练习socket通信时,可以使用网络调试助手辅助。
server.py代码
import socket
#客户端接收用户文件名
server=socket.socket()
server.bind(('localhost',8088))
server.listen(5)
print('waiting for connection...')
#接收到连接请求后,创建一个新的socket
client,addr=server.accept()
print('new connection from',addr)
filename=client.recv(1024).decode('utf-8')
#接收到文件名后读取
#判断有没有读取到,没有说明文件不存在,抛出异常
#客户端判断返回的数据是消息还是接收的文件数据
try:
with open(filename,'r') as f:
#一次读取1024个字节
while True:
data=f.read(1024)
if data:
print('正在发送数据...')
client.send(data.encode('utf-8'))
else:
break
except Exception as e:
print(f'客户端下载的文件:{filename}不存在!')
else:
print(f'文件{filename}发送成功!')
finally:
f.close()
client.close()
server.close()
client.py代码
#客户端发送一个文件名到服务端
import socket
client=socket.socket()#默认TCP连接
client.connect(('localhost',8088))
filename=input('请输入文件名:')
client.send(filename.encode('utf-8'))
#接收文件数据并保存
content=''
try:
while True:
data=client.recv(1024).decode('utf-8')
if data:
content+=data
else:
break
except Exception as e:
print('文件传输失败,下载的文件可能不存在!',e)
else:
#print('content:',content)
try:
with open('file/b.txt','w') as f:
f.write(content)
except Exception as e:
print('写入文件失败',e)
else:
print('写入文件成功!')
print('下载结束...')
finally:
content=""
client.close()
#1024传输效率高一些
在这两个py文件的同级目录下创建file文件夹用来存放下载的文件。
启动server.py和client.py后,在client.py控制台中输入要下载的文件名如a.txt
当然,这个程序还有很大缺陷,下载不存在的文件时服务端响应倒是很好,但客户端没有接收到下载的文件是否存在,程序也会在file文件下创建一个空的要下载的文件。
还请有识之士指正!