forked from c0de/BaseballBot
115 lines
3.8 KiB
Python
115 lines
3.8 KiB
Python
import math
|
|
|
|
# Import game functions
|
|
from database.models import (
|
|
GameModel as Game,
|
|
GuessModel as Guess,
|
|
PlayerModel as Player,
|
|
)
|
|
|
|
class ProcessGuess:
|
|
def __init__(self, game, **kwargs):
|
|
self.game_manager = game
|
|
|
|
self.message = kwargs.get("message")
|
|
self.pitch_value = kwargs.get("pitch_value")
|
|
self.difference = kwargs.get("difference")
|
|
self.difference_score = kwargs.get("difference_score")
|
|
self.guesses = kwargs.get("guesses")
|
|
self.guess = kwargs.get("guess")
|
|
|
|
def get_guesses(self):
|
|
# Get all guesses for this game as a list of combo Guess + Player models,
|
|
# excluding invalid results, from lowest to highest
|
|
# http://docs.peewee-orm.com/en/latest/peewee/query_examples.html#joins-and-subqueries
|
|
self.guesses = (
|
|
Guess.select(
|
|
Guess.guess,
|
|
Player.player_id,
|
|
Player.player_name,
|
|
Player.total_points,
|
|
)
|
|
.join(Player)
|
|
.where(
|
|
(Guess.game_id == self.game_manager.game.game_id)
|
|
& (Guess.guess > 0)
|
|
& (Guess.player.player_id == Player.player_id)
|
|
)
|
|
.order_by(Guess.guess)
|
|
)
|
|
return self.guesses
|
|
|
|
def update_difference_value(self):
|
|
# Update player's difference in guessed value
|
|
Guess.update({"difference": self.difference}).where(
|
|
(Guess.game_id == self.game_manager.game.game_id)
|
|
& (Guess.player.player_id == self.guess.player.player_id)
|
|
& (Guess.guess_id == self.guess.guess_id)
|
|
).execute()
|
|
|
|
def update_player_total_points(self):
|
|
# Update player's total score
|
|
Player.update(
|
|
{
|
|
"total_points": math.floor(
|
|
self.guess.player.total_points + self.difference_score
|
|
)
|
|
}
|
|
).where(Player.player_id == self.guess.player.player_id).execute()
|
|
|
|
def get_difference(self, guess=None):
|
|
"""Difference calculation, includes "loop over" effect"""
|
|
if not guess:
|
|
guess = self.guess
|
|
|
|
difference = abs(guess - self.pitch_value)
|
|
|
|
if difference > 500:
|
|
return 1000 - difference
|
|
|
|
self.difference = difference
|
|
|
|
self.update_difference_value()
|
|
return self.difference
|
|
|
|
def get_difference_score(self):
|
|
self.difference_score = 0
|
|
|
|
self.update_player_total_points()
|
|
return self.difference_score
|
|
|
|
def get_winner_loser(self):
|
|
# Determine which guesses are closest and furthest from the pitch_value
|
|
guess_values = [record.guess for record in self.guesses]
|
|
# Closest to the pitch_value
|
|
winner = min(guess_values, key=lambda guess: self.get_difference(guess))
|
|
# Furthest from the pitch_value
|
|
loser = max(guess_values, key=lambda guess: self.get_difference(guess))
|
|
|
|
return winner, loser
|
|
|
|
def process_guesses(self):
|
|
"""
|
|
Iterates through the guesses for this game, and appends to the message string
|
|
the results of how well that player performed this round.
|
|
|
|
Uses the pitch_value to determine the difference from their guess to the correct score
|
|
"""
|
|
winner, loser = self.get_winner_loser()
|
|
|
|
for guess in self.get_guesses():
|
|
self.guess = guess
|
|
|
|
difference = self.get_difference()
|
|
difference_score = self.get_difference_score()
|
|
|
|
self.message += f"{guess.player.player_name} | {guess.guess} | {difference} | {difference_score} | {guess.player.total_points}\n"
|
|
|
|
if guess.guess == winner:
|
|
closest_player_id = guess.player.player_id
|
|
|
|
if guess.guess == loser:
|
|
furthest_player_id = guess.player.player_id
|
|
|
|
return self.message, closest_player_id, furthest_player_id
|