17 lines
691 B
Python
17 lines
691 B
Python
from cryptoai.utils.db_manager import get_db_manager
|
|
from cryptoai.api.adata_api import AStockAPI
|
|
import json
|
|
from time import sleep
|
|
if __name__ == "__main__":
|
|
print("开始获取A股数据")
|
|
api = AStockAPI()
|
|
stock_codes = api.get_all_stock_codes()
|
|
list = json.loads(stock_codes.to_json(orient="records"))
|
|
# print(list[0])
|
|
|
|
# 保存到数据库
|
|
for stock in list:
|
|
print(f"创建股票: {stock['stock_code']} - {stock['short_name']}")
|
|
get_db_manager().create_stock(stock["stock_code"], stock["short_name"], stock["exchange"], stock["list_date"])
|
|
print(f"创建股票: {stock['stock_code']} - {stock['short_name']} 完成")
|
|
sleep(1) |