socket_server应用:发送命令,下载文件

import SocketServer,commands,time
class MyTCPHandler(SocketServer.BaseRequestHandler):

    def handle(self):
        while 1:
            self.data = self.request.recv(1024).strip()
            print "{} wrote:".format(self.client_address[0])
            if not self.data:
                print "client %s is dead!" %self.client_address[0]
                break
            user_input = self.data.split()
            if user_input[0] == "get":
                with open(user_input[1],'rb') as f:
                    self.request.sendall(f.read())
                time.sleep(0.5)
                self.request.send('FileTransFerDone')
                continue
            cmd_status,result = commands.getstatusoutput(self.data)
            if len(result.strip()) !=0:
                self.request.sendall(result)
            else:
                self.request.sendall("DONE")
if __name__ == "__main__":
    HOST,PORT = "localhost",9999
    server = SocketServer.ThreadingTCPServer((HOST, PORT),MyTCPHandler)
    server.serve_forever()
socket_server

time.sleep 休眠0.5秒

import socket

HOST = "localhost"
PORT = 9999
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect((HOST,PORT))

while 1:
    cmd = raw_input("your command:").strip()
    if len(cmd) == 0:continue
    s.sendall(cmd)
    if cmd.split()[0] == "get":
        with open(cmd.split()[1],'wb') as f:
            while 1:
                data = s.recv(1024)
                if data =='FileTransFerDone':break
                f.write(data)
        continue
    else:
        print s.recv(1024)
s.close()
socket_client.py

get 关键字

posted @ 2016-01-31 13:39  杨点风007  阅读(303)  评论(0编辑  收藏  举报