95 lines
2.7 KiB
Python
95 lines
2.7 KiB
Python
from TwitchChannelPointsMiner.classes.entities.Bet import Bet
|
|
from TwitchChannelPointsMiner.classes.entities.Streamer import Streamer
|
|
from TwitchChannelPointsMiner.classes.Settings import Settings
|
|
from TwitchChannelPointsMiner.utils import _millify, float_round
|
|
|
|
|
|
class EventPrediction(object):
|
|
__slots__ = [
|
|
"streamer",
|
|
"event_id",
|
|
"title",
|
|
"created_at",
|
|
"prediction_window_seconds",
|
|
"status",
|
|
"result",
|
|
"box_fillable",
|
|
"bet_confirmed",
|
|
"bet_placed",
|
|
"bet",
|
|
]
|
|
|
|
def __init__(
|
|
self,
|
|
streamer: Streamer,
|
|
event_id,
|
|
title,
|
|
created_at,
|
|
prediction_window_seconds,
|
|
status,
|
|
outcomes,
|
|
):
|
|
self.streamer = streamer
|
|
|
|
self.event_id = event_id
|
|
self.title = title.strip()
|
|
self.created_at = created_at
|
|
self.prediction_window_seconds = prediction_window_seconds
|
|
self.status = status
|
|
self.result: dict = {"string": "", "type": None, "gained": 0}
|
|
|
|
self.box_fillable = False
|
|
self.bet_confirmed = False
|
|
self.bet_placed = False
|
|
self.bet = Bet(outcomes, streamer.settings.bet)
|
|
|
|
def __repr__(self):
|
|
return f"EventPrediction(event_id={self.event_id}, streamer={self.streamer}, title={self.title})"
|
|
|
|
def __str__(self):
|
|
return (
|
|
f"EventPrediction: {self.streamer} - {self.title}"
|
|
if Settings.logger.less
|
|
else self.__repr__()
|
|
)
|
|
|
|
def elapsed(self, timestamp):
|
|
return float_round((timestamp - self.created_at).total_seconds())
|
|
|
|
def closing_bet_after(self, timestamp):
|
|
return float_round(self.prediction_window_seconds - self.elapsed(timestamp))
|
|
|
|
def print_recap(self) -> str:
|
|
return f"{self}\n\t\t{self.bet}\n\t\tResult: {self.result['string']}"
|
|
|
|
def parse_result(self, result) -> dict:
|
|
result_type = result["type"]
|
|
|
|
points = {}
|
|
points["placed"] = (
|
|
self.bet.decision["amount"] if result_type != "REFUND" else 0
|
|
)
|
|
points["won"] = (
|
|
result["points_won"]
|
|
if result["points_won"] or result_type == "REFUND"
|
|
else 0
|
|
)
|
|
points["gained"] = (
|
|
points["won"] - points["placed"] if result_type != "REFUND" else 0
|
|
)
|
|
points["prefix"] = "+" if points["gained"] >= 0 else ""
|
|
|
|
action = (
|
|
"Lost"
|
|
if result_type == "LOSE"
|
|
else ("Refunded" if result_type == "REFUND" else "Gained")
|
|
)
|
|
|
|
self.result = {
|
|
"string": f"{result_type}, {action}: {points['prefix']}{_millify(points['gained'])}",
|
|
"type": result_type,
|
|
"gained": points["gained"],
|
|
}
|
|
|
|
return points
|