naja/options.py

   1 import optparse
   2 import os
   3 import sys
   4
   5 from naja.attrdict import AttrDict
   6 from naja.constants import DEFAULTS
   7
   8
   9 options = AttrDict()
  10
  11
  12 def load_game(parser, slot_num):
  13     '''
  14     Load a save game and store it in parser.values.game_state.
  15     '''
  16     from naja.scenes.load_save import SaveGameSlot
  17     if not (0 <= slot_num <= 7):
  18         parser.error("--load accepts a slot number from 0 to 7.")
  19     state = SaveGameSlot(slot_num).load()
  20     if state is None:
  21         raise parser.error(
  22             "Could not load game from slot %s" % (slot_num,))
  23     options.game_state = state
  24
  25
  26 def load_deck(parser, deck):
  27     '''
  28     Create a new game for a specific deck  and store it in
  29     parser.values.game_state.
  30     '''
  31     from naja.gamestate import GameState
  32     try:
  33         state = GameState.new(deck=deck, max_health=4, wins_required=4)
  34     except:
  35         parser.error("Could not load deck %r" % (deck,))
  36     options.game_state = state
  37
  38
  39 def parse_args(args):
  40     '''
  41     Parse arguments and store them in the options dictionary.
  42
  43     Note: If you add arguments, you need to add an appropriate default to the
  44     DEFAULTS dict.
  45     '''
  46     options.update(DEFAULTS)
  47
  48     options.debug = 'DEBUG' in os.environ
  49
  50     parser = optparse.OptionParser()
  51     parser.add_option('--no-sound',
  52                       dest='sound', action='store_false', default=True,
  53                       help='Disable all sound, including music')
  54
  55     parser.add_option('--no-music',
  56                       dest='music', action='store_false', default=True,
  57                       help='Disable music (but not sound)')
  58
  59     parser.add_option("--save-location", default=_get_default_save_location(),
  60                       dest="save_location", help="Saved game location")
  61
  62     parser.add_option('--deck', default=None,
  63                       help='Start with a new game for a specific deck'
  64                       ' (bypassing the menu).')
  65
  66     parser.add_option('--load', default=None, type=int,
  67                       help='Start with a specific save game loaded'
  68                       ' (bypassing the menu). Slot values are 0-7.')
  69
  70     if options.debug:
  71         parser.add_option('--initial-bits', type=int,
  72                           help='Initial player bits')
  73         parser.add_option('--cheat-enabled', default=False,
  74                           action='store_true',
  75                           help='For those too lazy to type the KONAMI code')
  76
  77     opts, _ = parser.parse_args(args)
  78
  79     for k in DEFAULTS:
  80         if getattr(opts, k, None) is not None:
  81             options[k] = getattr(opts, k)
  82
  83     if opts.load is not None:
  84         load_game(parser, opts.load)
  85     if opts.deck is not None:
  86         load_deck(parser, opts.deck)
  87
  88
  89 def _get_default_save_location():
  90     """Return a default save game location."""
  91     app = "naja"
  92     if sys.platform.startswith("win"):
  93         if "APPDATA" in os.environ:
  94             return os.path.join(os.environ["APPDATA"], app)
  95         return os.path.join(os.path.expanduser("~"), "." + app)
  96     elif 'XDG_DATA_HOME' in os.environ:
  97         return os.path.join(os.environ["XDG_DATA_HOME"], app)
  98     return os.path.join(os.path.expanduser("~"), ".local", "share", app)