Python對Windows服務進行監控
阿新 • • 發佈：2019-01-08
#-*- encoding: utf-8 -*-
import logging
import wmi
import os
import time
from ConfigParser import ConfigParser
import smtplib
from email.mime.text import MIMEText
import socket
from datetime import datetime
import re
import sys
import time
import string
import psutil
import threading
from threading import Timer
import logging
# 建立一個logger
logger = logging.getLogger('Monitor')
logger.setLevel(logging.DEBUG)
# 建立一個handler，用於寫入日誌檔案
fh = logging.FileHandler('test.log')
fh.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
fh.setFormatter(formatter)
logger.addHandler(fh)
reload(sys) # Python2.5 初始化後會刪除 sys.setdefaultencoding 這個方法，我們需要重新載入
sys.setdefaultencoding('utf-8')
def send_mail(to_list,sub,content):
CONFIGFILE = 'config.ini'
config = ConfigParser()
config.read(CONFIGFILE)
mail_host=config.get('Mail','mail_host') #使用的郵箱的smtp伺服器地址，這裡是163的smtp地址
mail_user=config.get('Mail','mail_user') #使用者名稱
mail_pass=config.get('Mail','mail_pass') #密碼
mail_postfix=config.get('Mail','mail_postfix') #郵箱的字尾，網易就是163.com
me=sub+"<"+mail_user+"@"+mail_postfix+">"
msg = MIMEText(content,_subtype='plain',_charset='utf-8')
msg['Subject'] = sub
msg['From'] = me
msg['To'] = ";".join(to_list) #將收件人列表以‘；’分隔
try:
server = smtplib.SMTP()
server.connect(mail_host) #連線伺服器
server.login(mail_user,mail_pass) #登入操作
server.sendmail(me, to_list, msg.as_string())
server.close()
return True
except Exception, e:
print str(e)
logger.info(str(e))
return False
#讀取配置檔案中的程序名和系統路徑，這2個引數都可以在配置檔案中修改
ProList = []
#定義一個列表
c = wmi.WMI()
#獲取程序所用記憶體
def countProcessMemoey(processName):
try:
CONFIGFILE = 'config.ini'
config = ConfigParser()
config.read(CONFIGFILE)
pattern = re.compile(r'([^\s]+)\s+(\d+)\s.*\s([^\s]+\sK)')
cmd = 'tasklist /fi "imagename eq ' + processName + '"' + ' | findstr.exe ' + processName
result = os.popen(cmd).read()
resultList = result.split("\n")
totalMem = 0.0
totalCpu = 0.0
print "*" * 80
for srcLine in resultList:
srcLine = "".join(srcLine.split('\n'))
if len(srcLine) == 0:
break
m = pattern.search(srcLine)
if m == None:
continue
#由於是檢視python程序所佔記憶體，因此通過pid將本程式過濾掉
if str(os.getpid()) == m.group(2):
continue
p = psutil.Process(int(m.group(2)))
cpu = p.cpu_percent(interval=1)
ori_mem = m.group(3).replace(',','')
ori_mem = ori_mem.replace(' K','')
ori_mem = ori_mem.replace(r'\sK','')
memEach = string.atoi(ori_mem)
totalMem += (memEach * 1.0 /1024)
totalCpu += cpu
print 'ProcessName:'+ m.group(1) + '\tPID:' + m.group(2) + '\tmemory size:%.2f'% (memEach * 1.0 /1024), 'M' + ' CPU:'+str(cpu)+'%'
print 'ProcessName:'+ m.group(1)+' TotalMemory:'+str(totalMem)+'M'+' totalCPU:'+str(totalCpu)+'%'
logger.info('ProcessName:'+ m.group(1)+' TotalMemory:'+str(totalMem)+'M'+' totalCPU:'+str(totalCpu)+'%')
print "*" * 80
if totalMem> float(config.get('MonitorProcessValue','Memory')):
print 'Memory Exceed!'
IP = socket.gethostbyname(socket.gethostname())
now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
subject = IP +' ' + processName + '記憶體使用量過高！'
content = now + ' ' + IP +' ' + processName + '記憶體使用量過高,達到'+str(totalMem) +'M\n請儘快處理！'
logger.info(processName +'記憶體使用量過高,達到'+str(totalMem) +'M')
send_mail([' 所用配置檔案config.ini
[MonitorProgramPath]
ProgramPath: C:\Windows\System32\services.exe|C:\Program Files (x86)\Google\Chrome\Application\chrome.exe
[MonitorProcessName]
ProcessName: services.exe|chrome.exe
[MonitorProcessValue]
Memory:5000.0
CPU:50.0
Time:60
[Mail]
mail_host: smtp.163.com
mail_user:
mail_pass:
mail_postfix: 163.com