DB系统预警联系人API

拼搏现实的明天。 2022-08-06 03:09 204阅读 0赞

Author：Skate  
Time:2014/12/16

**DB系统预警联系人API**

在我们维护系统时，需要把系统的报警信息即时传递给相应同学，如果把联系方式直接写到脚本里，对以后的维护变更将埋下祸根，尤其是成百上千的系统。  
为此这里写了个获取联系人信息的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][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'][skate1_163.com], 'email', 'skate1', 'p1'),  
 (2, 2, ['skate2@163.com'][skate1_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'][skate1_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][--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][--load_root_root_10.20.0.55_3306_test6]  
[skate2@163.com,skate6@163.com][skate2_163.com_skate6_163.com],  
\[root@skatedb55 pytest\]\#

优点：  
1.在变更联系人或联系方式不需要修改代码  
2.联系人的相关信息存储在配置中心数据库，为了减少对数据库的查询,默认每天查一次数据库(自己可以指定)，把联系信息放在本地，既提高了速度，也减少了对配置中心的依赖  
3.如果想在变更联系信息及时生效，只需把本地的临时文件"/tmp/contact\_dbinfo"删除即可

contactlist.py:

# -*- 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

大家有好的意见，欢迎提出

------end-------

[load_user_pass_ip_port_dbname]: mailto:load=user/pass@ip:port:dbname
[skate1_163.com]: mailto:'%20rel=
[--load_root_root_10.20.0.55_3306_test6]: mailto:--load=root/root@10.20.0.55:3306:test6
[skate2_163.com_skate6_163.com]: mailto:skate2@163.com,skate6@163.com