from __future__importunicode_literals,absolute_import

fromsiximportiterkeys,iteritems

fromsix.moves.configparserimportConfigParser

from .autocompleteimportSIMPLEasdefault_completion,ALL_MODES

from .keysimportcli_key_dispatchascli_key_dispatch

classStruct(object):

defgetpreferredencoding():

returnlocale.getpreferredencoding()orsys.getdefaultencoding()

defcan_encode(c):

try:

c.encode(getpreferredencoding())

exceptUnicodeEncodeError:

defsupports_box_chars():

returnall(map(can_encode,u'│─└┘┌┐'))

defget_config_home():

xdg_config_home=os.environ.get('XDG_CONFIG_HOME','~/.config')

returnos.path.join(xdg_config_home,'bpython')

defdefault_config_path():

returnos.path.join(get_config_home(),'config')

deffill_config_with_default_values(config,default_values):

forsectioniniterkeys(default_values):

ifnotconfig.has_section(section):

config.add_section(section)

for (opt,val)initeritems(default_values[section]):

ifnotconfig.has_option(section,opt):

config.set(section,opt,'%s'% (val, ))

defloadini(struct,configfile):

config_path=os.path.expanduser(configfile)

config=ConfigParser()

defaults= {

'general': {

'arg_spec':True,

'auto_display_list':True,

'autocomplete_mode':default_completion,

'color_scheme':'default',

'complete_magic_methods':True,

'dedent_after':1,

'default_autoreload':False,

'editor':os.environ.get('VISUAL',os.environ.get('EDITOR','vi')),

'flush_output':True,

'highlight_show_source':True,

'hist_duplicates':True,

'hist_file':'~/.pythonhist',

'hist_length':100,

'paste_time':0.02,

'pastebin_confirm':True,

'pastebin_expiry':'1week',

'pastebin_helper':'',

'pastebin_removal_url':'https://bpaste.net/remove/$removal_id',

'pastebin_show_url':'https://bpaste.net/show/$paste_id',

'pastebin_url':'https://bpaste.net/json/new',

'save_append_py':False,

'single_undo_time':1.0,

'syntax':True,

'tab_length':4,

'unicode_box':True

},

'keyboard': {

'backspace':'C-h',

'beginning_of_line':'C-a',

'clear_line':'C-u',

'clear_screen':'C-l',

'clear_word':'C-w',

'copy_clipboard':'F10',

'cut_to_buffer':'C-k',

'delete':'C-d',

'down_one_line':'C-n',

'edit_config':'F3',

'edit_current_block':'C-x',

'end_of_line':'C-e',

'exit':'',

'external_editor':'F7',

'help':'F1',

'incremental_search':'M-s',

'last_output':'F9',

'left':'C-b',

'pastebin':'F8',

'reimport':'F6',

'reverse_incremental_search':'M-r',

'right':'C-f',

'save':'C-s',

'search':'C-o',

'show_source':'F2',

'suspend':'C-z',

'toggle_file_watch':'F5',

'transpose_chars':'C-t',

'undo':'C-r',

'up_one_line':'C-p',

'yank_from_buffer':'C-y'

},

'cli': {

'suggestion_width':0.8,

'trim_prompts':False,

},

'curtsies': {

'list_above':False,

'right_arrow_completion':True,

}}

default_keys_to_commands=dict((value,key)for (key,value)

initeritems(defaults['keyboard']))

fill_config_with_default_values(config,defaults)

ifnotconfig.read(config_path):

ifos.path.isfile(os.path.expanduser('~/.bpython.ini')):

sys.stderr.write("Error: It seems that you have a config file at "

"%s\n"%default_config_path())

sys.exit(1)

defget_key_no_doublebind(command):

default_commands_to_keys=defaults['keyboard']

requested_key=config.get('keyboard',command)

try:

default_command=default_keys_to_commands[requested_key]

if (default_commands_to_keys[default_command]==

config.get('keyboard',default_command)):

setattr(struct,'%s_key'%default_command,'')

exceptKeyError:

struct.config_path=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.single_undo_time=config.getfloat('general','single_undo_time')

struct.highlight_show_source=config.getboolean('general',

'highlight_show_source')

struct.hist_file=config.get('general','hist_file')

struct.editor=config.get('general','editor')

struct.hist_length=config.getint('general','hist_length')

struct.hist_duplicates=config.getboolean('general','hist_duplicates')

struct.flush_output=config.getboolean('general','flush_output')

struct.default_autoreload=config.getboolean('general','default_autoreload')

struct.pastebin_key=get_key_no_doublebind('pastebin')

struct.copy_clipboard_key=get_key_no_doublebind('copy_clipboard')

