利用Socketserver实现简单的文件上传

server.py

#!/usr/bin/env python
#coding:utf-8
import SocketServer
import os
class MyServer(SocketServer.BaseRequestHandler):def handle(self):base_path = 'C:/temp'conn = self.request        print 'connected...'while True:pre_data = conn.recv(1024)            #获取请求方法、文件名、文件大小cmd,file_name,file_size = pre_data.split('|')            #已经接收文件的大小recv_size = 0#上传文件路径拼接file_dir = os.path.join(base_path,file_name)f = file(file_dir,'wb')Flag = Truewhile Flag:                #未上传完毕，if int(file_size)>recv_size:                    #最多接收1024，可能接收的小于1024data = conn.recv(1024)recv_size+=len(data)                #上传完毕，则退出循环else:recv_size = 0Flag = Falsecontinue#写入文件f.write(data)            print 'upload successed.'f.close()
if __name__ == '__main__':instance = SocketServer.ThreadingTCPServer(('127.0.0.1',9999),MyServer)instance.serve_forever()

client.py

#!/usr/bin/env python
# coding:utf-8
__author__ = 'aolens'
import socket
import sys
import osip_port = ('127.0.0.1',9999)
s = socket.socket()
s.connect(ip_port)container = {'key':'','data':''}
while True:input = raw_input('path:')cmd,path = input.split('|')file_name = os.path.basename(path)file_size=os.stat(path).st_sizes.send(cmd+"|"+file_name+'|'+str(file_size))send_size = 0f= file(path,'rb')Flag = Truewhile Flag:        if send_size + 1024 >file_size:data = f.read(file_size-send_size)Flag = Falseelse:data = f.read(1024)send_size+=1024s.send(data)f.close()s.close()

客户端输出path的格式为:方法|路径

path:PUT|C:/oem8.log

查看上传文件路径C:/temp下，发现有了一个oem8.log文件，上传成功