#!/usr/bin/env python
# -*- coding:utf8 -*-
import urllib
import datetime
import os
import re
import json
import sys
import logging
import Daemon
from providers.s_email import SendMailHandle
from providers.s_sms import SendsmsHandle
from providers.s_im import SendImHandle
default_encoding = 'utf-8'
from BaseHTTPServer import HTTPServer, BaseHTTPRequestHandler
with open('./cfg.json', 'r') as f:
config = json.load(f)
data = config
class HTTPHandle(BaseHTTPRequestHandler):
def transDicts(self, params, logger):
dicts = {}
if len(params) == 0:
return
params = params.split('&')
for param in params:
if 'tos' in param:
dicts['tos'] = param.replace('tos=', '')
if 'content' in param:
dicts['content'] = param.replace('content=', '')
if 'subject' in param:
dicts['subject'] = param.replace('subject=', '')
if 'type' in param:
dicts['type'] = param.replace('type=', '')
return dicts
def do_POST(self):
mpath, margs = urllib.splitquery(self.path)
logging.config.fileConfig("./logging.ini")
logger = logging.getLogger(__name__)
logger.info("start to handle request")
logger.info("self.path %s" % self.path)
datas = self.rfile.read(int(self.headers['content-length']))
logger.info("datas %s" % datas)
datas = urllib.unquote(datas).decode("utf-8", 'ignore') # 指定编码方式
logger.info("datas %s" % datas)
logger.info("data----=" + repr(datas))
if not datas:
request_data = self.transDicts(self.path.split("?")[1], logger)
type_args = request_data.get("type")
else:
request_data = self.transDicts(datas, logger)
type_args = margs.replace('type=', '')
logger.info("request_data %s %s " % (type(request_data.get('content')), request_data.get('content')))
logger.info("dict----:" + repr(request_data))
pattern = r"(\[.*?\])"
guid = re.findall(pattern, request_data.get('content'), re.M)
content = ''
if len(guid) > 2:
dt = datetime.datetime.now()
# try:
t = guid[-1].replace('[', '').replace(']', '').split("+")
dt = (" ").join(x for x in t[1:len(t)])
# except BaseException,e:
# logger.error("error:%s"%e)
content = u'优先级：{0}\r\n状态：{1}\r\n监控对象：{2}\r\n日期：{3}\r\nMetric：{4}'.format(
guid[0].replace('[', '').replace(']', ''),
guid[1].replace('[', '').replace(']', ''),
guid[2].replace('[', '').replace(']', ''),
dt,
request_data.get('content').split("[")[5].split("]")[0])
if type_args == 'mail':
mail_config = data.get('mail')
try:
mail_handle = SendMailHandle(mail_config.get('mail_host'), mail_config.get('mail_user'),
mail_config.get('mail_pass'), mail_config.get('mail_postfix'))
mail_handle.send(request_data.get('content'), request_data.get('tos'), request_data.get('subject'))
except Exception as e:
logger.info('ERROR: ' + repr(e))
if type_args == 'sms':
sms_config = data.get('sms')
try:
logger.info('data' + repr(request_data))
sms_handle = SendsmsHandle(sms_config.get('api_url'))
for phone in request_data.get('tos').split(','):
logger.info('phone:' + phone)
if phone:
sms_handle.send(phone, content if content else request_data.get('content'))
except Exception as e:
logger.info('ERROR: ' + repr(e))
if type_args == 'im':
im_config = data.get('im')
try:
im_handle = SendImHandle(im_config.get('api_email_url'), im_config.get('api_login_url'),
im_config.get('api_send_url'), im_config.get('username'),
im_config.get('password'), logger)
im_handle.send(content if content else request_data.get('content'), request_data.get('tos'))
except Exception as e:
logger.info('ERROR: ' + repr(e))
# request_data = json.loads(datas,encoding="utf-8")
self.send_response(200)
self.end_headers()
logger.info("send successfully")
def port_to_pid(port):
output = os.popen('lsof -i tcp:' + str(port))
res = output.read()
line = res.split("\n")
if len(line) > 1:
return line[1].split()[1]
else:
return 0
def kill_process(pid):
if pid != 0:
os.popen('kill -9 ' + pid)
print "killed pid: " + pid
def main(argv):
# with open('./cfg.json', 'r') as f:
# data = json.load(f)
port = data["port"];
if len(argv) > 1:
action = argv[1]
else:
action = ""
if action == "" or action == "start":
pid = port_to_pid(port)
if pid != 0:
print "sms provider has started already, pid: " + pid
return
http_server = HTTPServer(('127.0.0.1', port), HTTPHandle)
http_server.serve_forever()
elif action == "stop":
pid = port_to_pid(port)
kill_process(pid)
elif action == "restart":
pid = port_to_pid(port)
kill_process(pid)
http_server = HTTPServer(('127.0.0.1', port), HTTPHandle)
http_server.serve_forever()
class HttpServerDaemon(Daemon.Daemon):
def _run(self):
# with open('./cfg.json', 'r') as f:
# data = json.load(f)
port = data["port"];
http_server = HTTPServer(('0.0.0.0', port), HTTPHandle)
self.logger.info('provider start runing')
http_server.serve_forever()
if __name__ == '__main__':
if sys.getdefaultencoding() != default_encoding:
reload(sys)
sys.setdefaultencoding(default_encoding)
daemon = HttpServerDaemon("./var/smsprovider.pid")
if len(sys.argv) > 1:
action = sys.argv[1]
else:
action = ""
if action == "" or action == "start":
daemon.start()
elif action == "stop":
daemon.stop()
elif action == "restart":
daemon.restart()
elif action == "status":
print daemon.status()
else:
print "Unknown command"
print "usage: %s start|stop|restart|status" % sys.argv[0]
sys.exit(2)
sys.exit(0)
