socket上传文件

来源：互联网发布：java 矩阵乘法编辑：程序博客网时间：2024/05/19 17:08

server.py文件

#!/usr/bin/env python#coding:utf-8#上传文件(至服务器)import SocketServer,osclass 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-8import socket,osip_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()

1 0