socket上傳文件

server.py文件

#!/usr/bin/env python
#coding:utf-8
#上傳文件(至服務器)
import SocketServer,os
class MyServer(SocketServer.BaseRequestHandler):
    def handle(self):
        base_path = r'C:\Users\91135\Desktop\test'
        conn = self.request
        print 'has 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 = True
            while Flag:
                if int(file_size)>recv_size:#未上傳完畢
                    data = conn.recv(1024)#最多接收1024，可能接收的小於1024
                    recv_size+=len(data)
                    f.write(data)#寫入文件
                else:#上傳完畢，則退出循環
                    #recv_size = 0
                    Flag = False
            print 'upload successed.'
            f.close()
instance = SocketServer.ThreadingTCPServer(('127.0.0.1',9999),MyServer)
instance.serve_forever()

client.py文件

#!/usr/bin/env python
#coding:utf-8
import socket,os
ip_port = ('127.0.0.1',9999)
sk = socket.socket()
sk.connect(ip_port)
#container = {'key':'','data':''}
while True:
    input = raw_input('cmd and path:')#cmd和path之間用'/'分割
    cmd,path = input.split('/') 
    file_name = os.path.basename(path)
    file_size=os.stat(path).st_size
    sk.send(cmd+"/"+file_name+'/'+str(file_size))
    send_size = 0
    f= file(path,'rb')
    Flag = True
    while Flag:
        if send_size + 1024 >=file_size:
            data = f.read(file_size-send_size)
            Flag = False
        else:
            data = f.read(1024)
            send_size+=1024
        sk.send(data)
    f.close()
sk.close()