keezen/player.py

from enum import Enum
from command import Option
import json
import jsonpickle

class Color(str, Enum):
    RED = "red"
    BLUE = "blue"
    YELLOW = "yellow"
    GREEN = "green"

    def __getstate__(self):
        return "hallo"

    def __setstate__(state):
        return Color(state)


class Player(object):
    def __init__(self, color = None, name = None):
        self.color = color
        self.name = name
        self.hand = []
        self.options = []
        self.message = "Even wachten"
        self.selectedCard = None
        self.cardIsChanged = False

    def checkOption(self, option):
        foundOption = any(option.command == o.command and o.checkArgs(option) for o in self.options)

        if not foundOption:
            self.message = f"{option.text} is niet toegestaan. " + self.message
            
        return foundOption


class DogEncoder(json.JSONEncoder):
    def default(self, obj):
        if hasattr(obj, '__dict__'):
            return obj.__dict__
        return json.JSONEncoder.default(self, obj)


if __name__ == "__main__":
   player = Player(Color.BLUE, "Groen")
   player.options = [Option("newgame", None, "text")]
   print(player)
   print(player.__dict__)
   print(json.dumps(player, cls=DogEncoder))
   #print(json.dumps(player))
   print(jsonpickle.encode(player))