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)