import os import sys from ConfigParser import ConfigParser from itertools import chain from bpython.keys import key_dispatch import errno class Struct(object): """Simple class for instantiating objects we can add arbitrary attributes to and use for various arbitrary things.""" def fill_config_with_default_values(config, default_values): for section in default_values.iterkeys(): if not config.has_section(section): config.add_section(section) for (opt, val) in default_values[section].iteritems(): if not config.has_option(section, opt): config.set(section, opt, str(val)) def loadini(struct, configfile): """Loads .ini configuration file and stores its values in struct""" config_path = os.path.expanduser(configfile) if not os.path.isfile(config_path) and configfile == '~/.bpython/config': # FIXME: I decided ~/.bpython.ini was a crappy place for a config file, # so this is just a fallback if the default is passed - remove this # eventually please. config_path = os.path.expanduser('~/.bpython.ini') config = ConfigParser() fill_config_with_default_values(config, { 'general': { 'arg_spec': True, 'auto_display_list': True, 'color_scheme': 'default', 'dedent_after': 1, 'flush_output': True, 'highlight_show_source': True, 'hist_file': '~/.pythonhist', 'hist_length': 100, 'paste_time': 0.02, 'syntax': True, 'tab_length': 4, 'pastebin_confirm': True, 'pastebin_private': False, 'pastebin_url': 'http://bpaste.net/xmlrpc/', 'pastebin_show_url': 'http://bpaste.net/show/$paste_id/', }, 'keyboard': { 'clear_line': 'C-u', 'clear_screen': 'C-l', 'clear_word': 'C-w', 'cut_to_buffer': 'C-k', 'delete': 'C-d', 'down_one_line': 'C-n', 'exit': '', 'last_output': 'F9', 'pastebin': 'F8', 'save': 'C-s', 'show_source': 'F2', 'undo': 'C-r', 'up_one_line': 'C-p', 'yank_from_buffer': 'C-y'}}) config.read(config_path) struct.dedent_after = config.getint('general', 'dedent_after') struct.tab_length = config.getint('general', 'tab_length') struct.auto_display_list = config.getboolean('general', 'auto_display_list') struct.syntax = config.getboolean('general', 'syntax') struct.arg_spec = config.getboolean('general', 'arg_spec') struct.paste_time = config.getfloat('general', 'paste_time') struct.highlight_show_source = config.getboolean('general', 'highlight_show_source') struct.hist_file = config.get('general', 'hist_file') struct.hist_length = config.getint('general', 'hist_length') struct.flush_output = config.getboolean('general', 'flush_output') struct.pastebin_key = config.get('keyboard', 'pastebin') struct.save_key = config.get('keyboard', 'save') struct.show_source_key = config.get('keyboard', 'show_source') struct.undo_key = config.get('keyboard', 'undo') struct.up_one_line_key = config.get('keyboard', 'up_one_line') struct.down_one_line_key = config.get('keyboard', 'down_one_line') struct.cut_to_buffer_key = config.get('keyboard', 'cut_to_buffer') struct.yank_from_buffer_key = config.get('keyboard', 'yank_from_buffer') struct.clear_word_key = config.get('keyboard', 'clear_word') struct.clear_line_key = config.get('keyboard', 'clear_line') struct.clear_screen_key = config.get('keyboard', 'clear_screen') struct.delete_key = config.get('keyboard', 'delete') struct.exit_key = config.get('keyboard', 'exit') struct.last_output_key = config.get('keyboard', 'last_output') struct.pastebin_confirm = config.getboolean('general', 'pastebin_confirm') struct.pastebin_private = config.getboolean('general', 'pastebin_private') struct.pastebin_url = config.get('general', 'pastebin_url') struct.pastebin_show_url = config.get('general', 'pastebin_show_url') color_scheme_name = config.get('general', 'color_scheme') default_colors = { 'keyword': 'y', 'name': 'c', 'comment': 'b', 'string': 'm', 'error': 'r', 'number': 'G', 'operator': 'Y', 'punctuation': 'y', 'token': 'C', 'background': 'd', 'output': 'w', 'main': 'c', 'paren': 'R', 'prompt': 'c', 'prompt_more': 'g', } if color_scheme_name == 'default': struct.color_scheme = default_colors else: path = os.path.expanduser('~/.bpython/%s.theme' % (color_scheme_name,)) load_theme(struct, path, config_path, default_colors) # checks for valid key configuration this part still sucks for key in (struct.pastebin_key, struct.save_key): key_dispatch[key] def load_theme(struct, path, inipath, default_colors): theme = ConfigParser() try: f = open(path, 'r') except (IOError, OSError), e: sys.stdout.write("Error loading theme file specified in '%s':\n%s\n" % (inipath, e)) sys.exit(1) theme.readfp(f) struct.color_scheme = {} for k, v in chain(theme.items('syntax'), theme.items('interface')): if theme.has_option('syntax', k): struct.color_scheme[k] = theme.get('syntax', k) else: struct.color_scheme[k] = theme.get('interface', k) # Check against default theme to see if all values are defined for k, v in default_colors.iteritems(): if k not in struct.color_scheme: struct.color_scheme[k] = v f.close() def migrate_rc(path): """Use the shlex module to convert the old configuration file to the new format. The old configuration file is renamed but not removed by now.""" import shlex f = open(path) parser = shlex.shlex(f) bools = { 'true': True, 'yes': True, 'on': True, 'false': False, 'no': False, 'off': False} config = ConfigParser() config.add_section('general') while True: k = parser.get_token() v = None if not k: break k = k.lower() if parser.get_token() == '=': v = parser.get_token() or None if v is not None: try: v = int(v) except ValueError: if v.lower() in bools: v = bools[v.lower()] config.set('general', k, v) f.close() try: os.makedirs(os.path.expanduser('~/.bpython')) except OSError, e: if e.errno != errno.EEXIST: raise f = open(os.path.expanduser('~/.bpython/config'), 'w') config.write(f) f.close() os.rename(path, os.path.expanduser('~/.bpythonrc.bak')) print ("The configuration file for bpython has been changed. A new " "config file has been created as ~/.bpython/config") print ("The existing .bpythonrc file has been renamed to .bpythonrc.bak " "and it can be removed.") print "Press enter to continue." raw_input()