X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Fgamestate.py;h=a32438170e3c5663366850cde744027507b57ceb;hb=3e460450c1baa9af43f5391ba4673a599071656b;hp=19051a42b8c1d450d4f4b43a0662610f4ad56825;hpb=aa858388f65344396a7b81a82198e3a6fbc1efca;p=naja.git diff --git a/naja/gamestate.py b/naja/gamestate.py index 19051a4..a324381 100644 --- a/naja/gamestate.py +++ b/naja/gamestate.py @@ -2,15 +2,23 @@ The current game state. """ -import yaml +try: + import yaml +except ImportError: + yaml = None + import json from naja.gameboard import GameBoard from naja.resources import resources def load_location_deck(name): - with resources.get_file('location_decks', '%s.yaml' % (name,)) as deck_fp: - return yaml.safe_load(deck_fp) + if yaml: + with resources.get_file('location_decks', '%s.yaml' % name) as deck_fp: + return yaml.safe_load(deck_fp) + else: + with resources.get_file('location_decks', '%s.json' % name) as deck_fp: + return json.load(deck_fp) class GameState(object):