使用python socket分發(fā)大文件的實現(xiàn)方法

更新時間：2019年07月08日 18:38:10 作者：幸福丶如此

今天小編就為大家分享一篇使用python socket分發(fā)大文件的實現(xiàn)方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

思路：

使用socket傳輸文件過程中，如果單次傳輸每次只能發(fā)送一部分數(shù)據(jù)，如果針對大文件，一次傳輸肯定是不行的，所以需要我們在傳輸?shù)臅r候提前把傳輸內(nèi)容的大小先發(fā)送給客戶端，在客戶端循環(huán)接收數(shù)據(jù)即可。

代碼部分：

一、服務器

#!/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)  #獲取文件內(nèi)容大小
#發(fā)送給客戶端的頭部信息
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)
  # 把頭部內(nèi)容發(fā)送過去
  conn.send(json.dumps(header_data))
  request_data1 = conn.recv(1024)
  print(request_data1)
  f = open(file_path,'r')
  content = f.read()
  #發(fā)送文件內(nèi)容
  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('已經(jīng)收到頭部信息,你可以發(fā)送數(shù)據(jù)了')
  #下面是循環(huán)接收文件內(nèi)容的部分
  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)

備注：在接收服務器端發(fā)送來的文件也按照判斷每次接收數(shù)據(jù)的大小來考慮，大致思路如下

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,'='))

以上這篇使用python socket分發(fā)大文件的實現(xiàn)方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: