python类get_stock_basics()的实例源码

def get_basic_info(show_progress=True):
    # ?????
    if show_progress:
        click.echo("????????")
    ts_symbols = ts.get_stock_basics()
    if show_progress:
        click.echo("??????")

    symbols = []

    histories = {}

    # ??????
    i = 0
    total = len(ts_symbols)
    for index, row in ts_symbols.iterrows():
        i = i +1
        if i > 10:
            break

        srow = {}
        # ????????
        click.echo("??????%s(%s)??????? (%d/%d)" % (index, row['name'], i, total))
        histories[index] = ts.get_hist_data(index)
        srow['start_date'] = histories[index].index[-1]
        srow['end_date'] = histories[index].index[0]
        srow['symbol'] = index
        srow['asset_name'] = row['name']
        symbols.append(srow)

    df_symbols = pd.DataFrame(data=symbols).sort_values('symbol')
    symbol_map = pd.DataFrame.copy(df_symbols.symbol)

    # fix the symbol exchange info
    df = df_symbols.apply(func=convert_symbol_series, axis=1)


    return df, histories, symbol_map

def QA_fetch_get_stock_info(name):
    data = QATs.get_stock_basics()
    data_json = json.loads(data.to_json(orient='records'))

    for i in range(0, len(data_json) - 1, 1):
        data_json[i]['code'] = data.index[i]
    return data_json

def QA_fetch_get_stock_list():
    df = QATs.get_stock_basics()
    return list(df.index)

def fetch_stock_basics(conn):
    logging.debug("Fetch stocks")
    df = ts.get_stock_basics()
    df['timeToMarket'] = df['timeToMarket'].map(lambda s: datetime.strptime(str(s), '%Y%m%d') if s > 0 else None)
    df.to_sql('stock_basics', conn, if_exists="replace", dtype={"code": VARCHAR(32)})

def Get_Stock_List():
    df = ts.get_stock_basics()
    return df

#?????????????????
#??MACD?KDJ?????

def getCompany():
    """code ??, name ??, industry ????, area ??, pe ???, outstanding ????(?), totals ???(?),
    totalAssets ???(?), liquidAssets ????, fixedAssets ????, reserved ???, reservedPerShare ?????,
    esp ????, bvps ????, pb ???, timeToMarket ????, undp ????, perundp ?????, rev ????(%),
    profit ????(%), gpr ???(%), npr ????(%), holders ???? """
    try:
        basics = ts.get_stock_basics()
        index = list(basics.index)  #????
        name = list(basics.name)  #????
        industry = list(basics.industry)  #????
        indexLen = len(index)

        cur1 = conn.cursor()
        cur1.execute("DROP TABLE IF EXISTS basics;")  #????????????
        sqlCreate = """create table basics
                       (b_id int comment '??',
                       b_index varchar(100) comment '????',
                       b_name varchar(100) comment '????',
                       b_industry varchar(100) comment '????',
                       PRIMARY key(b_id))"""
        cur1.execute(sqlCreate)

        for i in range(0, indexLen):
            print (i+1, index[i], name[i], industry[i])

            cur1.execute(
                "INSERT INTO basics(b_id, b_index, b_name, b_industry) "
                "VALUES('%d', '%s', '%s', '%s');"
                 % (i+1, index[i], name[i], industry[i]))

        cur1.execute('commit;')
        print ('?????')
    except Exception:
        pass

def get_all_company():
    df = ts.get_stock_basics()
    # print(pinyin(stockName))
    print("?????")
    return df

def stat_all(tmp_datetime):
    # ????
    data = ts.get_deposit_rate()
    common.insert_db(data, "ts_deposit_rate", False, "`date`,`deposit_type`")

    # ????
    data = ts.get_loan_rate()
    common.insert_db(data, "ts_loan_rate", False, "`date`,`loan_type`")

    # ??????
    data = ts.get_rrr()
    common.insert_db(data, "ts_rrr", False, "`date`")

    # ?????
    data = ts.get_money_supply()
    common.insert_db(data, "ts_money_supply", False, "`month`")

    # ?????(????)
    data = ts.get_money_supply_bal()
    common.insert_db(data, "ts_money_supply_bal", False, "`year`")

    # ??????(??)
    data = ts.get_gdp_year()
    common.insert_db(data, "ts_gdp_year", False, "`year`")

    # ??????(??)
    data = ts.get_gdp_quarter()
    common.insert_db(data, "ts_get_gdp_quarter", False, "`quarter`")

    # ?????GDP??
    data = ts.get_gdp_for()
    common.insert_db(data, "ts_gdp_for", False, "`year`")

    # ?????GDP??
    data = ts.get_gdp_pull()
    common.insert_db(data, "ts_gdp_pull", False, "`year`")

    # ???????
    data = ts.get_gdp_contrib()
    common.insert_db(data, "ts_gdp_contrib", False, "`year`")

    # ????????
    data = ts.get_cpi()
    common.insert_db(data, "ts_cpi", False, "`month`")

    # ?????????
    data = ts.get_ppi()
    common.insert_db(data, "ts_ppi", False, "`month`")

    #############################????? http://tushare.org/fundamental.html
    # ????
    data = ts.get_stock_basics()
    print(data.index)
    common.insert_db(data, "ts_stock_basics", True, "`code`")




# main????