Python全栈之路系列之基于socket实现文件上传
发布时间:2017年3月16日 00:04 浏览(106) 评论(0) 分类:Python
前言
此处没有前言
粘包
在实现发送文件功能之前我们先来理解下粘包的问题,下面有两张图,我觉得很清晰的就可以理解到了。
-
正常情况下发送文件
-
第一步: 客户端把获取到的文件总大小(size=65426)先放到缓冲区,然后发送给服务端
-
第二步: 此时客户端接收到的文件总大小就是65426
-
粘包的问题下发送文件
-
第一步: 客户端把获取到的文件总大小(size=65426)先放到缓冲区
-
第二步: 此时可能由于文件读取太快,导致缓存区的内容还没有发送到服务端,客户端就把读取到的文件内容(hello)也放到缓存区;
-
第三步: 然后客户端就把缓存区的全部内容都发送到服务端,那么客户端本来第一次应该接收到的数据室文件大小(size=65426),但实际接收到的数据确实:65426+hello,那么这个流程就是粘包的问题;
-
解决粘包问题
如果出现粘包的问题,那么传输的数据就有问题了,如何解决这个问题呢?看下图:
-
第一步: 客户端把文件大小放到缓冲区
-
第二步: 放入缓冲区之后立刻陷入阻塞的状态,登台服务端回复已收到文件大小,此时是不会再向服务端发送任何数据的
-
第三步: 缓存区的数据会发送到服务端
-
第四步: 服务端接收到客户端发来的文件大小之后立刻回复客户端,说我收到你发过来的文件大小了;
文件上传
执行结果如下
client.py
文件内容
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
ansheng@Darker:~ / socket_file$ cat client.py
#!/usr/bin/env python # _*_coding:utf-8 _*_ import socket
import os
# 创建一个socket对象 obj = socket.socket()
# 服务端的IP和端口 obj.connect(( '127.0.0.1' , 6542 ))
# 用os模块获取要传送的文件总大小 size = os.stat( "old_file.txt" ).st_size
# 把文件总大小发送给服务端 obj.sendall(bytes( str (size), encoding = "utf-8" ))
# 接受服务端返回的信息 obj.recv( 1024 )
# 以rb的模式打开一个要发送的文件d with open ( "old_file.txt" , "rb" ) as f:
# 循环文件的所有内容
for line in f:
# 发送给服务端
obj.sendall(line)
# 关闭退出 obj.close() |
service.py
文件内容
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
ansheng@Darker:~ / socket_file$ cat service.py
#!/usr/bin/env python # _*_coding:utf-8 _*_ import socket
# 创建一个socket对象 sk = socket.socket()
# 允许连接的IP和端口 sk.bind(( '127.0.0.1' , 6542 ))
# 最大连接数 sk.listen( 5 )
while True :
# 会一直阻塞,等待接收客户端的请求,如果有客户端连接会获取两个值,conn=创建的连接,address=客户端的IP和端口
conn, address = sk.accept()
# 客户端发送过来的文件大小
file_size = str (conn.recv( 1024 ),encoding = "utf-8" )
# 给客户端发送已经收到文件大小
conn.sendall(bytes( "ack" , encoding = "utf-8" ))
# 文件大小转换成int类型
total_size = int (file_size)
# 创建一个默认的值
has_recv = 0
# 打开一个新文件,以wb模式打开
f = open ( 'new_file.txt' , 'wb' )
# 进入循环
while True :
# 如果传送过来的大小等于文件总大小,那么就退出
if total_size = = has_recv:
break
# 接受客户端发送过来的内容
data = conn.recv( 1024 )
# 写入到文件当中
f.write(data)
# 现在的大小加上客户端发送过来的大小
has_recv + = len (data)
# 关闭
f.close()
|
本文转自 Edenwy 51CTO博客,原文链接:http://blog.51cto.com/edeny/1918823,如需转载请自行联系原作者