add export to json of games

This commit is contained in:
Maximilian Keßler 2023-08-08 13:53:48 +02:00
parent 330baff33c
commit 5a2329fa0b
Signed by: max
GPG Key ID: BCC5A619923C0BA5
2 changed files with 20 additions and 2 deletions

View File

@ -1,6 +1,6 @@
[project] [project]
name = "hanabi" name = "hanabi"
version = "1.1.2" version = "1.1.3"
description = "Hanabi interface" description = "Hanabi interface"
readme = "README.md" readme = "README.md"
license = { file = "LICENSE" } license = { file = "LICENSE" }

View File

@ -3,7 +3,7 @@ from typing import List, Dict, Tuple
from hanab_game import Action, ParseError from hanab_game import Action, ParseError
from hanabi import hanab_game from hanabi import hanab_game
from hanabi import constants from hanabi import constants
from hanabi.live import variants from hanabi.live import variants, compress
class HanabLiveInstance(hanab_game.HanabiInstance): class HanabLiveInstance(hanab_game.HanabiInstance):
@ -16,6 +16,8 @@ class HanabLiveInstance(hanab_game.HanabiInstance):
one_less_card: bool = False, one_less_card: bool = False,
*args, **kwargs *args, **kwargs
): ):
self.one_extra_card = one_extra_card
self.one_less_card = one_less_card
assert 2 <= num_players <= 6 assert 2 <= num_players <= 6
hand_size = constants.HAND_SIZES[num_players] hand_size = constants.HAND_SIZES[num_players]
if one_less_card: if one_less_card:
@ -87,6 +89,22 @@ class HanabLiveGameState(hanab_game.GameState):
super().__init__(instance) super().__init__(instance)
self.instance: HanabLiveInstance = instance self.instance: HanabLiveInstance = instance
def to_json(self):
return {
"actions": compress.compress_actions(self.actions),
"deck": compress.compress_deck(self.deck),
"players": ["Alice", "Bob", "Cathy", "Donald", "Emily", "Frank"][:self.num_players],
"notes": [[]] * self.num_players,
"options": {
"variant": self.instance.variant_id,
"deckPlays": self.instance.deck_plays,
"oneExtraCard": self.instance.one_extra_card,
"oneLessCard": self.instance.one_less_card,
"allOrNothing": self.instance.all_or_nothing,
"startingPlayer": self.instance.starting_player
}
}
def make_action(self, action): def make_action(self, action):
match action.type: match action.type:
case hanab_game.ActionType.ColorClue | hanab_game.ActionType.RankClue: case hanab_game.ActionType.ColorClue | hanab_game.ActionType.RankClue: