#!/usr/bin/env python#coding:utf-8import os,re,sysdef numSort(): global game_gen_path,noStart,alreadyStart alreadyStart = set() allservice = set() game_gen_path = '/data/bz-tw-and/' command="ps aux | grep bz-tw-and-s | awk '{print $11}' | awk -F'/' '{print $2}' | grep -v '^$' | cut -d'-' -f 4" service = os.popen('%s' % command,'r').readlines() #popen需要用表格式形式阅读，会有\n出现，需要strip()去除 for service in service: service_name = service.strip().strip()[1:] alreadyStart.add(service_name) for qu_name in os.listdir(game_gen_path): if re.match('s\d+',qu_name): allservice.add(qu_name[1:]) noStart = sorted(list(map(int,alreadyStart^allservice))) alreadyStart = sorted(list(map(int,alreadyStart)))def outer(fun): def wrapper(): fun() for qu_name in qu_service: excutePath = game_gen_path + 's' + str(qu_name) + '/bin/' for fileName in os.listdir(excutePath): if fileName == script: print('##### s\033[31m%s\033[0m now %s... #####' % (qu_name,status)) os.system('sh %s' % (excutePath + fileName)) return wrapper@outerdef start(): global qu_service,script,status numSort() qu_service = noStart script = 'start.sh' status = 'start'@outerdef stop(): global qu_service,script,status numSort() qu_service = alreadyStart script = 'stop.sh' status = 'stop'def status(): command = "ps aux | grep bz-tw-and-s | awk '{print $11}' | awk -F'/' '{print $2}' | grep -v '^$' | sort -t's' -k 2n | awk -F's' '{print $2}'" service = os.popen('%s' % command, 'r').readlines() if service == []: print '进程已经全部关闭' else: for process in service: fushu = process.strip() print('bz-tw-and-s'+'\033[31m%s\033[0m' % fushu)def callable(input_key): script_name = sys.argv[0] try: service_option = {'start': start, 'stop': stop, 'status': status} return service_option[input_key]() except KeyError as e: print e print ('Example: %s start|stop|status' % script_name)if __name__ == '__main__': callable(sys.argv[1])