DB系统预警联系人API

CREATE TABLE `db_alertcontact` (
`id` INT(11) NULL DEFAULT NULL,
`levelid` INT(11) NULL DEFAULT NULL COMMENT 'contact level',
`contact` VARCHAR(50) NULL DEFAULT NULL COMMENT 'email or phone information',
`type` VARCHAR(50) NULL DEFAULT NULL COMMENT 'phone/email',
`username` VARCHAR(100) NULL DEFAULT NULL,
`group` VARCHAR(80) NULL DEFAULT NULL COMMENT 'contact group'
)
COLLATE='utf8_general_ci'
ENGINE=InnoDB;

CREATE TABLE `db_alertlevel` (
`id` INT(11) NULL DEFAULT NULL,
`levelname` VARCHAR(50) NULL DEFAULT NULL COMMENT 'info/warn/err'
)
COLLATE='utf8_general_ci'
ENGINE=InnoDB;

[root@skatedb55 pytest]# python contactlist.py --help
usage: Contanct API v0.1 ,(C) Copyright Skate 2014 [-h] --group GROUP --type
TYPE --level LEVEL
[--interval INTERVAL]
[--load LOAD]

optional arguments:
-h, --help show this help message and exit
--group GROUP = The contact group
--type TYPE = The mode of contact
--level LEVEL = alarm level,info/warn/err
--interval INTERVAL = Database query interval(s)
--load LOAD = The configure center database,eg:
load=user/pass@ip:port:dbname
[root@skatedb55 pytest]#

INSERT INTO `db_alertcontact` (`id`, `levelid`, `contact`, `type`, `username`, `group`) VALUES
(1, 1, 'skate1@163.com', 'email', 'skate1', 'p1'),
(2, 2, 'skate2@163.com', 'email', 'skate2', 'p2'),
(3, 1, '1300000000', 'phone', 'skate3', 'p2'),
(4, 1, '1311111111', 'phone', 'skate4', 'p2'),
(5, 1, '1322222222', 'phone', 'skate5', 'p2'),
(6, 2, 'skate6@163.com', 'email', 'skate6', 'p2');


INSERT INTO `db_alertlevel` (`id`, `levelname`) VALUES
(1, 'info'),
(2, 'warn'),
(3, 'error');


[root@skatedb55 pytest]# python contactlist.py --group=p2 --type=phone --level=info --interval=10--load=root/root@10.20.0.55:3306:test6
1300000000,1311111111,1322222222,
[root@skatedb55 pytest]#

[root@skatedb55 pytest]# python contactlist.py --group=p2 --type=email --level=warn --interval=10--load=root/root@10.20.0.55:3306:test6
skate2@163.com,skate6@163.com,
[root@skatedb55 pytest]#

# -*- coding: utf-8 -*-
#!/usr/bin/python
#
# Author:Skate
# Time:2014/12/10
# Function: Contact API

import MySQLdb,sys
import argparse
import os
import datetime

class database:
 def __int__(self,host,user,passwd,port,dbname):
 self.conn = None
 pass
 def conn(self,host,user,passwd,port,dbname):
 self.host=host
 self.user=user
 self.passwd=passwd
 self.port=port
 self.dbname=dbname
 try: 
 self.conn = MySQLdb.connect(host=self.host, user=self.user, passwd=self.passwd, db=self.dbname,port=self.port) 
 
 except MySQLdb.Error, e: 
 print "MySQL Connect Error: %s" % (e.args[1]) 
 return self.conn 
 def closeConn(self):
 self.conn.close()
 def execute(self,sql,param):
 if self.conn==None or self.conn.open==False :
 return -1
 sys.exit
 cur = self.conn.cursor()
 cur.execute(sql,param)
 self.closeConn()
 return cur

