from django.core.management.base import BaseCommand
from health_monitoring.services.mset_service import MSETService

class Command(BaseCommand):
    help = '测试Excel数据加载和MSET计算（原代码逻辑验证）'

    def handle(self, *args, **options):
        # 1. 初始化服务
        service = MSETService()
        
        # 2. 测试发电机组数据
        self.stdout.write("=== 测试发电机组数据 ===")
        service.train_from_excel('generator')
        test_data_gen = {
            'U_temp': 65, 
            'V_temp': 67, 
            'W_temp': 63, 
            'bearing_temp': 45
        }
        score_gen, weights_gen = service.evaluate_health(test_data_gen)
        self.stdout.write(f"发电机组健康度: {score_gen}")
        self.stdout.write(f"权重分布: {weights_gen}")

        # 3. 测试机舱系统数据（可选）
        self.stdout.write("\n=== 测试机舱系统数据 ===")
        service.train_from_excel('nacelle')
        test_data_nac = {
            'vibration_x': 2.3,
            'vibration_y': 1.8,
            'position': 102,
            'temp': 28
        }
        score_nac, weights_nac = service.evaluate_health(test_data_nac)
        self.stdout.write(f"机舱系统健康度: {score_nac}")
        self.stdout.write(f"权重分布: {weights_nac}")