feat: add method to recalculate streak from recent matches
All checks were successful
continuous-integration/drone/push Build is passing

- Add recalculate_streak_from_recent_matches to Friend class
- Fetches last 20 matches and recalculates win/loss streaks
- Sorts matches chronologically to ensure accurate calculation
This commit is contained in:
Ching L 2025-09-11 15:26:18 +08:00
parent 7498f413bf
commit bb4ee378d9

42
dota.py
View File

@ -220,6 +220,48 @@ class Friend(BaseModel):
return data return data
def recalculate_streak_from_recent_matches(self):
"""获取近20场比赛并重新计算连胜连败记录"""
try:
# 获取近20场比赛
recent_matches = self.get_recent_matches(limit=20)
if not recent_matches:
logger.warning(f"No recent matches found for {self.name}")
return False
# 按时间从旧到新排序start_time升序
recent_matches.sort(key=lambda x: x.start_time)
# 重置连胜连败计数
self.win_streak = 0
self.loss_streak = 0
# 从最旧的比赛开始计算连胜连败
for match in recent_matches:
# 判断是否获胜
player_won = match.radiant_win == (match.player_slot < 128)
if player_won:
self.win_streak += 1
self.loss_streak = 0
else:
self.loss_streak += 1
self.win_streak = 0
# 更新最后一场比赛的ID避免重复计算
if recent_matches:
self.last_match_id = recent_matches[-1].match_id
# 保存更新后的数据
self.save()
logger.info(f"Updated streak for {self.name}: {self.win_streak} wins, {self.loss_streak} losses")
return True
except Exception as e:
logger.error(f"Failed to recalculate streak for {self.name}: {e}")
return False
def update_streak(self, match_id, win): def update_streak(self, match_id, win):
"""更新连胜连败计数""" """更新连胜连败计数"""
# 避免重复计算同一场比赛 # 避免重复计算同一场比赛