yu/real_trader
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

删除 导入chinese_calendar失败的逻辑

Browse Source
This commit is contained in:
zhiyong 2025-04-30 23:05:42 +08:00
parent 95cd287675
commit 519d4d2021
1 changed files with 4 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/trade_server.py
View File

@ -90,50 +90,10 @@ def should_use_simulation():
# 判断当前是否为交易日(只基于日期，不考虑时间)
now = datetime.datetime.now()
# 尝试导入chinese_calendar判断是否为交易日
try:
# 使用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}")
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()} 在已知节假日列表中，判定为非交易日")
# 如果不是交易日，返回True（使用模拟交易）
if not is_trading_day: