diff --git a/trade_day.py b/trade_day.py new file mode 100644 index 0000000..86f5dca --- /dev/null +++ b/trade_day.py @@ -0,0 +1,38 @@ +import time +import baostock as bs + +def is_trade_day(date=None): + if date is None: + date = time.strftime("%Y-%m-%d", time.localtime()) + #### 登陆系统 #### + lg = bs.login() + # 显示登陆返回信息 + if lg.error_code != str(0): + print('login respond error_code:'+lg.error_code) + print('login respond error_msg:'+lg.error_msg) + return False, False + + #### 获取交易日信息 #### + rs = bs.query_trade_dates(start_date=date, end_date=date) + if rs.error_code != str(0): + print('query_trade_dates respond error_code:'+rs.error_code) + print('query_trade_dates respond error_msg:'+rs.error_msg) + bs.logout() + return False, False + + #### 打印结果集 #### + trade = False + while (rs.error_code == '0') & rs.next(): + # 获取一条记录,将记录合并在一起 + data_record = rs.get_row_data() + if data_record[0] == date: + trade = data_record[1] == '1' + break + + #### 登出系统 #### + bs.logout() + return True, trade + + +# ok, trade = is_trade_day("2025-04-30") +# print("result:{0}, trading:{1}".format(ok, trade)) \ No newline at end of file