Update vps_monitor.py

vps_monitor.py

@@ -6,32 +6,21 @@ import sys
 from flask import Flask, jsonify, render_template_string
 from threading import Thread
 import logging
-from colorama import init, Fore, Back, Style
-
-init(autoreset=True)  # 初始化 colorama
 
 app = Flask(__name__)
 
 vps_status = {}
 
 # 设置日志
-class ColoredFormatter(logging.Formatter):
-    def format(self, record):
-        if record.levelno == logging.INFO:
-            record.msg = f"{Fore.GREEN}{record.msg}{Style.RESET_ALL}"
-        elif record.levelno == logging.WARNING:
-            record.msg = f"{Fore.YELLOW}{record.msg}{Style.RESET_ALL}"
-        elif record.levelno == logging.ERROR:
-            record.msg = f"{Fore.RED}{record.msg}{Style.RESET_ALL}"
-        return super().format(record)
-
-formatter = ColoredFormatter('%(asctime)s - %(levelname)s - %(message)s')
-handler = logging.StreamHandler(sys.stdout)
-handler.setFormatter(formatter)
-
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.StreamHandler(sys.stdout),
+        logging.StreamHandler(sys.stderr)
+    ]
+)
 logger = logging.getLogger()
-logger.setLevel(logging.INFO)
-logger.addHandler(handler)
 
 def get_vps_configs():
     configs = []
@@ -89,7 +78,7 @@ def check_and_run_script(config):
         check_command = f"ps aux | grep {script_name} | grep -v grep"
         stdin, stdout, stderr = client.exec_command(check_command)
         if stdout.read():
-            status = f"{Fore.GREEN}Running{Style.RESET_ALL}"
+            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.")
@@ -98,13 +87,13 @@ def check_and_run_script(config):
             exit_status = stdout.channel.recv_exit_status()
             
             if exit_status == 0:
-                status = f"{Fore.YELLOW}Restarted{Style.RESET_ALL}"
+                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 = f"{Fore.RED}Error{Style.RESET_ALL}"
+                status = "Error"
                 logger.error(f"Error executing restart script {script_name} on {config['hostname']}: {error_output}")
         
         key = f"{config['hostname']}:{script_name}"
@@ -121,7 +110,7 @@ def check_and_run_script(config):
         key = f"{config['hostname']}:{script_name}"
         vps_status[key] = {
             'index': config['index'],
-            'status': f"{Fore.RED}Error: {str(e)}{Style.RESET_ALL}",
+            'status': f"Error: {str(e)}",
             'last_check': time.strftime('%Y-%m-%d %H:%M:%S'),
             'username': config['username'],
             'script_name': script_name
@@ -133,11 +122,11 @@ def check_and_run_script(config):
         logger.info(f"Finished checking VPS {config['index']}: {config['hostname']} - {script_name}")
 
 def check_all_vps():
-    logger.info(f"{Back.BLUE}{Fore.WHITE}Starting VPS check{Style.RESET_ALL}")
+    logger.info("Starting VPS check")
     vps_configs = get_vps_configs()
     for config in vps_configs:
         check_and_run_script(config)
-    logger.info(f"{Back.BLUE}{Fore.WHITE}Finished VPS check{Style.RESET_ALL}")
+    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']}")