脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Python - 代码讲解Python对Windows服务进行监控

代码讲解Python对Windows服务进行监控

2021-01-15 00:12脚本之家 Python

本篇文章给大家分享了通过Python对Windows服务进行监控的实例代码,对此有兴趣的朋友可以学习参考下。

我们首先来看下python的全部代码,大家可以直接复制后测试:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
#-*- 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(['sunwei_work@163.com','sunweiworkplace@gmail.com'],subject, content)
    if totalCpu > float(config.get('MonitorProcessValue','CPU')):
      print 'CPU Exceed!'
      IP = socket.gethostbyname(socket.gethostname())
      now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
      subject = IP +' ' + processName + 'CPU使用率过高!'
      content = now + ' ' + IP +' ' + processName + 'CPU使用率过高,达到'+str(totalCpu)+'%\n请尽快处理!'
      logger.info(processName +'CPU使用率过高,达到'+str(totalMem) +'M')
      send_mail(['sunwei_work@163.com','sunweiworkplace@gmail.com'],subject, content)
  except Exception, e: 
    print str(e)
    logger.info(str(e))  
  
#判断进程是否存活
def judgeIfAlive(ProgramPath,ProcessName):
  try:
    print datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    for process in c.Win32_Process(): 
      ProList.append(str(process.Name)) 
    #把所有任务管理器中的进程名添加到列表
 
    if ProcessName in ProList:
      countProcessMemoey(ProcessName) 
    #判断进程名是否在列表中,如果是True,则所监控的服务正在 运行状态,
    #打印服务正常运行
      print '' 
      print ProcessName+" Server is running..."
      print '' 
      logger.info(ProcessName+" Server is running...")
    else
      #如果进程名不在列表中,即监控的服务挂了,则在log文件下记录日志
      #日志文件名是以年月日为文件名
      IP = socket.gethostbyname(socket.gethostname())
      now = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
      subject = IP +' ' + ProcessName + '已停止运行!'
      logger.info( ProcessName + '已停止运行!')
      content = now + ' ' + IP +' ' + ProcessName + '已停止运行!' +'\n请尽快处理!'
      send_mail(['sunwei_work@163.com','sunweiworkplace@gmail.com'],subject, content)
      print ProcessName+' Server is not running...'
      #打印服务状态
      logger.info('\n'+'Server is not running,Begining to Restart Server...'+'\n'+(time.strftime('%Y-%m-%d %H:%M:%S --%A--%c', time.localtime()) +'\n'))
      #写入时间和服务状态到日志文件中
      os.startfile(ProgramPath) 
      #调用服务重启
      logger.info(ProcessName+'Restart Server Success...'+'\n'+time.strftime('%Y-%m-%d %H:%M:%S --%A--%c', time.localtime()))
      print ProcessName+'Restart Server Success...'
      print time.strftime('%Y-%m-%d %H:%M:%S --%A--%c', time.localtime()) 
    del ProList[:] 
    #清空列表,否则列表会不停的添加进程名,会占用系统资源
  except Exception, e: 
    print str(e)
    logger.info(str(e))  
def startMonitor(ProgramPathDict,ProcessNameDict) : 
  for i in range(0,len(ProcessNameDict)):
    judgeIfAlive(ProgramPathDict[i],ProcessNameDict[i])
if __name__=="__main__"
  CONFIGFILE = 'config.ini'
  config = ConfigParser() 
  config.read(CONFIGFILE) 
  ProgramPathDict = config.get('MonitorProgramPath','ProgramPath').split("|"
  ProcessNameDict = config.get('MonitorProcessName','ProcessName').split("|")
  while True
    startMonitor(ProgramPathDict,ProcessNameDict) 
    time.sleep(int(config.get('MonitorProcessValue','Time')))

所用配置文件config.ini

?
1
2
3
4
5
6
7
8
9
10
11
12
13
[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

以上就是本次小编整理的关于Python对Windows服务进行监控的全部代码内容,感谢你对服务器之家的支持。

延伸 · 阅读

精彩推荐