database.py 文件源码

def get_symbols_save_into_db(mode = 'replace'):
    '''
    ?????????????.

    Parameters
    -----------
        mode
            ??,replace????,append???????
    '''
    now = dt.datetime.now()

    data = ts.get_stock_basics()

    con = pymysql.connect(
            host = db_host,
            user = db_user,
            passwd = db_pass,
            db = db_name,
            charset = 'utf8')

    data['created_time'] = now
    data['last_updated_time'] = now
    data['id'] = range(1,len(data) + 1)
    data['ticker'] = data.index
    data['timeToMarket'] = data['timeToMarket'].apply(date_time_str_convertor)
    data.to_sql('symbols',con,if_exists = 'replace',
                index = False,flavor = 'mysql')

    sql = '''
    ALTER TABLE symbols
    ADD PRIMARY KEY (id);
    '''

    cur = con.cursor()
    cur.execute(sql)
    con.close()
    print('????symbols?(??????)')