Chinaunix首页 | 论坛 | 博客
  • 博客访问: 43501
  • 博文数量: 26
  • 博客积分: 20
  • 博客等级: 民兵
  • 技术积分: 245
  • 用 户 组: 普通用户
  • 注册时间: 2010-03-17 15:52
文章分类

全部博文(26)

文章存档

2016年(23)

2015年(3)

我的朋友

分类: Python/Ruby

2016-06-23 20:52:41

#!/usr/bin/env python    
#_*_coding:utf-8_*_


'''
@author ocean


'''


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 = True
            while Flag:
                #未上传完毕,
                if int(file_size)>recv_size:
                    #最多接收1024,可能接收的小于1024
                    data = conn.recv(1024) 
                    recv_size+=len(data)
                #上传完毕,则退出循环
                else:
                    recv_size = 0
                    Flag = False
                    continue
        
                #写入文件
                f.write(data)
            print 'upload successed.'
            f.close()
    
instance = SocketServer.ThreadingTCPServer(('127.0.0.1',9999),MyServer)
instance.serve_forever() 


客户端:
#!/usr/bin/env python    
#_*_coding:utf-8_*_


'''
@author ocean


'''


import socket
import sys
import os


ip_port = ('127.0.0.1',9999)
#ip_port = ('10.58.137.82',9999)
sk = socket.socket()
sk.connect(ip_port)


container = {'key':'','data':''}
while True:
    input = raw_input('path:')  #format:put|c:\123.txt
    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()


阅读(804) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~