Mercurial > mamba

annotate mamba/options.py @ 285:cacc03748580

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Move world initialisation to restart
author Stefano Rivera <stefano@rivera.za.net>
date Thu, 15 Sep 2011 23:02:47 +0200
parents bea4857487de
children cc8be536a7fc
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
39
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
1 # global options that are set on startup
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
2
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
3 import optparse
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
4 import os
166
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
5 import sys
39
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
6 from mamba.constants import DEFAULTS
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
7
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
8
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
9 class OptionGetter(object):
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
10 """Object grabber."""
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
11
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
12 def __init__(self, options):
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
13 self._options = options
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
14 self._finalized = False
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
15
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
16 def set_option(self, name, value):
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
17 if self._finalized:
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
18 raise RuntimeError("Cannot set option %r to %r after"
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
19 "options are finalized." % (name, value))
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
20 self._options[name] = value
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
21
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
22 def finalize(self):
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
23 self._finalized = True
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
24
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
25 def __getattr__(self, name):
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
26 if name in self._options:
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
27 return self._options[name]
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
28 raise AttributeError("Unknown option %r" % (name,))
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
29
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
30 options = OptionGetter(DEFAULTS.copy())
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
31
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
32
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
33 def parse_args(args, options=options):
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
34 options.set_option('debug', 'DEBUG' in os.environ)
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
35
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
36 parser = optparse.OptionParser()
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
37 parser.add_option("--no-sound", action="store_false", default=True,
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
38 dest="sound", help="disable sound")
166
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
39 parser.add_option("--save-location", default=_get_default_save_location(),
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
40 dest="save_location", help="Saved game location")
39
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
41
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
42 if options.debug:
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
43 parser.add_option("--level", type="str", default=None,
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
44 dest="level", help="Initial level")
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
45 parser.add_option('--edit', action="store_true", default=False,
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
46 dest="edit", help="Edit given level")
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
47
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
48 opts, _ = parser.parse_args(args)
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
49
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
50 options.set_option('sound', opts.sound)
166
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
51 options.set_option('save_location', opts.save_location)
39
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
52
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
53 if options.debug:
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
54 options.set_option('level', opts.level)
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
55 options.set_option('edit', opts.edit)
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
56
3ab5097e8757 Refactor options.
Simon Cross <hodgestar@gmail.com>
parents:
diff changeset
57 options.finalize()
48
a70ded879f46 Move argument checking into mamba.options.
Simon Cross <hodgestar@gmail.com>
parents: 39
diff changeset
58
a70ded879f46 Move argument checking into mamba.options.
Simon Cross <hodgestar@gmail.com>
parents: 39
diff changeset
59
a70ded879f46 Move argument checking into mamba.options.
Simon Cross <hodgestar@gmail.com>
parents: 39
diff changeset
60 def check_args(options=options):
a70ded879f46 Move argument checking into mamba.options.
Simon Cross <hodgestar@gmail.com>
parents: 39
diff changeset
61 """Check options and complain if they're invalid."""
a70ded879f46 Move argument checking into mamba.options.
Simon Cross <hodgestar@gmail.com>
parents: 39
diff changeset
62 if options.edit and options.level is None:
a70ded879f46 Move argument checking into mamba.options.
Simon Cross <hodgestar@gmail.com>
parents: 39
diff changeset
63 print "You must supply --level when using --edit."
a70ded879f46 Move argument checking into mamba.options.
Simon Cross <hodgestar@gmail.com>
parents: 39
diff changeset
64 return False
a70ded879f46 Move argument checking into mamba.options.
Simon Cross <hodgestar@gmail.com>
parents: 39
diff changeset
65 return True
166
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
66
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
67
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
68 def _get_default_save_location():
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
69 """Return a default save game location."""
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
70 app = "mamba"
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
71 if sys.platform.startswith("win"):
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
72 if "APPDATA" in os.environ:
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
73 return os.path.join(os.environ["APPDATA"], app)
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
74 return os.path.join(os.path.expanduser("~"), "." + app)
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
75 elif 'XDG_DATA_HOME' in os.environ:
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
76 return os.path.join(os.environ["XDG_DATA_HOME"], app)
bea4857487de Game state and level list
Stefano Rivera <stefano@rivera.za.net>
parents: 48
diff changeset
77 return os.path.join(os.path.expanduser("~"), ".local", "share", app)