# -*- 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))