334 lines
10 KiB
Python
334 lines
10 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
工具函数模块
|
||
"""
|
||
|
||
import os
|
||
import json
|
||
import subprocess
|
||
import re
|
||
import signal
|
||
import shutil
|
||
from datetime import datetime
|
||
from config import LOG_FILE, PROJECTS_FILE, CRON_BACKUP_DIR, ALERT_CONFIG_FILE
|
||
|
||
|
||
# 进程状态缓存
|
||
process_cache = {}
|
||
|
||
|
||
def log_message(msg):
|
||
"""写入日志"""
|
||
timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
||
log_line = f"[{timestamp}] {msg}\n"
|
||
print(log_line.strip())
|
||
try:
|
||
os.makedirs(os.path.dirname(LOG_FILE), exist_ok=True)
|
||
with open(LOG_FILE, 'a') as f:
|
||
f.write(log_line)
|
||
except:
|
||
pass
|
||
|
||
|
||
def load_projects():
|
||
"""加载项目配置"""
|
||
try:
|
||
with open(PROJECTS_FILE, 'r', encoding='utf-8') as f:
|
||
data = json.load(f)
|
||
return data.get('projects', [])
|
||
except:
|
||
return []
|
||
|
||
|
||
def load_projects_full():
|
||
"""加载完整项目配置（包含server和external_services）"""
|
||
try:
|
||
with open(PROJECTS_FILE, 'r', encoding='utf-8') as f:
|
||
return json.load(f)
|
||
except:
|
||
return {'server': {}, 'projects': [], 'external_services': []}
|
||
|
||
|
||
def save_projects(projects):
|
||
"""保存项目配置（保留 server 和 external_services）"""
|
||
full_data = load_projects_full()
|
||
full_data['projects'] = projects
|
||
with open(PROJECTS_FILE, 'w', encoding='utf-8') as f:
|
||
json.dump(full_data, f, ensure_ascii=False, indent=2)
|
||
|
||
|
||
def check_port(port):
|
||
"""检查端口是否有进程监听"""
|
||
try:
|
||
import socket
|
||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
sock.settimeout(1)
|
||
result = sock.connect_ex(('localhost', port))
|
||
sock.close()
|
||
return result == 0
|
||
except:
|
||
return False
|
||
|
||
|
||
def stop_process(port):
|
||
"""停止占用指定端口的进程"""
|
||
try:
|
||
# 使用 ss 找到占用端口的进程
|
||
result = subprocess.run(f"ss -tlnp | grep :{port}", shell=True, capture_output=True, text=True)
|
||
if result.stdout:
|
||
# 解析 pid
|
||
import re
|
||
match = re.search(r'pid=(\d+)', result.stdout)
|
||
if match:
|
||
pid = int(match.group(1))
|
||
os.kill(pid, signal.SIGTERM)
|
||
log_message(f"已停止进程 {pid} (端口 {port})")
|
||
return 1
|
||
except Exception as e:
|
||
log_message(f"停止进程失败: {e}")
|
||
return 0
|
||
|
||
|
||
def run_command(cmd, cwd, project_id, action):
|
||
"""运行命令"""
|
||
log_message(f"[{project_id}] 执行 {action}: {cmd}")
|
||
try:
|
||
subprocess.Popen(cmd, shell=True, cwd=cwd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||
return True
|
||
except Exception as e:
|
||
log_message(f"命令执行失败: {e}")
|
||
return False
|
||
|
||
|
||
def backup_crontab():
|
||
"""备份当前 crontab"""
|
||
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
|
||
backup_file = os.path.join(CRON_BACKUP_DIR, f'crontab_{timestamp}.txt')
|
||
|
||
try:
|
||
os.makedirs(CRON_BACKUP_DIR, exist_ok=True)
|
||
result = subprocess.run("crontab -l 2>/dev/null", shell=True, capture_output=True, text=True)
|
||
with open(backup_file, 'w') as f:
|
||
f.write(result.stdout)
|
||
log_message(f"crontab 已备份到 {backup_file}")
|
||
except Exception as e:
|
||
log_message(f"备份 crontab 失败: {e}")
|
||
|
||
|
||
def get_system_crons():
|
||
"""获取系统所有 cron 任务"""
|
||
crons = []
|
||
|
||
try:
|
||
result = subprocess.run("crontab -l 2>/dev/null", shell=True, capture_output=True, text=True, timeout=5)
|
||
for line in result.stdout.strip().split('\n'):
|
||
line = line.strip()
|
||
if line and not line.startswith('#'):
|
||
parts = line.split()
|
||
if len(parts) >= 6:
|
||
crons.append({'source': 'user', 'expression': ' '.join(parts[:5]), 'command': ' '.join(parts[5:]), 'line': line})
|
||
except:
|
||
pass
|
||
|
||
try:
|
||
if os.path.exists('/etc/crontab'):
|
||
with open('/etc/crontab', 'r') as f:
|
||
for line in f:
|
||
line = line.strip()
|
||
if line and not line.startswith('#'):
|
||
parts = line.split()
|
||
if len(parts) >= 7:
|
||
crons.append({'source': 'system', 'expression': ' '.join(parts[:5]), 'user': parts[5], 'command': ' '.join(parts[6:]), 'line': line})
|
||
except:
|
||
pass
|
||
|
||
try:
|
||
if os.path.exists('/etc/cron.d'):
|
||
for filename in os.listdir('/etc/cron.d'):
|
||
filepath = os.path.join('/etc/cron.d', filename)
|
||
if os.path.isfile(filepath):
|
||
with open(filepath, 'r') as f:
|
||
for line in f:
|
||
line = line.strip()
|
||
if line and not line.startswith('#'):
|
||
parts = line.split()
|
||
if len(parts) >= 7:
|
||
crons.append({'source': 'cron.d', 'file': filename, 'expression': ' '.join(parts[:5]), 'user': parts[5], 'command': ' '.join(parts[6:]), 'line': line})
|
||
except:
|
||
pass
|
||
|
||
return crons
|
||
|
||
|
||
def parse_cron_expression(expr):
|
||
"""解析 cron 表达式为人类可读格式"""
|
||
parts = expr.split()
|
||
if len(parts) != 5:
|
||
return expr
|
||
|
||
minute, hour, day, month, weekday = parts
|
||
desc = []
|
||
|
||
if minute == '*':
|
||
desc.append('每分钟')
|
||
elif minute.startswith('*/'):
|
||
desc.append(f'每{minute[2:]}分钟')
|
||
else:
|
||
desc.append(f'{minute}分')
|
||
|
||
if hour != '*':
|
||
if hour.startswith('*/'):
|
||
desc.append(f'每{hour[2:]}小时')
|
||
else:
|
||
desc.append(f'{hour}时')
|
||
|
||
if day != '*' and not day.startswith('*'):
|
||
desc.append(f'{day}日')
|
||
|
||
if month != '*' and not month.startswith('*'):
|
||
desc.append(f'{month}月')
|
||
|
||
if weekday != '*' and not weekday.startswith('*'):
|
||
weekdays = ['周日', '周一', '周二', '周三', '周四', '周五', '周六']
|
||
try:
|
||
desc.append(weekdays[int(weekday)])
|
||
except:
|
||
pass
|
||
|
||
return ' '.join(desc) if desc else expr
|
||
|
||
def cron_expression_to_human(expr):
|
||
"""更友好的人类可读格式"""
|
||
parts = expr.split()
|
||
if len(parts) != 5:
|
||
return expr
|
||
|
||
minute, hour, day, month, weekday = parts
|
||
|
||
# 特殊模式
|
||
patterns = [
|
||
('0 4 * * *', '每天凌晨4点'),
|
||
('0 0 * * *', '每天午夜0点'),
|
||
('*/5 * * * *', '每5分钟'),
|
||
('*/10 * * * *', '每10分钟'),
|
||
('*/20 * * * *', '每20分钟'),
|
||
('*/30 * * * *', '每30分钟'),
|
||
('0 */1 * * *', '每小时'),
|
||
('0 */2 * * *', '每2小时'),
|
||
('0 9 * * 1', '每周一9点'),
|
||
('0 9 * * 1-5', '每周一到五9点'),
|
||
]
|
||
|
||
for pattern, desc in patterns:
|
||
if expr == pattern:
|
||
return desc
|
||
|
||
# 通用解析
|
||
return parse_cron_expression(expr)
|
||
|
||
|
||
def guess_cron_name(command):
|
||
"""从命令推断任务名称"""
|
||
if 'openclaw' in command or 'agent' in command:
|
||
return 'OpenClaw Agent'
|
||
elif 'backup' in command:
|
||
return '备份任务'
|
||
elif 'sync' in command:
|
||
return '同步任务'
|
||
elif 'clean' in command or 'cleanup' in command:
|
||
return '清理任务'
|
||
elif 'report' in command:
|
||
return '报告任务'
|
||
elif 'monitor' in command or 'check' in command:
|
||
return '监控任务'
|
||
elif 'git' in command:
|
||
return 'Git 任务'
|
||
return '自定义任务'
|
||
|
||
|
||
def guess_cron_category(command):
|
||
"""从命令推断分类"""
|
||
if 'openclaw' in command:
|
||
return 'agent'
|
||
elif 'backup' in command:
|
||
return 'backup'
|
||
elif 'sync' in command:
|
||
return 'sync'
|
||
elif 'clean' in command or 'cleanup' in command:
|
||
return 'cleanup'
|
||
elif 'report' in command:
|
||
return 'report'
|
||
elif 'git' in command:
|
||
return 'git'
|
||
return 'other'
|
||
|
||
|
||
def sync_crontab_from_db(db):
|
||
"""从数据库同步到系统 crontab"""
|
||
c = db.cursor()
|
||
c.execute("SELECT expression, command, enabled FROM cron_tasks WHERE enabled = 1")
|
||
tasks = c.fetchall()
|
||
|
||
lines = []
|
||
for task in tasks:
|
||
lines.append(f"{task['expression']} {task['command']}")
|
||
|
||
# 写入临时文件
|
||
tmp_file = '/tmp/crontab_sync.tmp'
|
||
with open(tmp_file, 'w') as f:
|
||
f.write('\n'.join(lines) + '\n')
|
||
|
||
# 安装 crontab
|
||
subprocess.run(f"crontab {tmp_file}", shell=True)
|
||
os.remove(tmp_file)
|
||
|
||
# 备份
|
||
backup_crontab()
|
||
|
||
|
||
def get_next_run_time(expression):
|
||
"""获取下次执行时间"""
|
||
try:
|
||
from croniter import croniter
|
||
base = datetime.now()
|
||
cron = croniter(expression, base)
|
||
next_time = cron.get_next(datetime)
|
||
return next_time.strftime('%Y-%m-%d %H:%M:%S')
|
||
except:
|
||
return '无法计算'
|
||
|
||
|
||
def get_cpu_temperatures():
|
||
"""获取 CPU Package 温度"""
|
||
temps = {'package_0': None, 'package_1': None}
|
||
try:
|
||
result = subprocess.run(['sensors'], capture_output=True, text=True, timeout=5)
|
||
output = result.stdout
|
||
# 解析 Package id 0 和 Package id 1
|
||
for line in output.split('\n'):
|
||
if 'Package id 0:' in line:
|
||
match = re.search(r'Package id 0:\s*([+\d.]+)°C', line)
|
||
if match:
|
||
temps['package_0'] = float(match.group(1).replace('+', ''))
|
||
elif 'Package id 1:' in line:
|
||
match = re.search(r'Package id 1:\s*([+\d.]+)°C', line)
|
||
if match:
|
||
temps['package_1'] = float(match.group(1).replace('+', ''))
|
||
except Exception as e:
|
||
log_message(f"获取CPU温度失败: {e}")
|
||
return temps
|
||
|
||
|
||
def load_alert_config():
|
||
"""加载预警配置"""
|
||
try:
|
||
with open(ALERT_CONFIG_FILE, 'r', encoding='utf-8') as f:
|
||
return json.load(f)
|
||
except:
|
||
return {}
|
||
|
||
|
||
def save_alert_config(config):
|
||
"""保存预警配置"""
|
||
with open(ALERT_CONFIG_FILE, 'w', encoding='utf-8') as f:
|
||
json.dump(config, f, ensure_ascii=False, indent=2) |