38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
|
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))
|