# -*- coding: utf-8 -*-
"""
第一步:更新股票池和日历
"""
import akshare as ak
# 删除本地数据----------------------------------------------------------------------
def dele():
import pymysql
connection = pymysql.connect(host='localhost', port=3306, user='root', passwd='root@123', db='fund_new', charset='utf8')
driver = connection.cursor()
sql="DELETE FROM days"
driver.execute(sql)
connection.commit()
print("删除本地数据成功")
connection.close()
def one(con1):
#dele()
#交易日历,全量更新
index_zh_a_hist_df = ak.index_zh_a_hist(symbol="000016", period="daily", start_date="20210101", end_date="22220101")
index_zh_a_hist_df['last_day'] = index_zh_a_hist_df['日期'].shift(1)
temp = index_zh_a_hist_df
for inde in ['涨跌幅']:
for i in [5,10,20,40]:
temp[inde + str(i)] = temp[inde].rolling(i).sum()
for inde in ['换手率']:
for i in [5,10,20,40]:
temp[inde + str(i)] = temp[inde].rolling(i).mean()
temp.to_sql(name='days', con=con1, if_exists='replace', index=False)
print ('更新日历', len(temp))
stock_comment_em_df = ak.stock_comment_em()
codes = stock_comment_em_df[['代码','名称']]
codes.to_sql(name='codes', con=con1, if_exists='replace', index=False)
print ('股票池', len(codes))