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

可以这样写：import bs_inferencedef test_download(stock_code):
data = bs_inference.download(stock_code)
print(data)

test_download('000001')

相关问题

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

import baostock as bsdef download(code, start_date, end_date):
# 登陆系统
lg = bs.login()
# 显示登陆返回信息
print('login respond error_code:'+lg.error_code)
print('login respond error_msg:'+lg.error_msg)

# 获取股票数据
rs = bs.query_history_k_data_plus(code,
"date,code,open,high,low,close,preclose,volume,amount,adjustflag,turn,tradestatus,pctChg,isST",
start_date=start_date, end_date=end_date,
frequency="d", adjustflag="3")
data_list = []
while (rs.error_code == '0') & rs.next():
# 获取一条记录，将记录合并在一起
data_list.append(rs.get_row_data())
# 登出系统
bs.logout()
return data_list

用python写1个代码，bs_inference.py，里面有一个download 函数，这个函数的功能是调用 baostock接口来进行数据的下载，最终返回的是一个数据list。

我可以为你提供一个bs_inference.py的python代码示例供你参考：import baostock as bs

def download(code, start_date, end_date):
# 登陆系统
lg = bs.login()
# 显示登陆返回信息
print('login respond error_code:'+lg.error_code)
print('login respond error_msg:'+lg.error_msg)

# 获取股票数据
rs = bs.query_history_k_data_plus(code,
"date,code,open,high,low,close,preclose,volume,amount,adjustflag,turn,tradestatus,pctChg,isST",
start_date=start_date, end_date=end_date,
frequency="d", adjustflag="3")
print('query_history_k_data_plus respond error_code:'+rs.error_code)
print('query_history_k_data_plus respond error_msg:'+rs.error_msg)

result_list = []
while (rs.error_code == '0') & rs.next():
# 获取一条记录，将记录合并在一起
result_list.append(rs.get_row_data())
# 结果集定位到第一条记录
rs.seek(0)
# 登出系统
bs.logout()
return result_list