天天看點

python socket傳檔案

python傳輸檔案最重要的有兩步:

1).将要傳輸的檔案的資訊發送過去,包括檔案包,大小以及其它資訊;

2).發送端讀取檔案内容并發送過去,接受端将緩存裡面的内容寫入檔案.

發送端:

# -*- coding: cp936 -*-

from socket import *

import os

import struct

ADDR = ('192.168.0.178',8000)

BUFSIZE = 1024

filename = 'wubi.exe'

FILEINFO_SIZE=struct.calcsize('128s32sI8s')

sendSock = socket(AF_INET,SOCK_STREAM)

sendSock.connect(ADDR)

fhead=struct.pack('128s11I',filename,0,0,0,0,0,0,0,0,os.stat(filename).st_size,0,0)

sendSock.send(fhead)

fp = open(filename,'rb')

while 1:

    filedata = fp.read(BUFSIZE)

    if not filedata: break

    sendSock.send(filedata)

print "檔案傳送完畢,正在斷開連接配接..."

fp.close()

sendSock.close()

print "連接配接已關閉..."

接收端:

recvSock = socket(AF_INET,SOCK_STREAM)

recvSock.bind(ADDR)

recvSock.listen(True)

print "等待連接配接..."

conn,addr = recvSock.accept()

print "用戶端已連接配接—> ",addr

fhead = conn.recv(FILEINFO_SIZE)

filename,temp1,filesize,temp2=struct.unpack('128s32sI8s',fhead)

#print filename,temp1,filesize,temp2

print filename,len(filename),type(filename)

print filesize

filename = 'new_'+filename.strip('\00') #...

fp = open(filename,'wb')

restsize = filesize

print "正在接收檔案... ",

    if restsize > BUFSIZE:

        filedata = conn.recv(BUFSIZE)

    else:

        filedata = conn.recv(restsize)

    fp.write(filedata)

    restsize = restsize-len(filedata)

    if restsize == 0:

     break

print "接收檔案完畢,正在斷開連接配接..."

conn.close()

recvSock.close()

該程式經過實際測試可用,大家經過修改相應代碼即可在本地進行測試使用。

友情提示:

用于傳送比較大的檔案時會比較慢,還可能照成指令行假死的情況...