OrgTJWater/TJWaterServer
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refine

Browse Source
This commit is contained in:
DingZQ
2025-04-07 22:14:00 +08:00
parent e07bbb70f3
commit 603cb1f2aa
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@@ -2326,7 +2326,7 @@ async def fastapi_query_all_records_by_date(querydate: str) -> dict[str, list]:
global redis_client global redis_client
# 今天的不要去缓存 # 今天的不要去缓存
if not time_api.is_today(querydate): if not time_api.is_today_or_future(querydate):
cache_key = f"queryallrecordsbydate_{querydate}" cache_key = f"queryallrecordsbydate_{querydate}"
data = redis_client.get(cache_key) data = redis_client.get(cache_key)
if data: if data:
@@ -2339,7 +2339,7 @@ async def fastapi_query_all_records_by_date(querydate: str) -> dict[str, list]:
"links": nodes_links[1] } "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)) redis_client.set(cache_key, msgpack.packb(results, default=encode_datetime))
return results return results

6
time_api.py
View File

@@ -134,4 +134,10 @@ def is_tomorrow(query_date: str) -> bool:
dt = parse_beijing_time(query_date) dt = parse_beijing_time(query_date)
return dt.date() == (datetime.now().date() + timedelta(days=1)) 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()