🐛 修复 histories 只有一条时推算数据出现除数为0的bug

This commit is contained in:
2024-05-15 08:56:22 +08:00
parent b8cf10b45d
commit 4c7cd00a76

View File

@@ -172,9 +172,10 @@ async def query_historical_data(user: User, user_info: UserInfoSuccess) -> list[
# 按照时间排序
histories = sorted(histories, key=lambda x: x.record_at)
print(histories)
for index, value in enumerate(histories):
# 在历史记录里找有没有今天0点后的数据
if value.record_at > today:
# 在历史记录里找有没有今天0点后的数据, 并且至少要有两个数据点
if value.record_at > today and len(histories) >= 2: # noqa: PLR2004
histories = histories[:index] + [
get_specified_point(histories[index - 1], histories[index], today.replace(microsecond=1000))
]