import requests
import json

# 请将此处的 URL 替换为您实际的 API 基础域名
api_url = "http://10.252.130.135:8000/api/v1/sources/"

# 请求头
headers = {
    "Content-Type": "application/json",
    # "Authorization": "Bearer YOUR_TOKEN" # 如果接口需要鉴权，请取消注释并填入 Token
}

# 解析后的数据源列表
sources_data = [
    {"name": "今日头条", "url": "toutiao"},
    {"name": "百度热搜", "url": "baidu"},
    {"name": "华尔街见闻", "url": "wallstreetcn-hot"},
    {"name": "澎湃新闻", "url": "thepaper"},
    {"name": "bilibili 热搜", "url": "bilibili-hot-search"},
    {"name": "财联社热门", "url": "cls-hot"},
    {"name": "凤凰网", "url": "ifeng"},
    {"name": "贴吧", "url": "tieba"},
    {"name": "微博", "url": "weibo"},
    {"name": "抖音", "url": "douyin"},
    {"name": "知乎", "url": "zhihu"}
]

# 遍历数据并发送 POST 请求
for item in sources_data:
    payload = {
        "source_name": item["name"],
        "source_type": "HOT_TREND",
        "home_url": item["url"],
        "is_enabled": True
    }

    try:
        response = requests.post(api_url, headers=headers, data=json.dumps(payload))
        if response.status_code in (200, 201):
            print(f"✅ 成功创建: {item['name']}")
        else:
            print(f"❌ 创建失败: {item['name']} - 状态码: {response.status_code} - 详情: {response.text}")
    except Exception as e:
        print(f"⚠️ 请求异常: {item['name']} - 错误: {e}")

print("执行完毕！")