-# 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 default
+default_config = os.path.dirname(__file__) + '/colorset/default.json'
+data = load_config(default_config)
+for d in data:
+ c[d] = data[d]
+c['theme'] = 'default'
+# Load init if exist
+try:
+ path = os.path.dirname(__file__) + '/colorset/init'
+ f = open(path)
+ lines = f.readlines()
+ if len(lines) > 1:
+ raise Exception('More than 1 default theme')
+ theme_name = lines[0].strip()
+ default_config = os.path.dirname(
+ __file__) + '/colorset/' + theme_name + '.json'
+ data = load_config(default_config)
+ for d in data:
+ c[d] = data[d]
+ c['theme'] = theme_name
+ f.close()
+except:
+ pass
+# Load user's colorset
+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]
+ c['theme'] = 'custom'
+except:
+ pass