使用python socket分发大文件

思路：
使用socket传输文件过程中，如果单次传输每次只能发送一部分数据，如果针对大文件，一次传输肯定是不行的，所以需要我们在传输的时候提前把传输内容的大小先发送给客户端，在客户端循环接收数据即可。

代码部分：
一、服务器

#!/usr/bin/env python
#-*- coding:utf-8 -*-
#environment：2.7

import os,json,time,socket
sk = socket.socket()
ip_port=('127.0.0.1',9600)
sk.bind(ip_port)
sk.listen(5)
file_path = '/root/update/123.txt'
file_name = file_path.rsplit(os.sep,1)[1] 
file_size = os.path.getsize(file_path)   #获取文件内容大小
#发送给客户端的头部信息
header_data = {
    'file_name': file_name,
    'file_size': file_size,
    'date': time.strftime('%Y-%m-%d %X',time.localtime()),
    'charset': 'utf-8'
}
while True:
    conn, addr = sk.accept()
    print("%s:%s is connect"% addr)
    request_data = conn.recv(1024)
    print(request_data)
    # 把头部内容发送过去
    conn.send(json.dumps(header_data))
    request_data1 = conn.recv(1024)
    print(request_data1)
    f = open(file_path,'r')
    content = f.read()
    #发送文件内容
    conn.sendall(content)
    conn.close()

二、客户端

#!/usr/bin/env python
#-*- coding:utf-8 -*-
#environment：2.7

import socket,json
while True:
    sk = socket.socket()
    ip_port = ('127.0.0.1',9600)
    sk.connect(ip_port)
    user_input = raw_input('>>>:').strip()
    if len(user_input) == 0:continue
    if user_input == 'q':break
    sk.send(user_input)
    server_head_msg = json.loads(sk.recv(1024))
    print(server_head_msg)
    #文件名res_name，文件大小res_size
    res_name = server_head_msg['file_name']
    res_size = server_head_msg['file_size']
    sk.send('已经收到头部信息,你可以发送数据了')
    #下面是循环接收文件内容的部分
    num = res_size/1024.0
    if num != int(num):
        num = int(num) +1
    else:
        num = int(num)
    for i in range(num):
        content = sk.recv(1024)
        print(content)

备注：在接收服务器端发送来的文件也按照判断每次接收数据的大小来考虑，大致思路如下

res = ''
recive_size = 0
while recive_size <= res_size:
    data = sk.recv(1024)
    recive_size += len(data)
    res += data
else:
    print(res)
    print('done'.center(40,'='))