supervision.py

import os
import time
import getopt
import sys
import os
from pathlib import Path
import schedule

Mib = 1024 * 1024


def kill_process(SupervisionFileName, FileName):
	info = os.popen("ps -ef | grep {0}".format(FileName)).readlines()
	if info:
		for i in info:
			if SupervisionFileName not in i:
				try:
					j = i.split()[1].strip()
					os.system("kill -9 " + j)
					print("kill process! {0}".format(i))
				except Exception as e:
					print("Err occur when kill process! e = {0}, info = {1}".format(e, i))


def check_log_file(LogFileName):
	log_file_size = os.path.getsize(LogFileName)
	print("log file size is: {0}".format(log_file_size))
	if log_file_size > LogFileMaxSize * Mib:
		os.system("rm -rf {0}".format(LogFileName))


def job():

	# 跳转到新的文件夹
	new_dir = os.path.dirname(FileName)
	if len(new_dir) != 0:
		os.chdir(new_dir)

	log_file = r'{0}.log'.format(FileName)

	# 检查日志文件 如果太大 那就删除日志文件
	check_log_file(log_file)

	# 日志信息（日志文件最后一条数据)
	log_info = None
	# 上一条日志的信息
	last_log = None
	same_count = 0
	stop_loop = False

	# 循环检查10次数据
	for i in range(10):
		try:
			with open(log_file, "r") as f:
				#  获取u最后一条日志数据
				file = f.readlines()
				log_info = file[-1]
				print("log info: {0}".format(log_info))
				start_logs = file[:100]
				log_infos = file[-50:]
				file.clear()

				# # 启动时候就没有手续费 可以直接退出
				# for start_log in start_logs:
				# 	if "fees" in start_log:
				# 		exit("Inability to pay some fees.")

				# 日志不同 说明正常 反之需要处理
				if log_info != last_log:
					last_log = log_info
					same_count = 0
				else:
					same_count += 1

				# 如果有异常 那么就重新启动
				if (same_count >= 5):

					print("warn: mining abnormal. Now restart mining, and please wait a moment.")
					start(FileName, SupervisionFileName)
					break

				# 如果日志中有报错信息或是手续费不足 那么退出
				for log in log_infos:
					# print("log:", log)
					if ("fees" in log) or ("Err" in log) or( "err" in log):
						print("Inability to pay some fees or some error occur.")
						start(FileName, SupervisionFileName)
						stop_loop = True
						break

		except Exception as e:
			print("warn: log file does not exists. Now restart mining, and please wait a moment.", e)
			start(FileName, SupervisionFileName)
			break

		if stop_loop:
			break

		# 每5秒钟去检查一次日志
		time.sleep(5)


def start(FileName, SupervisionFileName):

	# 改变当前路径
	print("filename:", FileName)
	new_dir = os.path.dirname(FileName)
	print("new_dir:", new_dir, len(new_dir))
	if len(new_dir) != 0:
		os.chdir(new_dir)

	kill_process(SupervisionFileName, FileName)
	print("stop mining success!")
	time.sleep(5)
	# os.system(r'./{0} > {1}.log 2>&1 &'.format(FileName, FileName))
	os.system(r'{0} > {1}.log 2>&1 &'.format(FileName, FileName))
	print("start mining success!")


def stop(FileName, SupervisionFileName):
	info = os.popen("ps -ef | grep {0}".format(FileName)).readlines()#.extend(os.popen("ps -ef | grep supervision.py").readlines())
	info1 = os.popen("ps -ef | grep {0}\.py".format(SupervisionFileName)).readlines()
	info.extend(info1)
	if info:
		for i in info:
			print(i)
			try:
				j = i.split()[1].strip()
				os.system("kill -9 " + j)
				print("kill process! {0}".format(i))
			except Exception as e:
				print("Err occur when kill process! e = {0}, info = {1}".format(e, i))


def first_start():

	global FileName
	global LogFileMaxSize
	global StopMining
	opts, args = getopt.getopt(sys.argv[1:], "", ["stop", "mining=", "log-max-size="])
	print(opts)

	# 检查是否有文件参数
	for opt, arg in opts:
		if opt == "--mining" and len(arg) != 0 and "--" not in arg:
			FileName = arg
			break
	else:
		exit("please add '--mining' in your command line, and the value can not empty!")

	# 检查是否有log文件大小限制值 如果输入零则使用默认值
	for opt, arg in opts:
		if opt == "--log-max-size" and int(arg) != 0:
			LogFileMaxSize = int(arg)
			break

	# 检查是否有停止命令 有的话直接停止
	for opt, arg in opts:
		if opt == "--stop":
			stop(FileName, SupervisionFileName)
			StopMining = True
			exit("stop mining!")
			break
	else:
		start(FileName, SupervisionFileName)


if __name__ == "__main__":
	# 监控节点 放在与挖矿软件相同的文件夹中

	# 使用方法：
		# 开启挖矿： python3 supervision.py --mining 挖矿软件名称 [--log-max-size 数值(默认值是20)] (Mib为基本单位， 比如数值为1， 代表log文件最大空间允许值是1Mib)
		# 结束挖矿： python3 supervision.py --mining 挖矿软件名称 --stop

	FileName = ""  # 挖矿软件名称
	LogFileMaxSize = 100  # 日志文件大小最大允许值(多少Mib)
	SupervisionFileName = Path(__file__).name.split(".")[0]
	StopMining = False

	# 检查命令行参数， 并启动挖矿
	first_start()

	if not StopMining:
		# 每十分钟去执行一次
		schedule.every(2).minutes.do(job)
		# schedule.every().seconds.do(job)

		while True:
			schedule.run_pending()