python socket上传文件的服务端和客户端实现

ycvodzf

代码均已测试：希望高人能指点一二 　　服务端： 　　 #coding=gbk'''等待连接等待发送文件读取数据写入文件并且保存等待连接'''import socketimport threadingimport timeimport structdef 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: ",addressprint filename, len(filename),type(filename)print filesizefilename = 'new_'+filename.strip('\00')#命名新文件new_传送的文件fp = open(filename,'wb')#新建文件，并且准备写入restsize = filesizeprint "recving..."while 1:if restsize > 1024:#如果剩余数据包大于1024，就去1024的数据包filedata = newsock.recv(1024)else:filedata = newsock.recv(restsize)fp.write(filedata)breakif not filedata:breakfp.write(filedata)restsize = testsize - len(filedata)#计算剩余数据包大小if restsize <= 0:breakfp.close()print "recv succeeded !!File named:",filenameexcept:print "he socket partner maybe closed"newsock.close()breaksock=socket.socket(socket.AF_INET, socket.SOCK_STREAM)#创建tcp连接sock.bind(('127.0.0.1',8887))#定于端口和ipsock.listen(5)#监听while True:newsock, address = sock.accept()print "accept another connection"tmpThread = threading.Thread(target=function,args=(newsock,address)) #如果接收到文件，创建线程tmpThread.start()#执行线程print 'end'

客户端： #coding=gbk'''输入文件名，并且上传'''import socketimport timeimport structimport ossock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)sock.settimeout(1)e=0try:sock.connect(('127.0.0.1',8887))print 'connect...'except socket.timeout,e:print 'timeout',eexcept socket.error,e:print 'error',eexcept e:print 'any',eif not e:while (1):filename = raw_input('input your filename------->')#输入文件名FILEINFO_SIZE = struct.calcsize('128sI')#编码格式大小fhead = struct.pack('128sI',filename,os.stat(filename).st_size)#按照规则进行打包sock.send(fhead)#发送文件基本信息数据fp = open(filename,'rb')while 1:        #发送文件filedata = fp.read(1024)if not filedata:breaksock.send(filedata)print "sending over..."fp.close()