#!/usr/bin/python
#-*- coding:utf-8 -*-
 
"""
名稱(chēng)：進(jìn)程檢查腳本
作者：wjzhu
時(shí)間：2014-06-30
功能：根據(jù)進(jìn)程名稱(chēng)，判斷進(jìn)程是否存在，執(zhí)行相應(yīng)操作
參數(shù)：p_name：進(jìn)程名稱(chēng)|p_path：進(jìn)程啟動(dòng)路徑
返回值：0:進(jìn)程存在，正常退出|1：進(jìn)程不存在，執(zhí)行命令后，進(jìn)程存在|2：其他異常情況
 
update:2014-07-29 啟動(dòng)進(jìn)程前調(diào)用os.chdir()方法，改變當(dāng)前工作目錄，解決部分進(jìn)程啟動(dòng)時(shí)依賴(lài)工作目錄問(wèn)題
 
"""
 
import os
import sys
 
#初始化進(jìn)程名稱(chēng)和命令路徑
 
#使用參數(shù)方式傳遞程序名稱(chēng)和程序路徑
#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平臺(tái)調(diào)用ps命令/Win平臺(tái)調(diào)用tasklist命令，判斷進(jìn)程是否存在，傳入進(jìn)程名稱(chēng)，返回為查詢(xún)得到的進(jìn)程個(gè)數(shù)
def process_exit(process_name):
  #Linux
  p_checkresp = os.popen('ps aux | grep "' + process_name + '" | grep -v grep').readlines()
 
  #Windows，為避免進(jìn)程名稱(chēng)被截?cái)?，輸出格式為csv，使用tasklist /fo csv
  #p_checkresp = os.popen('tasklist /fo csv | find "' + process_name + '"').readlines()
  return len(p_checkresp)
 
#Linx平臺(tái)調(diào)用os.system方法啟動(dòng)命令/Win平臺(tái)調(diào)用os.startfile方法啟動(dòng)命令，傳入命令路徑，無(wú)返回值
def process_exec(process_path):
  #將工作目錄切換到啟動(dòng)腳本所在目錄，解決部分進(jìn)程啟動(dòng)時(shí)依賴(lài)工作目錄問(wèn)題
  os.chdir(os.path.dirname(process_path))
 
  #Linxu
  os.system(process_path)
 
  #Windows
  #os.startfile(process_path)
 
#主函數(shù)
if __name__ == '__main__':
 
  #查詢(xún)進(jìn)程個(gè)數(shù)大于1，返回0，不做任何操作，退出
  if process_exit(p_name) >= 1:
    print 0
    sys.exit(0)
     
  #查詢(xún)進(jìn)程個(gè)數(shù)等于0
  elif process_exit(p_name) == 0:
    #執(zhí)行啟動(dòng)命令
    process_exec(p_path)
    #查詢(xún)進(jìn)程個(gè)數(shù)大于1，返回1，啟動(dòng)成功，退出
    if process_exit(p_name) >= 1:
      print 1
      sys.exit(0)
    #啟動(dòng)失敗，返回2，退出
    else:
      print 2
      sys.exit(0)
       
  #其他問(wèn)題，返回2，退出
  else:
    print 2
    sys.exit(0)