import pymongo
import akshare as ak
import json

stockNUm = 0
# 打开数据库
mongo_client = pymongo.MongoClient('mongodb://localhost:27017/')
# 若没有Database 和Collection，则会自动创建
mongo_db = mongo_client['SinaData']

# mongo_client = pymongo.MongoClient('mongodb://localhost:27017/')
# # 若没有Database 和Collection，则会自动创建
mongo_collection = mongo_db['SinaAllStockName']

# 查询第一条
# print(mongo_collection.find_one())
# 查询所有数据
mydoc = mongo_collection.find()
for x in mydoc:
    print(x)
    stockNUm = stockNUm + 1
    if stockNUm == 3:
        break
    # 在量化投资研究中普遍采用后复权数据  新浪财经
    stock_zh_a_daily_qfq_df = ak.stock_zh_a_daily(symbol=str(x['code']), adjust="qfq")  # 历史行情数据(前复权)
    print(stock_zh_a_daily_qfq_df)
    print(type(stock_zh_a_daily_qfq_df))  # DataFrame
    # DataFrame 日期列转换为字符串
    stock_zh_a_daily_qfq_df['date'] = stock_zh_a_daily_qfq_df['date'].astype(str)
    # 将 DataFrame 写入MongoDB
    data = json.loads(stock_zh_a_daily_qfq_df.T.to_json()).values()
    stock_trade = mongo_db[str(x['code'])]
    stock_trade.insert_many(data)