python：socket传输大文件示例

文件可以传输，但是对比传输前后的文件：socket_test.txt，末尾有一些不一致服务端代码：

    #!/usr/bin/python
    # -*- coding: utf-8 -*-
    import sys
    reload(sys)
    sys.setdefaultencoding("utf-8")
    import time
    '''
    等待连接
    等待发送文件
    读取数据
    写入文件并且保存
    等待连接
    '''
    import socket
    import threading
    import time
    import struct


    def function(newsock, address):
      FILEINFO_SIZE = struct.calcsize('128sI')
      '''定义文件信息（包含文件名和文件大小）大小。128s代表128个char[]（文件名），I代表一个integer or long（文件大小）'''
      while 1:
        try:
          fhead = newsock.recv(FILEINFO_SIZE)
          filename, filesize = struct.unpack('128sI', fhead)
          '''把接收到的数据库进行解包，按照打包规则128sI'''
          print "address is: ", address
          print filename, len(filename), type(filename)
          print filesize
          #filename = 'new_'+filename.strip('\00') # 命名新文件new_传送的文件
          filename = filename.strip('\00')
          fp = open(filename, 'wb') # 新建文件，并且准备写入
          restsize = filesize
          print "recving..."
          while 1:
            if restsize > 102400: # 如果剩余数据包大于1024，就去1024的数据包
              filedata = newsock.recv(10240)
            else:
              filedata = newsock.recv(restsize)
              fp.write(filedata)
              #break
            if not filedata:
              break
            fp.write(filedata)
            restsize = restsize - len(filedata) # 计算剩余数据包大小
            if restsize <= 0:
              break
          fp.close()
          print "recv succeeded !!File named:", filename
        except Exception, e:
          print unicode(e).encode('gbk')
          print "the socket partner maybe closed"
          newsock.close()
          break
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 创建tcp连接
    sock.bind(('10.240.146.82', 8887)) # 定于端口和ip
    sock.listen(5) # 监听
    while True:
      newsock, address = sock.accept()
      print "accept another connection"
      tmpThread = threading.Thread(target=function, args=(newsock, address)) # 如果接收到文件，创建线程
      tmpThread.start() # 执行线程
    print 'end'

客户端代码：

    #!/usr/bin/python
    # -*- coding: utf-8 -*-
    import sys
    reload(sys)
    sys.setdefaultencoding("utf-8")
    '''
    输入文件名，并且上传
    '''
    import socket
    import time
    import struct
    import os
    f = open('socket_test.txt', 'wb')

    for i in range(1000000):
      f.write('for socket test, the line number is : ' + str(i) + '\n')

    f.close()

    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.settimeout(50)
    e = 0
    try:
      sock.connect(('10.240.146.82', 8887))
      print 'connect...'
    except socket.timeout, e:
      print 'timeout', e
    except socket.error, e:
      print 'error', e
    except e:
      print 'any', e
    if not e:
      #while (1):
        #filename = raw_input('input your filename------->') # 输入文件名
      filename = 'socket_test.txt'
      FILEINFO_SIZE = struct.calcsize('128sI') # 编码格式大小
      fhead = struct.pack('128sI', filename, os.stat(filename).st_size) # 按照规则进行打包
      sock.send(fhead) # 发送文件基本信息数据
      fp = open(filename, 'rb')
      fp2 = open('local_test.txt','wb')
      i = 0
      while 1: # 发送文件
        filedata = fp.read(10240)
        if not filedata:
          break
        sock.sendall(filedata)
        fp2.write(filedata)
        print i
        i = i + 1
      print "sending over..."
      fp.close()
      fp2.close()

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。