import os

import requests
from loguru import logger
import inspect
from functools import wraps

cwd = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
print(cwd)
logger.add(f"{cwd}/my.log",  # log文件地址
           level="DEBUG",  # log记录最低级别
           rotation="00:00",  # 将日志记录以大小、时间等方式进行分割或划分
           retention="3 days",  # 文件保留时间
           compression="zip",  # 文件压缩格式
           backtrace=True,
           # rotation="1 MB"  # 滚动大日志文件

           )
LOGGER = logger


def send_mag(content):
    for i in range(3):
        try:
            url = 'https://jdtbpdd.cn/wx/send_msg?'
            data = {'content': content}
            res = requests.post(url, data=data).json()
            logger.info(f'消息发送结果：{res}')
            return res
        except Exception as e:
            logger.error(f'推送消息异常：{e}')
    return None


def _get_caller_info():
    frame = inspect.currentframe()
    try:
        caller = frame.f_back.f_back  # 跳过 wrapper 自身
        return f"{caller.f_code.co_filename}:{caller.f_lineno}"
    finally:
        del frame


def retry(name='备注', log=True, result_log=False, over_msg=False, error_msg=True):
    """

    :param name:
    :param log: 日志是否打印
    :param result_log:  返回结果日志是否打印
    :param over_msg:    成功是否推送
    :param error_msg:   失败是否推送
    :return:
    """

    def retry_fun(fun):
        @wraps(fun)
        def wrapper(*args, **kwargs):

            caller_info = _get_caller_info() if log else ""

            if log:
                logger.info(f"[{name}] 开始: {caller_info}")

            try:
                result = fun(*args, **kwargs)
                if log:
                    if result_log:
                        logger.info(f"[{name}] 调用成功: {caller_info} result:{str(result)}")
                    else:
                        logger.info(f"[{name}] 调用成功: {caller_info}")
                if over_msg:
                    send_mag(f"[{name}] 调用成功:\n[执行程序]:{caller_info}\n[返回结果]:{str(result)}")
                return result
            except Exception as e:
                if log:
                    logger.exception(f"[{name}] 异常调用: {caller_info} - {e}")
                else:
                    logger.exception(f"[{name}] 异常: {e}")
                if error_msg:
                    send_mag(f"[{name}]: 异常调用:\n[执行程序]:{caller_info}\n[异常内容]:{e}")
                # raise   # ⚠️ 强烈建议保留（否则异常被吞掉）

        return wrapper

    return retry_fun


@retry(name='测试', over_msg=True)
def test():

    return 'mytest'


if __name__ == '__main__':
    test()