diff --git a/src/trade_server.py b/src/trade_server.py index 76b421a..42acabd 100644 --- a/src/trade_server.py +++ b/src/trade_server.py @@ -90,50 +90,10 @@ def should_use_simulation(): # 判断当前是否为交易日(只基于日期,不考虑时间) now = datetime.datetime.now() - # 尝试导入chinese_calendar判断是否为交易日 - try: - from chinese_calendar import is_workday, is_holiday - is_trading_day = is_workday(now) and not is_holiday(now) - logger.debug(f"使用chinese_calendar判断交易日: {now.date()}, 是交易日: {is_trading_day}") - except ImportError: - # 如果无法导入chinese_calendar,使用简单的工作日判断,并记录警告 - is_trading_day = now.weekday() < 5 # 0-4 为周一至周五 - logger.warning(f"无法导入chinese_calendar,使用简单工作日判断: {now.date()}, 是工作日: {is_trading_day}") - - # 额外检查是否为已知的中国法定节假日(如果必要,可以扩展此列表) - holidays_2023 = [ - datetime.date(2023, 1, 1), # 元旦 - datetime.date(2023, 1, 2), # 元旦调休 - datetime.date(2023, 1, 21), # 春节 - datetime.date(2023, 1, 22), # 春节 - datetime.date(2023, 1, 23), # 春节 - datetime.date(2023, 1, 24), # 春节 - datetime.date(2023, 1, 25), # 春节 - datetime.date(2023, 1, 26), # 春节 - datetime.date(2023, 1, 27), # 春节 - # ... 可以添加更多已知节假日 - ] - - holidays_2024 = [ - datetime.date(2024, 1, 1), # 元旦 - datetime.date(2024, 2, 10), # 春节 - datetime.date(2024, 2, 11), # 春节 - datetime.date(2024, 2, 12), # 春节 - datetime.date(2024, 2, 13), # 春节 - datetime.date(2024, 2, 14), # 春节 - datetime.date(2024, 2, 15), # 春节 - datetime.date(2024, 2, 16), # 春节 - datetime.date(2024, 2, 17), # 春节 - # ... 可以添加更多已知节假日 - ] - - # 合并所有年份的节假日 - all_holidays = holidays_2023 + holidays_2024 - - # 检查当前日期是否在已知节假日列表中 - if now.date() in all_holidays: - is_trading_day = False - logger.info(f"当前日期 {now.date()} 在已知节假日列表中,判定为非交易日") + # 使用chinese_calendar判断是否为交易日 + from chinese_calendar import is_workday, is_holiday + is_trading_day = is_workday(now) and not is_holiday(now) + logger.debug(f"使用chinese_calendar判断交易日: {now.date()}, 是交易日: {is_trading_day}") # 如果不是交易日,返回True(使用模拟交易) if not is_trading_day: