Python Socket傳輸文件

發送端可以不停的發送新文件，接收端可以不停的接收新文件。

例如：發送端輸入：e:\visio.rar，接收端會默認保存為 e:\new_visio.rar，支持多並發，具體實現如下；

接收端：

方法一：

#-*- coding: UTF-8 -*-
import socket,time,SocketServer,struct,os,thread
host='192.168.50.74'
port=12307
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM) #定義socket類型
s.bind((host,port)) #綁定需要監聽的Ip和端口號，tuple格式
s.listen(1)

 
def conn_thread(connection,address):  
    while True:
        try:
            connection.settimeout(600)
            fileinfo_size=struct.calcsize('128sl') 
            buf = connection.recv(fileinfo_size)
            if buf: #如果不加這個if，第一個文件傳輸完成后會自動走到下一句
                filename,filesize =struct.unpack('128sl',buf) 
                filename_f = filename.strip('\00')
                filenewname = os.path.join('e:\\',('new_'+ filename_f))
                print 'file new name is %s, filesize is %s' %(filenewname,filesize)
                recvd_size = 0 #定義接收了的文件大小
                file = open(filenewname,'wb')
                print 'stat receiving...'
                while not recvd_size == filesize:
                    if filesize - recvd_size > 1024:
                        rdata = connection.recv(1024)
                        recvd_size += len(rdata)
                    else:
                        rdata = connection.recv(filesize - recvd_size) 
                        recvd_size = filesize
                    file.write(rdata)
                file.close()
                print 'receive done'
                #connection.close()
        except socket.timeout:
            connection.close()


while True:
    connection,address=s.accept()
    print('Connected by ',address)
    #thread = threading.Thread(target=conn_thread,args=(connection,address)) #使用threading也可以
    #thread.start()
    thread.start_new_thread(conn_thread,(connection,address)) 

s.close()

方法二：

#-*- coding: UTF-8 -*-
import socket,time,SocketServer,struct,os
host='192.168.50.74'
port=12307
ADDR=(host,port)

class MyRequestHandler(SocketServer.BaseRequestHandler):    
    def handle(self):      
        print('connected from:', self.client_address)
        while True:
            fileinfo_size=struct.calcsize('128sl') #定義文件信息。128s表示文件名為128bytes長，l表示一個int或log文件類型，在此為文件大小
            self.buf = self.request.recv(fileinfo_size)
            if self.buf: #如果不加這個if，第一個文件傳輸完成后會自動走到下一句
                self.filename,self.filesize =struct.unpack('128sl',self.buf) #根據128sl解包文件信息，與client端的打包規則相同
                print 'filesize is: ',self.filesize,'filename size is: ',len(self.filename) #文件名長度為128，大於文件名實際長度
                self.filenewname = os.path.join('e:\\',('new_'+ self.filename).strip('\00')) #使用strip()刪除打包時附加的多余空字符
                print self.filenewname,type(self.filenewname)
                recvd_size = 0 #定義接收了的文件大小
                file = open(self.filenewname,'wb')
                print 'stat receiving...'
                while not recvd_size == self.filesize:
                    if self.filesize - recvd_size > 1024:
                        rdata = self.request.recv(1024)
                        recvd_size += len(rdata)
                    else:
                        rdata = self.request.recv(self.filesize - recvd_size) 
                        recvd_size = self.filesize
                    file.write(rdata)
                file.close()
                print 'receive done'
        #self.request.close()

tcpServ = SocketServer.ThreadingTCPServer(ADDR, MyRequestHandler)  
print('waiting for connection...' )
tcpServ.serve_forever()

發送端：

#-*- coding: UTF-8 -*-
import socket,os,struct
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect(('192.168.50.74',12307))
while True:
    
    filepath = raw_input('Please Enter chars:\r\n')
    if os.path.isfile(filepath):
        fileinfo_size=struct.calcsize('128sl') #定義打包規則
        #定義文件頭信息，包含文件名和文件大小
        fhead = struct.pack('128sl',os.path.basename(filepath),os.stat(filepath).st_size)
        s.send(fhead) 
        print 'client filepath: ',filepath
        # with open(filepath,'rb') as fo: 這樣發送文件有問題，發送完成后還會發一些東西過去
        fo = open(filepath,'rb')
        while True:
            filedata = fo.read(1024)
            if not filedata:
                break
            s.send(filedata)
        fo.close()
        print 'send over...'
        #s.close()