如何在Python脚本中通过PRAW追踪API调用次数？

下面是几种实用的实现方式：

1. 直接读取PRAW内部统计值 #

PRAW底层的Requestor对象会自动累计API调用次数，你可以直接访问其内部属性获取数值：

import praw

# 初始化Reddit客户端
reddit = praw.Reddit(
    client_id="你的客户端ID",
    client_secret="你的客户端密钥",
    user_agent="自定义用户代理"
)

# 执行任意API操作示例
for post in reddit.subreddit("python").hot(limit=5):
    print(post.title)

# 获取累计调用次数
print(f"累计API调用次数: {reddit._core._requestor.num_requests}")

注意：该属性属于PRAW未公开的内部接口，当前版本稳定可用，但后续版本可能存在变更风险。

2. 自定义请求钩子统计 #

通过注册请求钩子，你可以在每次API请求发起时执行自定义计数逻辑，同时还能记录请求细节：

import praw

api_call_counter = 0

def track_api_calls(request):
    global api_call_counter
    api_call_counter += 1
    # 可选：记录请求方法和URL
    # print(f"请求: {request.method} {request.url}")

reddit = praw.Reddit(
    client_id="你的客户端ID",
    client_secret="你的客户端密钥",
    user_agent="自定义用户代理"
)

# 注册请求钩子
reddit._core._requestor.register_hook("request", track_api_calls)

# 测试操作
subreddit = reddit.subreddit("learnpython")
print(subreddit.description)

print(f"累计API调用次数: {api_call_counter}")

这种方式灵活性更高，适合需要自定义统计维度（如按请求类型分类统计）的场景。

3. 结合速率限制处理器统计 #

如果需要同时监控API速率限制和调用次数，可以自定义RateLimitHandler实现：

import praw
from prawcore import RateLimitHandler

class CountingRateLimiter(RateLimitHandler):
    def __init__(self):
        super().__init__()
        self.total_calls = 0

    def delay(self, request, response):
        self.total_calls += 1
        # 保留原速率限制延迟逻辑
        super().delay(request, response)

# 初始化带自定义处理器的客户端
reddit = praw.Reddit(
    client_id="你的客户端ID",
    client_secret="你的客户端密钥",
    user_agent="自定义用户代理",
    ratelimit_handler=CountingRateLimiter()
)

# 执行测试操作
for comment in reddit.subreddit("python").comments(limit=10):
    print(comment.body)

# 获取累计调用次数
print(f"累计API调用次数: {reddit._core._requestor.ratelimit_handler.total_calls}")

该方案适合需要在处理速率限制的同时统计调用量的场景。

内容的提问来源于stack exchange，提问作者pstacksio