当前位置：首页 > news >正文

通过Python脚本批量管理Taotoken上的API Key与用量

news 2026/5/14 8:35:33

🚀 告别海外账号与网络限制！稳定直连全球优质大模型，限时半价接入中。 👉 点击领取海量免费额度

通过Python脚本批量管理Taotoken上的API Key与用量

对于管理多个项目或团队的开发者而言，手动在控制台逐个创建、配置和监控API Key既繁琐又容易出错。Taotoken平台提供了与OpenAI兼容的管理API，允许你通过程序化方式自动化这些任务。本文将介绍如何编写Python脚本，利用这些API实现对API Key的批量创建、状态管理、用量查询等操作，从而将权限与成本管控集成到你的自动化运维流程中。

1. 准备工作：获取管理API凭证

要调用Taotoken的管理API，你需要一个具备相应权限的API Key。这个Key与你用于调用模型服务的Key不同，通常被称为“管理Key”或“主Key”。

登录Taotoken控制台。
进入“API密钥”管理页面。
创建一个新的API Key，并确保在创建时为其勾选或分配了管理权限（如“密钥管理”、“用量查询”等，具体权限名称请以控制台实际选项为准）。
妥善保存此Key，它将是脚本与平台交互的凭证。

此外，你还需要确认管理API的端点地址。Taotoken的管理API同样遵循OpenAI兼容格式，其基础地址（Base URL）为https://taotoken.net/api。

2. 编写Python脚本的核心步骤

我们将使用Python的requests库来发送HTTP请求。首先确保已安装该库：pip install requests。

2.1 初始化客户端与认证

脚本的第一步是设置基础URL和管理Key，并构建用于认证的请求头。

import requests import json # 配置信息 MANAGEMENT_API_KEY = "你的管理API_KEY" # 替换为你的管理Key BASE_URL = "https://taotoken.net/api" # 通用的请求头 headers = { "Authorization": f"Bearer {MANAGEMENT_API_KEY}", "Content-Type": "application/json" }

2.2 批量创建API Key

你可以通过循环调用创建接口，为不同项目或团队成员生成独立的Key。通常可以在创建时指定名称、额度限制、过期时间等属性。

def create_api_key(key_name, limit=None): """创建一个新的API Key""" url = f"{BASE_URL}/v1/api_keys" # 创建Key的端点 payload = { "name": key_name, # 其他可选参数，如 "limit", "expires_at" 等，请参考平台API文档 } if limit: payload["limit"] = limit # 例如设置额度限制 response = requests.post(url, headers=headers, json=payload) if response.status_code == 200: key_data = response.json() print(f"Key '{key_name}' 创建成功。Key: {key_data.get('key')}") return key_data else: print(f"创建Key '{key_name}' 失败: {response.status_code}, {response.text}") return None # 示例：批量创建Key project_keys = [] for project in ["project_alpha", "project_beta", "team_gamma"]: key_info = create_api_key(f"key_for_{project}", limit=1000000) # 假设设置100万token额度 if key_info: project_keys.append(key_info)

2.3 查询与列出所有API Key

为了集中管理，你需要能够获取所有已创建Key的列表及其状态。

def list_api_keys(): """列出所有API Key""" url = f"{BASE_URL}/v1/api_keys" response = requests.get(url, headers=headers) if response.status_code == 200: keys = response.json().get('data', []) # 注意响应结构可能为分页格式 for key in keys: print(f"ID: {key['id']}, 名称: {key.get('name')}, 状态: {key.get('status')}") return keys else: print(f"获取Key列表失败: {response.status_code}, {response.text}") return [] # 获取并查看所有Key all_keys = list_api_keys()

2.4 禁用/启用与删除API Key

当项目结束或成员离职时，需要及时禁用或删除对应的Key。

def modify_api_key_status(key_id, action): """修改API Key状态 (禁用/启用)""" # action 通常为 "disable" 或 "enable" url = f"{BASE_URL}/v1/api_keys/{key_id}/{action}" response = requests.post(url, headers=headers) if response.status_code == 200: print(f"Key {key_id} 已{action}。") else: print(f"操作Key {key_id} 失败: {response.status_code}, {response.text}") def delete_api_key(key_id): """删除一个API Key""" url = f"{BASE_URL}/v1/api_keys/{key_id}" response = requests.delete(url, headers=headers) if response.status_code == 200: print(f"Key {key_id} 已删除。") else: print(f"删除Key {key_id} 失败: {response.status_code}, {response.text}") # 示例：禁用某个特定的Key target_key_id = "需要操作的Key_ID" modify_api_key_status(target_key_id, "disable")

2.5 查询API Key用量与成本

定期拉取各Key的用量数据，是进行成本分析和预算控制的基础。

def get_key_usage(key_id, start_date=None, end_date=None): """查询指定API Key的用量详情""" url = f"{BASE_URL}/v1/api_keys/{key_id}/usage" params = {} if start_date: params["start_date"] = start_date # 格式参考API文档，如 "2024-01-01" if end_date: params["end_date"] = end_date response = requests.get(url, headers=headers, params=params) if response.status_code == 200: usage_data = response.json() # 用量数据的具体结构请参考平台API文档 print(f"Key {key_id} 用量数据: {json.dumps(usage_data, indent=2, ensure_ascii=False)}") return usage_data else: print(f"获取Key {key_id} 用量失败: {response.status_code}, {response.text}") return None # 示例：查询最近7天的用量（假设日期格式） import datetime end = datetime.datetime.now().strftime("%Y-%m-%d") start = (datetime.datetime.now() - datetime.timedelta(days=7)).strftime("%Y-%m-%d") for key in all_keys[:3]: # 查询前3个Key的用量 get_key_usage(key['id'], start, end)