update

TwitchChannelPointsMiner/classes/entities/EventPrediction.py

@@ -0,0 +1,94 @@
+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