#!/usr/bin/env python
# coding=utf-8

"""
@author: CNWei
@Software: PyCharm
@contact: t6i888@163.com
@file: case_validator
@date: 2025/2/27 17:25
@desc: 
"""
import logging
from typing import List, Union, Any

from pydantic import TypeAdapter

from core.exchange import Exchange
from core.models import ValidateItem

logger = logging.getLogger(__name__)


VALIDATE_LIST_ADAPTER = TypeAdapter(List[ValidateItem])

class CaseValidator:
    VALIDATORS = {}

    @classmethod
    def register(cls, name: str):
        def decorator(func):
            cls.VALIDATORS[name] = func
            return func

        return decorator

    @classmethod
    def validate(cls,response: Any, validate_list: List[ValidateItem]):
        """
        核心断言入口：适配 CaseInfo.validate_data (List[ValidateItem])
        """
        if not validate_list:
            return
        # dicts = [
        #     item.model_dump(by_alias=True) if isinstance(item, ValidateItem) else item for item in validate_list
        # ]
        # rendered = exchanger.replace(dicts)
        # # 触发 SmartInt/SmartDict 类型修复
        # final_list = VALIDATE_LIST_ADAPTER.validate_python(rendered)

        for item in validate_list:
            # 1. 提取模型中的数据
            # 此时 final_case 里的 item 已经是经过变量替换后的实体
            actual = item.check
            expect = item.expect
            method = item.assert_method  # 即模型中的 alias="assert"
            msg = item.msg or f"Assert {actual} {method} {expect}"

            # 2. 获取对应的断言函数
            validator = cls.VALIDATORS.get(method)
            if not validator:
                logger.error(f"❌ 不支持的断言方式: {method}")
                raise KeyError(f"Unsupported validator: {method}")

            # 3. 执行断言
            try:
                validator(actual, expect, msg)
            except AssertionError as e:
                logger.error(
                    f"❌ 断言失败: {msg} | 实际值: {actual} ({type(actual)}), 期望值: {expect} ({type(expect)})")
                raise e


@CaseValidator.register('equals')
def validate_equals(a, b, msg):
    logger.info(f"assert {a} == {b}, {msg} 执行这段代码")
    print(f"assert {a} == {b}, {msg} 执行这段代码")
    assert a == b, msg


@CaseValidator.register('not_equals')
def validate_not_equals(a, b, msg):
    logger.info(f"assert {a} != {b}, {msg}")
    assert a != b, msg


@CaseValidator.register('contains')
def validate_contains(a, b, msg):
    logger.info(f"assert {a} in {b}, {msg}")
    assert a in b, msg


@CaseValidator.register('not_contains')
def validate_not_contains(a, b, msg):
    logger.info(f"assert {a} not in {b}, {msg}")
    assert a not in b, msg


if __name__ == '__main__':
    resp=None
    mock_case = [
    {"check": 100, "expect": 100, "assert": "equals"},
    {"check": "success", "expect": "success", "assert": "contains"}
]
    final_validate_list = VALIDATE_LIST_ADAPTER.validate_python(mock_case)
    case_validator = CaseValidator()
    print(case_validator.VALIDATORS)
    case_validator.validate(resp,final_validate_list)