Method parse

codeclash/arenas/figgie/replay.py:79–130

(self, raw: bytes, players=None)

    DRAW_JS = DRAW_JS

    def parse(self, raw: bytes, players=None) -> ReplayData:
        d = json.loads(raw.decode())
        np_ = d.get("num_players", 4)
        goal = d.get("goal_suit")
        init = d.get("initial_hands", {})
        names = {i: (players[i]["name"] if players and len(players) > i else f"P{i + 1}") for i in range(np_)}
        ante = 50 if np_ == 4 else 40

        holdings = {i: dict(init.get(str(i), {})) for i in range(np_)}
        cash = {i: 350 - ante for i in range(np_)}
        frames = []
        for idx, e in enumerate(d.get("events", [])):
            trade = actions = None
            if e.get("type") == "trade":
                b, s, su, pr = e["buyer"], e["seller"], e["suit"], e["price"]
                holdings[b][su] = holdings[b].get(su, 0) + 1
                holdings[s][su] = holdings[s].get(su, 0) - 1
                cash[b] -= pr
                cash[s] += pr
                trade = {"buyer": b, "seller": s, "suit": su, "price": pr}
            else:
                actions = [
                    {"p": a["player"], "valid": a.get("valid", True), **a.get("action", {})}
                    for a in e.get("actions", [])
                ]
            frames.append(
                {
                    "turn": idx,
                    "tick": e.get("tick"),
                    "kind": e.get("type"),
                    "holdings": {i: dict(holdings[i]) for i in range(np_)},
                    "cash": dict(cash),
                    "trade": trade,
                    "actions": actions,
                }
            )

        scores = {int(k): v for k, v in d.get("scores", {}).items()}
        ranked = sorted(scores.items(), key=lambda kv: kv[1], reverse=True)
        if not ranked or (len(ranked) > 1 and ranked[0][1] == ranked[1][1]):
            winner, draw = None, bool(ranked)
        else:
            winner, draw = names.get(ranked[0][0], f"P{ranked[0][0] + 1}"), False

        return ReplayData(
            w=1,
            h=1,
            frames=frames,
            winner=winner,
            draw=draw,
            extra={"names": names, "goal": goal, "scores": scores, "np": np_},
        )