Method parse

codeclash/arenas/battlesnake/replay.py:69–120

(self, raw: bytes, players=None)

    DRAW_JS = DRAW_JS

    def parse(self, raw: bytes, players=None) -> ReplayData:
        rows = [json.loads(line) for line in raw.decode().splitlines() if line.strip()]
        # per-turn board states (dedupe: keep the last frame seen for each turn)
        by_turn = {}
        for r in rows:
            if isinstance(r, dict) and "board" in r and "turn" in r:
                by_turn[r["turn"]] = r["board"]
        turns = sorted(by_turn)
        result = next((r for r in reversed(rows) if isinstance(r, dict) and "winnerName" in r), {})
        if not turns:
            return ReplayData(w=0, h=0, frames=[], winner=result.get("winnerName"), draw=result.get("isDraw", False))

        # stable color per snake name (prefer the snake's own color from the log if present)
        names, colors = [], {}
        for t in turns:
            for s in by_turn[t]["snakes"]:
                if s["name"] not in names:
                    names.append(s["name"])
        for i, nm in enumerate(names):
            colors[nm] = PALETTE[i % len(PALETTE)]
        for t in turns:
            for s in by_turn[t]["snakes"]:
                if s.get("color"):
                    colors[s["name"]] = s["color"]

        b0 = by_turn[turns[0]]
        frames = []
        for t in turns:
            b = by_turn[t]
            frames.append(
                {
                    "turn": t,
                    "food": [[c["x"], c["y"]] for c in b.get("food", [])],
                    "hazards": [[c["x"], c["y"]] for c in b.get("hazards", [])],
                    "snakes": [
                        {
                            "name": s["name"],
                            "health": s.get("health", 0),
                            "body": [[c["x"], c["y"]] for c in s["body"]],
                        }
                        for s in b["snakes"]
                    ],
                }
            )
        return ReplayData(
            w=b0["width"],
            h=b0["height"],
            frames=frames,
            winner=result.get("winnerName"),
            draw=result.get("isDraw", False),
            extra={"colors": colors},
        )