def contactlist(group,type,level,host,user,passwd,port,dbname,interval=86400):
 tfile='/tmp/contact_dbinfo'
 list=''
 if os.path.isfile(tfile):
 a1=datetime.datetime.fromtimestamp(os.path.getctime(tfile))
 a2=datetime.datetime.now()
 diffsec = (a2 - a1).seconds
 if diffsec > interval:
 os.remove(tfile) 
 f=open(tfile,'a') 
 db=database()
 db.conn(host,user,passwd,port,dbname)
 sql="select t.contact,t.username,t.group,t.`type`,l.levelname from db_alertcontact t , db_alertlevel l where t.levelid=l.id and l.levelname=%s and t.group=%s and t.`type`=%s"
 param=(level,group,type)
 cur=db.execute(sql,param)
 results=cur.fetchall()
 for row in results:
 if row[3] =='phone':
 #for r in row:
 list = list + row[0] + ','
 elif row[3] == 'email':
 #for r in row:
 list = list + row[0] + ','
 if type =='phone':
 f.write('phonelist='+ group + ':' + list + '\n')
 f.close()
 elif type == 'email':
 f.write('emaillist='+ group + ':' +list + '\n')
 f.close()
 else:
 strsearch = type + 'list='+ group
 istype = os.popen('cat '+ tfile +' | grep ' + strsearch + ' | wc -l').readline().strip()
 if int(istype) > 0: 
 line = os.popen('cat '+ tfile +' | grep ' + strsearch).readline().strip()
 b=line.split('=')
 a=b[1].split(":")
 if b[0]=='phonelist':
 list=a[1]
 elif b[0]=='emaillist':
 list=a[1]
 elif int(istype) < 1:
 f=open(tfile,'a')
 db=database()
 db.conn(host,user,passwd,port,dbname)
 sql="select t.contact,t.username,t.group,t.`type`,l.levelname from db_alertcontact t , db_alertlevel l where t.levelid=l.id and l.levelname=%s and t.group=%s and t.`type`=%s"
 param=(level,group,type)
 cur=db.execute(sql,param)
 results=cur.fetchall()
 #list=''
 for row in results:
 if row[3] =='phone':
 list = list + row[0] + ','
 elif row[3] == 'email':
 list = list + row[0] + ','
 if type =='phone':
 f.write('phonelist='+ group + ':' + list + '\n')
 f.close()
 elif type == 'email':
 f.write('emaillist='+ group + ':' + list + '\n')
 f.close()

 else:
 f=open(tfile,'a')
 db=database()
 db.conn(host,user,passwd,port,dbname)
 sql="select t.contact,t.username,t.group,t.`type`,l.levelname from db_alertcontact t , db_alertlevel l where t.levelid=l.id and l.levelname=%s and t.group=%s and t.`type`=%s"
 param=(level,group,type)
 cur=db.execute(sql,param)
 results=cur.fetchall()

 for row in results:
 if row[3] =='phone':
 #for r in row:
 list = list + row[0] + ','
 elif row[3] == 'email':
 #for r in row:
 list = list + row[0] + ','

 if type =='phone':

 f.write('phonelist='+ group + ':' + list + '\n')
 f.close()
 elif type == 'email':
 f.write('emaillist='+ group + ':' + list + '\n')
 f.close()

 return list

if __name__ == "__main__":
 parser = argparse.ArgumentParser("Contanct API v0.1 ,(C) Copyright Skate 2014")
 parser.add_argument('--group', action='store', dest='group',required=True,
 help=" = The contact group")

 parser.add_argument('--type', action='store', dest='type',required=True,
 help=" = The mode of contact")

 parser.add_argument('--level', action='store', dest='level',required=True,
 help=" = alarm level,info/warn/err")

 parser.add_argument('--interval', action='store', dest='interval',type=int,default=86400,
 help=" = Database query interval")

 parser.add_argument('--load', action='store', dest='load',default='',
 help=" = The configure center database,eg: \n load=user/pass@ip:port:dbname")

 results = parser.parse_args()

 load = results.load
 group = results.group
 type = results.type
 level = results.level
 interval = results.interval 

 if (load !=''):
 user_info,url = load.split("@")
 host,port,db = url.split(":")
 port=int(port)
 user,passwd = user_info.split("/",1)

 str = contactlist(group,type,level,host,user,passwd,port,db,interval)
 print str