python函数装饰器,返回自定义类型

from typing import Any, Callable, TypeVar
import requests
T = TypeVar("T")

class MaxRetryError(Exception):
    def __init__(self):
        message = "超出最大重试次数"
        super().__init__(message)

class MaxRetry:
    def __init__(self, max_retry: int = 2):
        self.max_retry = max_retry

    def __call__(self, connect_once: Callable[..., T]) -> Callable[..., T]:
        def connect_n_times(*args: Any, **kwargs: Any) -> T:
            retry = self.max_retry + 1
            while retry:
                try:
                    return connect_once(*args, **kwargs)
                except requests.exceptions.Timeout as e:
                    print("抓取超时,正在尝试重新连接~")
                finally:
                    retry -= 1
            raise MaxRetryError()

        return connect_n_times
@MaxRetry(2)
def get_info(url: str) -> str:
    re = requests.get(url, timeout=2)
    return re.text
url = "https://www.abc.com/4de25977c7dce0"
get_info(url)

[output]:
抓取超时,正在尝试重新连接~
抓取超时,正在尝试重新连接~
抓取超时,正在尝试重新连接~
---------------------------------------------------------------------------
MaxRetryError                             Traceback (most recent call last)
<ipython-input-37-bc2dde821bd1> in <module>
----> 1 get_info(url)

<ipython-input-34-b40f19f5bf7c> in connect_n_times(*args, **kwargs)
     22                 finally:
     23                     retry -= 1
---> 24             raise MaxRetryError()
     25 
     26         return connect_n_times

MaxRetryError: 超出最大重试次数
posted @   JinX-Digital  阅读(180)  评论(0编辑  收藏  举报
编辑推荐:
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
阅读排行:
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· 阿里巴巴 QwQ-32B真的超越了 DeepSeek R-1吗?
· 如何调用 DeepSeek 的自然语言处理 API 接口并集成到在线客服系统
· 【译】Visual Studio 中新的强大生产力特性
· 2025年我用 Compose 写了一个 Todo App
点击右上角即可分享
微信分享提示
主题色彩