61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
import asyncio
|
||
import httpx
|
||
|
||
# 预置的眼镜行业上市公司列表
|
||
COMPANIES = [
|
||
{
|
||
"stock_code": "300622",
|
||
"company_name": "博士眼镜连锁股份有限公司",
|
||
"short_name": "博士眼镜",
|
||
"industry": "眼镜零售"
|
||
},
|
||
{
|
||
"stock_code": "301101",
|
||
"company_name": "明月镜片股份有限公司",
|
||
"short_name": "明月镜片",
|
||
"industry": "镜片制造"
|
||
},
|
||
{
|
||
"stock_code": "300595",
|
||
"company_name": "欧普康视科技股份有限公司",
|
||
"short_name": "欧普康视",
|
||
"industry": "医疗器械/OK镜"
|
||
},
|
||
{
|
||
"stock_code": "300015",
|
||
"company_name": "爱尔眼科医院集团股份有限公司",
|
||
"short_name": "爱尔眼科",
|
||
"industry": "眼科医疗"
|
||
},
|
||
{
|
||
"stock_code": "002223",
|
||
"company_name": "江苏鱼跃医疗设备股份有限公司",
|
||
"short_name": "鱼跃医疗",
|
||
"industry": "医疗器械"
|
||
}
|
||
]
|
||
|
||
API_URL = "http://localhost:8000/api/companies"
|
||
|
||
async def init_data():
|
||
print("开始添加预置公司...")
|
||
async with httpx.AsyncClient() as client:
|
||
for company in COMPANIES:
|
||
try:
|
||
# 检查是否已存在
|
||
response = await client.post(API_URL, json=company)
|
||
if response.status_code == 200:
|
||
print(f"✅ 成功添加: {company['short_name']} ({company['stock_code']})")
|
||
elif response.status_code == 400 and "已存在" in response.text:
|
||
print(f"ℹ️ 已存在: {company['short_name']} ({company['stock_code']})")
|
||
else:
|
||
print(f"❌ 添加失败: {company['short_name']} - {response.text}")
|
||
except Exception as e:
|
||
print(f"❌ 请求错误: {e}")
|
||
|
||
print("\n完成！请刷新「公司管理」页面查看。")
|
||
print("提示：添加完成后，请点击页面左下角的「立即同步」按钮开始抓取报告。")
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(init_data())
|