X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Factions.py;h=32ac03509c88527cf8d5205d0bfa0a490456c674;hb=7ba98383908eb9546ee78a14dad90057dc65757b;hp=5599bd73b2d724e6986bc8a58d3ecf7e3ba2da16;hpb=d9ded620200ed74517f17f83f01313470ffca222;p=naja.git diff --git a/naja/actions.py b/naja/actions.py index 5599bd7..32ac035 100644 --- a/naja/actions.py +++ b/naja/actions.py @@ -9,7 +9,11 @@ class LocationAction(object): TEXT = None def __init__(self, required_bits, **data): - self.required_bits = frozenset(required_bits) + bits = set() + for bit in required_bits: + # Convert names to numbers if applicable. + bits.add(BITS.get(bit, bit)) + self.required_bits = frozenset(bits) self.data = data def check_available(self, player): @@ -25,6 +29,11 @@ class LocationAction(object): else: return False + def export(self): + return {'required_bits': list(self.required_bits), + 'data': self.data, + 'action_class': self.__class__.__name__} + class DoNothing(LocationAction): TEXT = "No effect." @@ -33,7 +42,7 @@ class DoNothing(LocationAction): pass -class LoseHeathOrMSB(LocationAction): +class LoseHealthOrMSB(LocationAction): TEXT = "Lose health. If MSB is set, it will be cleared instead." def perform_action(self, board, location):