qutebrowser/qutebrowser/utils/config.py
2014-01-29 21:06:56 +01:00

171 lines
5.0 KiB
Python

"""Configuration storage and config-related utilities.
config -- The main Config object.
colordict -- All configured colors.
default_config -- The default config as dict.
MONOSPACE -- A list of suitable monospace fonts.
"""
import os.path
import os
import logging
from configparser import ConfigParser
config = None
colordict = {}
default_config = {
'general': {
'show_completion': 'true',
'space_scroll': '200',
},
'keybind': {
'o': 'open',
'O': 'tabopen',
'd': 'tabclose',
'J': 'tabnext',
'K': 'tabprev',
'r': 'reload',
'H': 'back',
'L': 'forward',
'h': 'scroll -50 0',
'j': 'scroll 0 50',
'k': 'scroll 0 -50',
'l': 'scroll 50 0',
'u': 'undo',
'gg': 'scroll_perc_y 0',
'G': 'scroll_perc_y',
'n': 'nextsearch',
},
'colors': {
'completion.fg': '#333333',
'completion.item.bg': 'white',
'completion.category.bg': ('qlineargradient(x1:0, y1:0, x2:0, y2:1, '
'stop:0 #e4e4e4, stop:1 #dbdbdb)'),
'completion.category.border.top': '#808080',
'completion.category.border.bottom': '#bbbbbb',
'completion.item.selected.fg': '#333333',
'completion.item.selected.bg': '#ffec8b',
'completion.item.selected.border.top': '#f2f2c0',
'completion.item.selected.border.bottom': '#e6e680',
'completion.match.fg': 'red',
'statusbar.progress.bg': 'white',
'statusbar.progress.bg.error': 'red',
'statusbar.bg': 'black',
'statusbar.fg': 'white',
'statusbar.bg.error': 'red',
'tab.bg': 'grey',
'tab.bg.selected': 'black',
'tab.fg': 'white',
'tab.seperator': 'white',
},
}
_MONOSPACE = ['Monospace', 'DejaVu Sans Mono', 'Consolas', 'Monaco',
'Bitstream Vera Sans Mono', 'Andale Mono', 'Liberation Mono',
'Courier New', 'Courier', 'monospace', 'Fixed', 'Terminal']
MONOSPACE = ', '.join(_MONOSPACE)
def init(confdir):
"""Initialize the global objects based on the config in configdir."""
global config, colordict
config = Config(confdir)
try:
colordict = ColorDict(config['colors'])
except KeyError:
colordict = ColorDict()
def get_stylesheet(template):
"""Return a formatted stylesheet based on a template."""
return template.strip().format(color=colordict, monospace=MONOSPACE)
class ColorDict(dict):
"""A dict aimed at Qt stylesheet colors."""
def __getitem__(self, key):
"""Override dict __getitem__.
If a value wasn't found, return an empty string.
(Color not defined, so no output in the stylesheet)
If the key has a .fg. element in it, return color: X;.
If the key has a .bg. element in it, return background-color: X;.
In all other cases, return the plain value.
"""
try:
val = super().__getitem__(key)
except KeyError:
return ''
if 'fg' in key.split('.'):
return 'color: {};'.format(val)
elif 'bg' in key.split('.'):
return 'background-color: {};'.format(val)
else:
return val
def getraw(self, key):
"""Get a value without the transformations done in __getitem__.
Returns a value, or None if the value wasn't found.
"""
try:
return super().__getitem__(key)
except KeyError:
return None
class Config(ConfigParser):
"""Our own ConfigParser subclass."""
configdir = None
FNAME = 'config'
def __init__(self, configdir):
"""Config constructor.
configdir -- directory to store the config in.
"""
super().__init__()
self.optionxform = lambda opt: opt # be case-insensitive
self.configdir = configdir
if self.configdir is None:
self.init_config()
return
self.configfile = os.path.join(self.configdir, self.FNAME)
if not os.path.isfile(self.configfile):
self.init_config()
logging.debug("Reading config from {}".format(self.configfile))
self.read(self.configfile)
def init_config(self):
"""Initialize Config from default_config and save it."""
logging.info("Initializing default config.")
if self.configdir is None:
self.read_dict(default_config)
return
cp = ConfigParser()
cp.optionxform = lambda opt: opt # be case-insensitive
cp.read_dict(default_config)
if not os.path.exists(self.configdir):
os.makedirs(self.configdir, 0o755)
with open(self.configfile, 'w') as f:
cp.write(f)
def save(self):
"""Save the config file."""
if self.configdir is None:
return
if not os.path.exists(self.configdir):
os.makedirs(self.configdir, 0o755)
logging.debug("Saving config to {}".format(self.configfile))
with open(self.configfile, 'w') as f:
self.write(f)