Mercurial > touhou
diff pytouhou/options.py @ 567:b2269b9c6119
Add a configuration parser, and pass those options to argparse as defaults. Also include an xdg helper.
author | Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> |
---|---|
date | Tue, 01 Jul 2014 23:17:40 +0200 |
parents | 5f7f859a72f9 |
children | e7a4731a278b |
line wrap: on
line diff
--- a/pytouhou/options.py +++ b/pytouhou/options.py @@ -12,25 +12,108 @@ ## GNU General Public License for more details. ## -from argparse import ArgumentParser +import os +from ConfigParser import RawConfigParser, NoOptionError + +from pytouhou.utils.xdg import load_config_paths + + +class Options(object): + def __init__(self, name, defaults): + load_paths = list(reversed([os.path.join(directory, '%s.cfg' % name) + for directory + in load_config_paths(name)])) + + self.config = RawConfigParser(defaults) + self.paths = self.config.read(load_paths) + self.section = name if self.config.has_section(name) else 'DEFAULT' + + def get(self, option): + try: + return self.config.get(self.section, option) + except NoOptionError: + return None + + +def patch_argument_parser(): + from argparse import ArgumentParser, _ActionsContainer + + original_method = _ActionsContainer.add_argument + + def add_argument(self, *args, **kwargs): + if 'default' not in kwargs: + dest = kwargs.get('dest') + if dest is None: + for dest in args: + dest = dest.lstrip('-') + value = self.default.get(dest) + if value is not None: + break + else: + dest = dest.replace('_', '-') + value = self.default.get(dest) + if value is not None: + argument_type = kwargs.get('type') + if argument_type is not None: + value = argument_type(value) + action = kwargs.get('action') + if action == 'store_true': + value = value.lower() == 'true' + elif action == 'store_false': + value = value.lower() != 'true' + if kwargs.get('nargs') == '*' and isinstance(value, str): + value = value.split() + kwargs['default'] = value + elif dest == 'double-buffer': + kwargs['default'] = None + return original_method(self, *args, **kwargs) + _ActionsContainer.add_argument = add_argument + + class Parser(ArgumentParser): + def __init__(self, *args, **kwargs): + self.default = kwargs.pop('default') + ArgumentParser.__init__(self, *args, **kwargs) + + def add_argument_group(self, *args, **kwargs): + group = ArgumentParser.add_argument_group(self, *args, **kwargs) + group.default = self.default + group.add_argument_group = self.add_argument_group + group.add_mutually_exclusive_group = self.add_mutually_exclusive_group + return group + + def add_mutually_exclusive_group(self, *args, **kwargs): + group = ArgumentParser.add_mutually_exclusive_group(self, *args, **kwargs) + group.default = self.default + group.add_argument_group = self.add_argument_group + group.add_mutually_exclusive_group = self.add_mutually_exclusive_group + return group + + return Parser + + +ArgumentParser = patch_argument_parser() + + +def parse_config(section, defaults): + return Options(section, defaults) def parse_arguments(defaults): - parser = ArgumentParser(description='Libre reimplementation of the Touhou 6 engine.') + parser = ArgumentParser(description='Libre reimplementation of the Touhou 6 engine.', default=defaults) - parser.add_argument('data', metavar='DAT', default=defaults['data'], nargs='*', help='Game’s data files') - parser.add_argument('-p', '--path', metavar='DIRECTORY', default='.', help='Game directory path.') + parser.add_argument('data', metavar='DAT', nargs='*', help='Game’s data files') + parser.add_argument('-p', '--path', metavar='DIRECTORY', help='Game directory path.') parser.add_argument('--debug', action='store_true', help='Set unlimited continues, and perhaps other debug features.') parser.add_argument('--verbosity', metavar='VERBOSITY', choices=['DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'], help='Select the wanted logging level.') game_group = parser.add_argument_group('Game options') - game_group.add_argument('-s', '--stage', metavar='STAGE', type=int, default=None, help='Stage, 1 to 7 (Extra), nothing means story mode.') - game_group.add_argument('-r', '--rank', metavar='RANK', type=int, default=0, help='Rank, from 0 (Easy, default) to 3 (Lunatic).') - game_group.add_argument('-c', '--character', metavar='CHARACTER', type=int, default=0, help='Select the character to use, from 0 (ReimuA, default) to 3 (MarisaB).') + game_group.add_argument('-s', '--stage', metavar='STAGE', type=int, help='Stage, 1 to 7 (Extra), nothing means story mode.') + game_group.add_argument('-r', '--rank', metavar='RANK', type=int, help='Rank, from 0 (Easy, default) to 3 (Lunatic).') + game_group.add_argument('-c', '--character', metavar='CHARACTER', type=int, help='Select the character to use, from 0 (ReimuA, default) to 3 (MarisaB).') game_group.add_argument('-b', '--boss-rush', action='store_true', help='Fight only bosses.') - game_group.add_argument('--game', metavar='GAME', choices=['EoSD'], default='EoSD', help='Select the game engine to use.') - game_group.add_argument('--interface', metavar='INTERFACE', choices=['EoSD', 'Sample'], default='EoSD', help='Select the interface to use.') - game_group.add_argument('--hints', metavar='HINTS', default=None, help='Hints file, to display text while playing.') + game_group.add_argument('--game', metavar='GAME', choices=['EoSD'], help='Select the game engine to use.') + game_group.add_argument('--interface', metavar='INTERFACE', choices=['EoSD', 'Sample'], help='Select the interface to use.') + game_group.add_argument('--hints', metavar='HINTS', help='Hints file, to display text while playing.') replay_group = parser.add_argument_group('Replay options') replay_group.add_argument('--replay', metavar='REPLAY', help='Select a file to replay.') @@ -38,23 +121,23 @@ def parse_arguments(defaults): replay_group.add_argument('--skip-replay', action='store_true', help='Skip the replay and start to play when it’s finished.') netplay_group = parser.add_argument_group('Netplay options') - netplay_group.add_argument('--port', metavar='PORT', type=int, default=0, help='Local port to use.') - netplay_group.add_argument('--remote', metavar='REMOTE', default=None, help='Remote address.') + netplay_group.add_argument('--port', metavar='PORT', type=int, help='Local port to use.') + netplay_group.add_argument('--remote', metavar='REMOTE', help='Remote address.') netplay_group.add_argument('--friendly-fire', action='store_true', help='Allow friendly-fire during netplay.') graphics_group = parser.add_argument_group('Graphics options') - graphics_group.add_argument('--backend', metavar='BACKEND', choices=['opengl', 'sdl'], default=['opengl', 'sdl'], nargs='*', help='Which backend to use (opengl or sdl).') - graphics_group.add_argument('--fps-limit', metavar='FPS', default=-1, type=int, help='Set fps limit. A value of 0 disables fps limiting, while a negative value limits to 60 fps if and only if vsync doesn’t work.') + graphics_group.add_argument('--backend', metavar='BACKEND', choices=['opengl', 'sdl'], nargs='*', help='Which backend to use (opengl or sdl).') + graphics_group.add_argument('--fps-limit', metavar='FPS', type=int, help='Set fps limit. A value of 0 disables fps limiting, while a negative value limits to 60 fps if and only if vsync doesn’t work.') graphics_group.add_argument('--no-background', action='store_false', help='Disable background display (huge performance boost on slow systems).') graphics_group.add_argument('--no-particles', action='store_false', help='Disable particles handling (huge performance boost on slow systems).') graphics_group.add_argument('--no-sound', action='store_false', help='Disable music and sound effects.') opengl_group = parser.add_argument_group('OpenGL backend options') - opengl_group.add_argument('--gl-flavor', choices=['core', 'es', 'compatibility', 'legacy'], default='compatibility', help='OpenGL profile to use.') - opengl_group.add_argument('--gl-version', default=2.1, type=float, help='OpenGL version to use.') + opengl_group.add_argument('--gl-flavor', choices=['core', 'es', 'compatibility', 'legacy'], help='OpenGL profile to use.') + opengl_group.add_argument('--gl-version', type=float, help='OpenGL version to use.') double_buffer = opengl_group.add_mutually_exclusive_group() - double_buffer.add_argument('--double-buffer', dest='double_buffer', action='store_true', default=None, help='Enable double buffering.') - double_buffer.add_argument('--single-buffer', dest='double_buffer', action='store_false', default=None, help='Disable double buffering.') + double_buffer.add_argument('--double-buffer', dest='double_buffer', action='store_true', help='Enable double buffering.') + double_buffer.add_argument('--single-buffer', dest='double_buffer', action='store_false', help='Disable double buffering.') return parser.parse_args()