Update vps_monitor.py

vps_monitor.py

@@ -52,13 +52,18 @@ def get_vps_configs():
             }
             configs.append(config)
         
+        # 添加调试输出
         logger.info(f"Config {index}: {hostname}, {username}, {script_paths}")
+        print(f"Config {index}: {hostname}, {username}, {script_paths}")
+        sys.stdout.flush()
         
         index += 1
     return configs
 
 def check_and_run_script(config):
     logger.info(f"Checking VPS {config['index']}: {config['hostname']} - {config['script_path']}")
+    print(f"Checking VPS {config['index']}: {config['hostname']} - {config['script_path']}")
+    sys.stdout.flush()
     client = None
     try:
         client = paramiko.SSHClient()
@@ -75,25 +80,24 @@ def check_and_run_script(config):
         script_path = config['script_path']
         script_name = os.path.basename(script_path)
         
+        # 检查脚本是否正在运行 (使用 ps 命令，适用于FreeBSD)
         check_command = f"ps aux | grep {script_name} | grep -v grep"
         stdin, stdout, stderr = client.exec_command(check_command)
         if stdout.read():
             status = "Running"
             logger.info(f"Script {script_name} is running on {config['hostname']}")
         else:
-            logger.warning(f"Script {script_name} not running on {config['hostname']}. Executing restart script.")
-            restart_command = f"/bin/sh {script_path}"
+            logger.info(f"Script {script_name} not running on {config['hostname']}. Executing restart script.")
+            restart_command = f"/bin/sh {script_path}"  # 使用完整路径的sh
             stdin, stdout, stderr = client.exec_command(restart_command)
             exit_status = stdout.channel.recv_exit_status()
             
             if exit_status == 0:
                 status = "Restarted"
                 logger.info(f"Restart script {script_name} executed successfully on {config['hostname']}")
-                output = stdout.read().decode('utf-8')
-                logger.info(f"Restart output: {output}")
             else:
                 error_output = stderr.read().decode('utf-8')
-                status = "Error"
+                status = f"Error: {error_output}"
                 logger.error(f"Error executing restart script {script_name} on {config['hostname']}: {error_output}")
         
         key = f"{config['hostname']}:{script_name}"
@@ -120,15 +124,20 @@ def check_and_run_script(config):
             client.close()
             logger.info(f"SSH connection closed for VPS {config['index']}: {config['hostname']}")
         logger.info(f"Finished checking VPS {config['index']}: {config['hostname']} - {script_name}")
+        print(f"Finished checking VPS {config['index']}: {config['hostname']} - {script_name}")
+        sys.stdout.flush()
 
 def check_all_vps():
     logger.info("Starting VPS check")
+    print("Starting VPS check")
+    sys.stdout.flush()
     vps_configs = get_vps_configs()
     for config in vps_configs:
         check_and_run_script(config)
     logger.info("Finished VPS check")
     for key, status in vps_status.items():
-        logger.info(f"VPS {status['index']} - {key}: Status: {status['status']}, Username: {status['username']}, Script: {status['script_name']}")
+        print(f"VPS {status['index']} - {key}: Status: {status['status']}, Username: {status['username']}, Script: {status['script_name']}")
+    sys.stdout.flush()
 
 @app.route('/')
 def index():
@@ -175,24 +184,37 @@ def main():
     global start_time
     start_time = time.time()
     
-    logger.info("===== VPS monitoring script is starting =====")
+    print("===== VPS monitoring script is starting =====")
+    sys.stdout.flush()
+    logger.info("===== VPS monitoring script started =====")
     
     flask_thread = Thread(target=run_flask)
     flask_thread.start()
     logger.info("Flask server started in background")
+    print("Flask server started in background")
+    sys.stdout.flush()
 
     vps_configs = get_vps_configs()
     logger.info(f"Found {len(vps_configs)} VPS configurations")
+    print(f"Found {len(vps_configs)} VPS configurations")
+    sys.stdout.flush()
     for config in vps_configs:
         logger.info(f"VPS configured: {config['hostname']} - {config['script_path']}")
+        print(f"VPS configured: {config['hostname']} - {config['script_path']}")
+        sys.stdout.flush()
 
     logger.info("Running initial VPS check")
+    print("Running initial VPS check")
+    sys.stdout.flush()
     check_all_vps()
 
     schedule.every(15).minutes.do(check_all_vps)
     logger.info("Scheduled VPS check every 15 minutes")
+    print("Scheduled VPS check every 15 minutes")
+    sys.stdout.flush()
     
-    logger.info("===== VPS monitoring script is running =====")
+    print("===== VPS monitoring script is running =====")
+    sys.stdout.flush()
     
     heartbeat_count = 0
     while True:
@@ -200,7 +222,8 @@ def main():
         time.sleep(60)
         heartbeat_count += 1
         if heartbeat_count % 5 == 0:  # 每5分钟输出一次心跳信息
-            logger.info(f"Heartbeat: Script is still running. Uptime: {heartbeat_count} minutes")
+            print(f"Heartbeat: Script is still running. Uptime: {heartbeat_count} minutes")
+            sys.stdout.flush()
 
 if __name__ == "__main__":
     main()