python限制调用接口速度

EaSoNgo1112023-11-13 12:46

复制代码

import time
import requests
import pandas as pd
from concurrent.futures import ThreadPoolExecutor


class RateLimiter:
    def __init__(self, max_calls, period):
        self.max_calls = max_calls
        self.period = period
        self.calls = 0
        self.reset_time = time.time()

    def can_call(self):
        current_time = time.time()
        if current_time - self.reset_time > self.period:
            self.reset_time = current_time
            self.calls = 0
        return self.calls < self.max_calls

    def called(self):
        self.calls += 1

在这里，RateLimiter类通过在指定时间段内限制最大调用次数来实现速率限制。构造函数__init__接受两个参数：max_calls表示最大调用次数，period表示时间段（以秒为单位）。

类中的can_call方法用于判断是否可以进行调用。它首先检查当前时间与上次重置时间之间的时间差是否超过设定的时间段。如果超过了时间段，则重置计数器，并更新重置时间和调用次数。最后，判断调用次数是否小于最大调用次数，如果是则返回True，表示可以进行调用；否则返回False，表示已经达到限制。

called方法用于在每次调用后更新调用次数。当函数或接口被调用时，应该在合适的位置调用called方法，以便在限制内适当增加调用次数。

这种速率限制器通常用于控制对外部API的访问次数，以免超出API提供商的规定限制。它能够平滑地限制请求速率，避免因短时间内大量请求而导致服务崩溃或被封禁。