#!/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)