一种新的获取数据的办法(api改进) about pytdx HOT 12 CLOSED

def QA_util_get_real_date(date, trade_list, towards):
        #print(date in trade_list)
    if towards == 1:
        while date not in trade_list:
            date = str(datetime.datetime.strptime(
                date, '%Y-%m-%d') + datetime.timedelta(days=1))[0:10]
        else:
            return date
    elif towards == -1:
        while date not in trade_list:
            date = str(datetime.datetime.strptime(
                date, '%Y-%m-%d') - datetime.timedelta(days=1))[0:10]
        else:
            return date

def QA_fetch_get_stock_day(code, start_date,end_date,ip='119.147.212.81',port=7709):
    start_date=QA_util_get_real_date(start_date,trade_date_sse,1)
    end_date=QA_util_get_real_date(end_date,trade_date_sse,-1)
    with api.connect(ip, port):

        # 判断end_date在哪个位置
        index_0=str(datetime.date.today())
        index_of_index_0=trade_date_sse.index(index_0)
        index_of_index_end=trade_date_sse.index(end_date)
        index_of_index_start=trade_date_sse.index(start_date)
        
        index_of_end=index_of_index_0-index_of_index_end
        index_length=index_of_index_end+1-index_of_index_start
        data = api.get_security_bars(9, 0, code,index_of_end, index_length)  # 返回普通list
        data = api.to_df(api.get_security_bars(9, 0, code,index_of_end, index_length))  # 返回DataFrame
    return data


print(QA_fetch_get_stock_day('000001','2017-07-03','2017-07-10'))
print(QA_fetch_get_stock_day('000001','2017-07-01','2017-07-09'))