基于Python的TestAgent实现

问题：

1、本人工作主要做自动化，经常要去Linux后台进行一些脚本操作，有时要去后台执行命令，如果逐个登陆比较费事，效率会大打折扣

2、虽然有可以直接去后台执行命令的AW，但是该AW存在很多问题，而且遇到交互式操作时不能很好的解决

基于以上问题，通过Python写了一个简单的CLI Agent，就叫做TestAgent吧，主要思路：

1、采用POST消息发送到TestAgent，TestAgent进行解析

2、TestAgent接受到消息后，把消息体存为一个文件

3、将文件更改为可执行的，然后启动一个进程去执行脚本

4、如果执行成功将结果返回给客户端，如果失败，同样将错误输出也返回给客户端

5、在POST消息的头域中可以设置超时时间，如果超时，返回“time out”，并将启动的进程给杀掉

代码如下：

 #! /usr/bin/env python
 import commands
 import socket
 import time
 import os
 import multiprocessing
 import uuid
 from BaseHTTPServer import BaseHTTPRequestHandler,HTTPServer

 #HTTPServer的监听端口
 PORT=12345

 class HttpHandler(BaseHTTPRequestHandler):
     tmpfile=''
     #处理POST消息
     def do_POST(self):
         print self.path
         content_len = int(self.headers.getheader('content-length',0))
         #获取timeout
         timeout = int(self.headers.getheader('timeout',0))
         if timeout==0:
             timeout=5
         #解析消息并存储为文件
         script=self.rfile.read(content_len)
         x=uuid.uuid4()
         self.tmpfile="."+str(x.int)
         fd=open(self.tmpfile,'w')
         fd.write(script)
         fd.close()
         os.system("chmod +x "+self.tmpfile)
         script="./"+self.tmpfile
         #执行脚本
         self.ExecuteScript(script,timeout)

     def ExecuteScript(self,script,timeout=5):
         #启动另一个进程执行脚本
         p=multiprocessing.Process(target=self.ScriptWorker,args=(script,))
         p.start()
         i=0
         while i<timeout:
             if(not p.is_alive()):
                 return "successful"
             else:
                 time.sleep(1)
             i=i+1
         #超时的话终止进程并杀掉执行任务的进程
         p.terminate()
         os.system("kill -9 "+str(p.pid))
         self.send_error(400,"time out")
         self.request.shutdown(socket.SHUT_RDWR)
         self.request.close()
         #删除临时文件
         if self.tmpfile != '':
             os.system("rm "+self.tmpfile)
         self.tmpfile=''

     def ScriptWorker(self,script):
         #执行脚本，返回状态码和输出
         (status,result)=commands.getstatusoutput(script)
         print script
         print result
         #如果成功返回200，如果失败返回400
         if status == 0:
             self.send_response(200)
         else:
             self.send_response(400)
         self.send_header('Content-type','text/html')
         self.end_headers()
         self.wfile.write(result)
         #删除临时文件
         if self.tmpfile != '':
             os.system("rm "+self.tmpfile)
         self.tmpfile=''

 if __name__=='__main__':
     os.system('rm .*')
     server_address=('0.0.0.0',PORT)
     http_server=HTTPServer(server_address,HttpHandler)
     http_server.serve_forever()
         

测试：

采用curl或者restful client进行测试

1、执行简单命令

2、执行的命令不存在

3、执行一个Python脚本

4、执行一个超时的脚本

5、执行一个带有timeout头域的脚本

至此，基本所有功能都验证过了

PS：该程序理论上可以执行任何脚本，只要脚本的解释器写正确

使用时一般会再写个monitor脚本，放在crontab中，这样就完全可以不登陆服务器了，可以自动拉起TestAgent

希望该程序可以帮助大家，^v^ !!

地址：https://github.com/litlefirefly/TestAgent