用python写1个交互代码，bs_inference.py，里面有一个download 函数，这个函数的功能是调用 baostock接口来进行数据的下载，最终返回的是一个数据list。函数的参数是（股票代码，开始日期，结束日期）。

# 在bs_inference.py中定义download函数：def download(code, start_date, end_date):
# 调用 baostock 接口来获取数据
lg = bs.login()
rs = bs.query_history_k_data_plus(code,
"date,code,open,high,low,close,preclose,volume,amount,adjustflag,turn,tradestatus,pctChg,peTTM,pbMRQ,psTTM,pcfNcfTTM,isST",
start_date=start_date, end_date=end_date,
frequency="d", adjustflag="3")
if rs.error_code != '0':
print('获取数据失败')
else:
# 返回获取的数据列表
data_list = []
while (rs.error_code == '0') & rs.next():
data_list.append(rs.get_row_data())
return data_list

相关问题

用python写1个测试代码：test.py。通过 import bs_inference来导入download函数，用户可以通过输入download函数的参数获取到对应股票的数据

示例：import bs_inferencedef test():
stock_data = bs_inference.download(stock_name="MSFT", start_date="2020-01-01", end_date="2020-04-30")
print(stock_data)if __name__ == "__main__":
test()

用python写1个测试代码：test.py。通过 import bs_inference来导入之前的download函数，从而获取到对应股票的数据

import bs_inferencedef test():
stock_data = bs_inference.download('AAPL')
print(stock_data)if __name__ == '__main__':
test()