X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Foptions.py;h=498a979d8ecae6299cbf4d5d280b89bd6bfb2a46;hb=HEAD;hp=d37f1be1d459861b37db7ff5552d6290b44f1170;hpb=a97bdc61b9c4049e2aa20278c814b8bb8b8273cc;p=naja.git

diff --git a/naja/options.py b/naja/options.py
index d37f1be..498a979 100644
--- a/naja/options.py
+++ b/naja/options.py
@@ -9,39 +9,33 @@ from naja.constants import DEFAULTS
 options = AttrDict()
 
 
-def check_min_max(option, value, min_value, max_value):
-    '''
-    Check value lies between min and max and raise OptionValueError if it does
-    not.
-    '''
-    if not (min_value <= value <= max_value):
-        raise optparse.OptionValueError(
-            "Value of %s should be between %s and %s but got: %r"
-            % (option.dest, min, max, value))
-
-
-def load_game(option, opt_str, value, parser):
+def load_game(parser, slot_num):
     '''
     Load a save game and store it in parser.values.game_state.
     '''
-    check_min_max(option, value, 0, 7)
-    # madness takes its toll ...
-    options.save_location = parser.values.save_location
     from naja.scenes.load_save import SaveGameSlot
-    state = SaveGameSlot(value).load()
+    if not (0 <= slot_num <= 7):
+        parser.error("--load accepts a slot number from 0 to 7.")
+    state = SaveGameSlot(slot_num).load()
     if state is None:
-        raise optparse.OptionValueError(
-            "Could not load game from slot %s" % (value,))
-    parser.values.game_state = state
+        raise parser.error(
+            "Could not load game from slot %s" % (slot_num,))
+    options.game_state = state
 
 
-def load_deck(option, opt_str, value, parser):
+def load_deck(parser, deck):
     '''
     Create a new game for a specific deck  and store it in
     parser.values.game_state.
     '''
-    raise optparse.OptionalValueError(
-        "Deck loading not implemented.")
+    from naja.gamestate import GameState
+    try:
+        state = GameState.new(deck=deck)
+    except:
+        if options.debug:
+            raise
+        parser.error("Could not load deck %r" % (deck,))
+    options.game_state = state
 
 
 def parse_args(args):
@@ -67,18 +61,20 @@ def parse_args(args):
     parser.add_option("--save-location", default=_get_default_save_location(),
                       dest="save_location", help="Saved game location")
 
+    parser.add_option('--deck', default=None,
+                      help='Start with a new game for a specific deck'
+                      ' (bypassing the menu).')
+
+    parser.add_option('--load', default=None, type=int,
+                      help='Start with a specific save game loaded'
+                      ' (bypassing the menu). Slot values are 0-7.')
+
     if options.debug:
         parser.add_option('--initial-bits', type=int,
                           help='Initial player bits')
         parser.add_option('--cheat-enabled', default=False,
                           action='store_true',
                           help='For those too lazy to type the KONAMI code')
-        parser.add_option('--deck', default=None, action="callback",
-                          callback=load_deck,
-                          help='Start with a new game for a specific deck')
-        parser.add_option('--load', default=None, type=int, action="callback",
-                          callback=load_game,
-                          help='Start with a specific save game loaded (0-7)')
 
     opts, _ = parser.parse_args(args)
 
@@ -86,6 +82,11 @@ def parse_args(args):
         if getattr(opts, k, None) is not None:
             options[k] = getattr(opts, k)
 
+    if opts.load is not None:
+        load_game(parser, opts.load)
+    if opts.deck is not None:
+        load_deck(parser, opts.deck)
+
 
 def _get_default_save_location():
     """Return a default save game location."""