import json
+import re
import os
import os.path
+from collections import OrderedDict
# Regular expression for comments
comment_re = re.compile(
)
def load_config(filepath):
- try:
- with open(filepath) as f:
- content = ''.join(f.readlines())
+ """
+ Load config from filepath
+ """
+ with open(filepath) as f:
+ content = ''.join(f.readlines())
+ match = comment_re.search(content)
+ while match:
+ content = content[:match.start()] + content[match.end():]
match = comment_re.search(content)
- while match:
- content = content[:match.start()] + content[match.end():]
- match = comment_re.search(content)
- data = json.loads(content)
- for d in data:
- globals()[d] = data[d]
- except:
- pass
+ return json.loads(content, object_pairs_hook=OrderedDict)
-# Load colorset
-default_config = 'rainbowstream/colorset/default.json'
-rainbow_config = os.environ.get('HOME', os.environ.get('USERPROFILE','')) + os.sep + '.rainbow_config.json'
-load_config(default_config)
-load_config(rainbow_config)
+# Config dictionary
+c = {}
+
+# Load the initial config
+config = os.path.dirname(
+ __file__) + '/colorset/config'
+try:
+ data = load_config(config)
+ for d in data:
+ c[d] = data[d]
+except:
+ pass
+
+# Load user's config
+rainbow_config = os.environ.get(
+ 'HOME',
+ os.environ.get(
+ 'USERPROFILE',
+ '')) + os.sep + '.rainbow_config.json'
+try:
+ data = load_config(rainbow_config)
+ for d in data:
+ c[d] = data[d]
+except:
+ print('It seems that ~/.rainbow_config.json has wrong format :(')
+
+# Load default theme
+theme_file = os.path.dirname(
+ __file__) + '/colorset/' + c['THEME'] + '.json'
+try:
+ data = load_config(theme_file)
+ for d in data:
+ c[d] = data[d]
+except:
+ pass