term color define only by number
[rainbowstream.git] / rainbowstream / config.py
CommitLineData
44d70096 1import json
72c02928 2import re
44d70096
VNM
3import os
4import os.path
1f2f6159 5from collections import OrderedDict
6fa09c14 6
0a0ee6db
VNM
7# Regular expression for comments
8comment_re = re.compile(
9 '(^)?[^\S\n]*/(?:\*(.*?)\*/[^\S\n]*|/[^\n]*)($)?',
10 re.DOTALL | re.MULTILINE
11)
12
13def load_config(filepath):
92be926e
VNM
14 """
15 Load config from filepath
16 """
ddb1e615
VNM
17 with open(filepath) as f:
18 content = ''.join(f.readlines())
19 match = comment_re.search(content)
20 while match:
21 content = content[:match.start()] + content[match.end():]
0a0ee6db 22 match = comment_re.search(content)
1f2f6159 23 return json.loads(content, object_pairs_hook=OrderedDict)
6fa09c14 24
1c8a5082 25# Config dictionary
632c6fa5 26c = {}
1f2f6159
O
27
28# Load user's config
c075e6dc
O
29rainbow_config = os.environ.get(
30 'HOME',
31 os.environ.get(
32 'USERPROFILE',
33 '')) + os.sep + '.rainbow_config.json'
ddb1e615
VNM
34try:
35 data = load_config(rainbow_config)
a5301bc0
VNM
36 for d in data:
37 c[d] = data[d]
1f2f6159
O
38except:
39 pass
40
41# Load default theme
42theme_file = os.path.dirname(
43 __file__) + '/colorset/' + c['THEME'] + '.json'
44try:
45 data = load_config(theme_file)
46 for d in data:
47 c[d] = data[d]
ddb1e615
VNM
48except:
49 pass