本文写于 2014年。
当前项目,进程守护一般使用 Supervisor。

需求说明

利用Hyperic调用Python,实现进程守护。

调用操作系统方法获取进程信息,判断进程是否存在。

Linux和Windows均支持,区别在于获取进程信息和启动进程的方法不同。

需求实现

代码如下:


#!/usr/bin/python
#-*- coding:utf-8 -*-

"""
名称:进程检查脚本
作者:wjzhu
时间:2014-06-30
功能:根据进程名称,判断进程是否存在,执行相应操作
参数:p_name:进程名称|p_path:进程启动路径
返回值:0:进程存在,正常退出|1:进程不存在,执行命令后,进程存在|2:其他异常情况

update:2014-07-29 启动进程前调用os.chdir()方法,改变当前工作目录,解决部分进程启动时依赖工作目录问题

"""

import os
import sys

#初始化进程名称和命令路径

#使用参数方式传递程序名称和程序路径
#p_name = sys.argv[1]
#p_path = sys.argv[2]

#Linux
p_name = "redis"
p_path = "/usr/local/redis-2.8.11/src/redis-server/usr/local/redis-2.8.11/redis.conf"

#Windows
#p_name = "filezilla.exe"
#p_path = "C:\\Program Files (x86)\\FileZilla FTP Client\\filezilla.exe"



#Linux平台调用ps命令/Win平台调用tasklist命令,判断进程是否存在,传入进程名称,返回为查询得到的进程个数
def process_exit(process_name):
#Linux
p_checkresp = os.popen('ps aux | grep "' + process_name + '" | grep -v grep').readlines()

#Windows,为避免进程名称被截断,输出格式为csv,使用tasklist /fo csv
#p_checkresp = os.popen('tasklist /fo csv | find "' + process_name + '"').readlines()
return len(p_checkresp)

#Linx平台调用os.system方法启动命令/Win平台调用os.startfile方法启动命令,传入命令路径,无返回值
def process_exec(process_path):
#将工作目录切换到启动脚本所在目录,解决部分进程启动时依赖工作目录问题
os.chdir(os.path.dirname(process_path))

#Linxu
os.system(process_path)

#Windows
#os.startfile(process_path)

#主函数
if __name__ == '__main__':

#查询进程个数大于1,返回0,不做任何操作,退出
if process_exit(p_name) >= 1:
print 0
sys.exit(0)

#查询进程个数等于0
elif process_exit(p_name) == 0:
#执行启动命令
process_exec(p_path)
#查询进程个数大于1,返回1,启动成功,退出
if process_exit(p_name) >= 1:
print 1
sys.exit(0)
#启动失败,返回2,退出
else:
print 2
sys.exit(0)

#其他问题,返回2,退出
else:
print 2
sys.exit(0)