struct.save_key=get_key_no_doublebind('save')

struct.search_key=get_key_no_doublebind('search')

struct.show_source_key=get_key_no_doublebind('show_source')

struct.suspend_key=get_key_no_doublebind('suspend')

struct.toggle_file_watch_key=get_key_no_doublebind('toggle_file_watch')

struct.undo_key=get_key_no_doublebind('undo')

struct.reimport_key=get_key_no_doublebind('reimport')

struct.reverse_incremental_search_key=get_key_no_doublebind(

'reverse_incremental_search')

struct.incremental_search_key=get_key_no_doublebind('incremental_search')

struct.up_one_line_key=get_key_no_doublebind('up_one_line')

struct.down_one_line_key=get_key_no_doublebind('down_one_line')

struct.cut_to_buffer_key=get_key_no_doublebind('cut_to_buffer')

struct.yank_from_buffer_key=get_key_no_doublebind('yank_from_buffer')

struct.clear_word_key=get_key_no_doublebind('clear_word')

struct.backspace_key=get_key_no_doublebind('backspace')

struct.clear_line_key=get_key_no_doublebind('clear_line')

struct.clear_screen_key=get_key_no_doublebind('clear_screen')

struct.delete_key=get_key_no_doublebind('delete')

struct.left_key=get_key_no_doublebind('left')

struct.right_key=get_key_no_doublebind('right')

struct.end_of_line_key=get_key_no_doublebind('end_of_line')

struct.beginning_of_line_key=get_key_no_doublebind('beginning_of_line')

struct.transpose_chars_key=get_key_no_doublebind('transpose_chars')

struct.exit_key=get_key_no_doublebind('exit')

struct.last_output_key=get_key_no_doublebind('last_output')

struct.edit_config_key=get_key_no_doublebind('edit_config')

struct.edit_current_block_key=get_key_no_doublebind('edit_current_block')

struct.external_editor_key=get_key_no_doublebind('external_editor')

struct.help_key=get_key_no_doublebind('help')

struct.pastebin_confirm=config.getboolean('general','pastebin_confirm')

struct.pastebin_url=config.get('general','pastebin_url')

struct.pastebin_show_url=config.get('general','pastebin_show_url')

struct.pastebin_removal_url=config.get('general','pastebin_removal_url')

struct.pastebin_expiry=config.get('general','pastebin_expiry')

struct.pastebin_helper=config.get('general','pastebin_helper')

struct.cli_suggestion_width=config.getfloat('cli',

'suggestion_width')

struct.cli_trim_prompts=config.getboolean('cli',

'trim_prompts')

struct.complete_magic_methods=config.getboolean('general',

'complete_magic_methods')

struct.autocomplete_mode=config.get('general','autocomplete_mode')

struct.save_append_py=config.getboolean('general','save_append_py')

struct.curtsies_list_above=config.getboolean('curtsies','list_above')

struct.curtsies_right_arrow_completion= \

config.getboolean('curtsies','right_arrow_completion')

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',

'right_arrow_suggestion':'K',

}

ifcolor_scheme_name=='default':

struct.color_scheme=default_colors

else:

struct.color_scheme=dict()

path=os.path.expanduser(os.path.join(get_config_home(),

theme_filename))

try:

load_theme(struct,path,struct.color_scheme,default_colors)

exceptEnvironmentError:

sys.stderr.write("Could not load theme '%s'.\n"%

(color_scheme_name, ))

sys.exit(1)

struct.hist_file=os.path.expanduser(struct.hist_file)

ifstruct.autocomplete_modenotinALL_MODES:

struct.autocomplete_mode=default_completion

ifconfig.getboolean('general','unicode_box')andsupports_box_chars():

struct.left_border='│'

struct.right_border='│'

struct.top_border='─'

struct.bottom_border='─'

struct.left_bottom_corner='└'

struct.right_bottom_corner='┘'

struct.left_top_corner='┌'

struct.right_top_corner='┐'

else:

struct.left_border='|'

struct.right_border='|'

struct.top_border='-'

struct.bottom_border='-'

struct.left_bottom_corner='+'

struct.right_bottom_corner='+'

struct.left_top_corner='+'

struct.right_top_corner='+'

defload_theme(struct,path,colors,default_colors):

theme=ConfigParser()

withopen(path,'r')asf:

theme.readfp(f)

fork,vinchain(theme.items('syntax'),theme.items('interface')):

iftheme.has_option('syntax',k):

colors[k]=theme.get('syntax',k)

else:

colors[k]=theme.get('interface',k)

fork,viniteritems(default_colors):

ifknotincolors:

colors[k]=v