From 603cb1f2aaa33e2fb93b2ef6c16e0c17addb9919 Mon Sep 17 00:00:00 2001 From: DingZQ Date: Mon, 7 Apr 2025 22:14:00 +0800 Subject: [PATCH] Refine --- main.py | 4 ++-- time_api.py | 6 ++++++ 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 896e7bf..f66a8ad 100644 --- a/main.py +++ b/main.py @@ -2326,7 +2326,7 @@ async def fastapi_query_all_records_by_date(querydate: str) -> dict[str, list]: global redis_client # 今天的不要去缓存 - if not time_api.is_today(querydate): + if not time_api.is_today_or_future(querydate): cache_key = f"queryallrecordsbydate_{querydate}" data = redis_client.get(cache_key) if data: @@ -2339,7 +2339,7 @@ async def fastapi_query_all_records_by_date(querydate: str) -> dict[str, list]: "links": nodes_links[1] } # 今天的不要去缓存 - if not time_api.is_today(querydate): + if not time_api.is_today_or_future(querydate): redis_client.set(cache_key, msgpack.packb(results, default=encode_datetime)) return results diff --git a/time_api.py b/time_api.py index b4eaf9b..85cc2f4 100644 --- a/time_api.py +++ b/time_api.py @@ -134,4 +134,10 @@ def is_tomorrow(query_date: str) -> bool: dt = parse_beijing_time(query_date) return dt.date() == (datetime.now().date() + timedelta(days=1)) +def is_today_or_future(query_date: str) -> bool: + ''' + 判断一个日期是否是今天或未来 + ''' + dt = parse_beijing_time(query_date) + return dt.date() >= datetime.now().date()