守护python, [my_py_defend.py]
import datetime
import logging
import subprocess
import time
# 指定要进入的目录和要执行的 Python 程序路径
directory = "D:\\pyworkspace\\project\\mytest"
# 服务名, 没啥用, 打印日志用到, 直接py文件名即可
service_name = 'mytest_main'
# py文件名
python_program = f"{service_name}.py"
# 监听端口, 根据端口判断目标是否存活
l_port = 8081
# 设置日志文件保存的天数
max_log_age = 15
# 创建一个logger
logger = logging.getLogger(f'{service_name}_defend_Logger')
# 设置日志记录级别为DEBUG
logger.setLevel(logging.DEBUG)
# 定义日志文件
log_filename = f'{service_name}_defend.log'
# 创建一个handler,用于写入日志文件
fh = logging.FileHandler(log_filename)
# 再创建一个handler,用于输出到控制台
ch = logging.StreamHandler()
# 定义handler的输出格式
formatter = logging.Formatter('%(asctime)s | %(levelname)s | %(filename)s | %(lineno)d | %(funcName)s | %(message)s')
fh.setFormatter(formatter)
ch.setFormatter(formatter)
# 给logger添加handler
logger.addHandler(fh)
logger.addHandler(ch)
current_date = datetime.date.today()
def get_current_log_first_line(file_path):
with open(file_path, 'r') as file:
lines = file.readlines()
for line_number, line in enumerate(reversed(lines), start=1):
asc_line = len(lines) - line_number + 1
log_date_str = line[:10]
try:
log_date = datetime.datetime.strptime(log_date_str, '%Y-%m-%d').date()
except Exception:
continue
if (current_date - log_date).days >= max_log_age:
return asc_line
# 清理历史日志压缩文件
def clear_log():
current_log_first_line = get_current_log_first_line(log_filename)
if current_log_first_line is None:
return None
# 删除旧日志
with open(log_filename, 'r') as fin:
data = fin.read().splitlines(True)
with open(log_filename, 'w') as fout:
fout.writelines(data[current_log_first_line:])
# 判断端口是否占用
def check_port():
result = subprocess.run(('netstat', '-ano'), capture_output=True, text=True,
creationflags=subprocess.CREATE_NO_WINDOW)
output = result.stdout
lines = output.split('\n')
for line in lines:
if str(l_port) in line and 'LISTENING' in line: # 注意,这里要确保状态是 LISTENING 才表示端口被占用
return True
return False
defend_run_count = 0
is_running = 0
not_is_running = 0
while True:
try:
if check_port():
is_running += 1
logger.info(
f"{service_name} is running, defend_run_count:{defend_run_count}, is_running:{is_running}, not_is_running:{not_is_running}")
else:
not_is_running += 1
logger.info(
f"{service_name} is not running, start it, defend_run_count:{defend_run_count}, not_is_running:{not_is_running}")
# 使用命令行切换目录并执行 Python 程序
command = f"cd /d {directory} && pythonw {python_program}"
subprocess.Popen(command, shell=True)
time.sleep(3)
defend_run_count += 1
if defend_run_count > 1000:
clear_log()
defend_run_count = 0
logger.info('is running for 10000, run clear log')
except Exception:
pass
下面这几个参数根据实际情况修改
# 指定要进入的目录和要执行的 Python 程序路径
directory = "D:\\pyworkspace\\project\\mytest"
# 服务名, 没啥用, 打印日志用到, 直接py文件名即可
service_name = 'mytest_main'
# py文件名
python_program = f"{service_name}.py"
# 监听端口, 根据端口判断目标是否存活
l_port = 8081
# 设置日志文件保存的天数
max_log_age = 15
安装pyinstaller打包exe工具
pip install pyinstaller -i https://mirrors.aliyun.com/pypi/simple/ requests
打包成exe
终端进入[my_py_defend.py]所在目录, 执行以下命令
pyinstaller --onefile -w my_py_defend.py
然后在该目录的dist文件夹或生成同名的exe文件
设置为计划任务
创建任务, 触发改为"启动时", 就是开机就执行, 操作选择该exe文件, 创建完成后运行该任务即可