python实现模拟用户监控直播服务

#!/usr/bin/env python
# -*- coding:utf-8 -*-
#  author: Dily
#  Email: [e-mail]
"""功能: 模拟用户监控直播平台服务存活"""

import logging
import re
import time
import subprocess
import smtplib
import os
import string
from email.mime.text import MIMEText

def mylog_init():
    """初始化日志格式"""
    logging.basicConfig(level=logging.DEBUG,
    format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s:%(name)s:%(message)s    ',
    datefmt='%a, %d %b %Y %H:%M:%S',
    filename='/work/dily/live/rtmp.log',
    filemode='w+')

def send_mail(sub,content):
  """发送报警邮件"""
  # 收件人列表
  mailto_list=['[e-mail]']
  # 使用的邮箱的smtp服务器地址
  mail_host="mail.xxx.com"
  # 用户名
  mail_user="xxxxxx"
  # 密码
  mail_pass="xxxxxx"
  # 邮箱主域名
  mail_postfix="xxx.com"

  me="Rtmp_Check_Mail"+"<"+mail_user+"@"+mail_postfix+">"
  msg = MIMEText(content,_subtype='plain')
  msg['Subject'] = sub
  msg['From'] = me
  # 将收件人列表以';'分隔
  msg['To'] = ";".join(mailto_list)
  try:
    server = smtplib.SMTP()
    # 连接邮件服务器
    server.connect(mail_host,"25")
    # 登录邮件服务器
    server.login("xxxxxx",mail_pass)
    server.sendmail(me, mailto_list, msg.as_string())
    server.close()
    return True
  except Exception, e:
    print str(e)
    logging.error(str(e))
    return False

def rtmp_logtime():
  """计算logtime时间"""
  mytime = list(time.strftime('%Y%m%d%H%M',time.localtime(time.time())))
  if mytime[-1] == '0' or mytime[-1] == '1' or mytime[-1] == '2' or mytime[-1] == '3' or mytime[-1] == '4':
    mytime[-1] = '0'
    logtime = ''.join(mytime)
    return logtime
  else:
    mytime[-1] = '5'
    logtime = ''.join(mytime)
    return logtime

def rtmp_domain():
  """截取rtmp实时URL"""
  result = 'sudo pssh -x "-p12321" -t 60 -H IP -q -P "cd /cache/logs/acc_log/ && ls -t |grep s.xxx.com_access |head -1 | xargs cat | grep PLAY | tail -1"'
  logging.info(result)
  rtmp_res = subprocess.Popen(result,shell=True,stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
  result_domain = rtmp_res.stdout.readlines()
  logging.info(result_domain)
  C = r'([a-zA-z]+://[^\"\s]+)'
  find = re.compile(C)
  m = find.search(str(result_domain))
  try:
    return m.group(1)
  except AttributeError,e:
    print '没有最新日志可收集'
    logging.error('没有最新日志可收集')

def rtmp_ck():
  """检查边缘节点服务是否正常"""
  result = '/usr/local/bin/rtmp_check %s' % (str(rtmp_domain()))
  logging.info('rtmp check cmd is : %s',result)
  rtmp_res = subprocess.Popen(result,shell=True,stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
  error = []
  pattern = re.compile(
    r"((?:(?:25[0-5]|2[0-4]\d|((1\d{2})|([1-9]?\d)))\.){3}(?:25[0-5]|2[0-4]\d|((1\d{2})|([1-9]?\d))))")
  for line in rtmp_res.stdout.readlines():
    logging.info(' %s',line.strip())
    if 'ERROR' in line:
      error.append(line.strip())
  logging.debug(error)
  if error:
    if send_mail('Fastweb Rtmp Check FAILURE','\n'.join(error)):
      logging.info('mail send success!')
    else:
      logging.error('mail send faild!')
  else:
    logging.info('rtmp check all success!')

def main():
  """主函数"""
  mylog_init()
  rtmp_ck()

if __name__ == '__main__':
  main()