Commit | Line | Data |
---|---|---|
44d70096 | 1 | import json |
72c02928 | 2 | import re |
44d70096 VNM |
3 | import os |
4 | import os.path | |
6fa09c14 | 5 | |
0a0ee6db VNM |
6 | # Regular expression for comments |
7 | comment_re = re.compile( | |
8 | '(^)?[^\S\n]*/(?:\*(.*?)\*/[^\S\n]*|/[^\n]*)($)?', | |
9 | re.DOTALL | re.MULTILINE | |
10 | ) | |
11 | ||
12 | def load_config(filepath): | |
92be926e VNM |
13 | """ |
14 | Load config from filepath | |
15 | """ | |
0a0ee6db VNM |
16 | try: |
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():] | |
22 | match = comment_re.search(content) | |
632c6fa5 | 23 | return json.loads(content) |
0a0ee6db VNM |
24 | except: |
25 | pass | |
6fa09c14 | 26 | |
632c6fa5 O |
27 | # Load default colorset |
28 | c = {} | |
0a0ee6db | 29 | default_config = 'rainbowstream/colorset/default.json' |
632c6fa5 O |
30 | data = load_config(default_config) |
31 | for d in data: | |
32 | c[d] = data[d] | |
33 | # Load user's colorset | |
2e73187a | 34 | rainbow_config = os.environ.get('HOME', os.environ.get('USERPROFILE','')) + os.sep + '.rainbow_config.json' |
632c6fa5 O |
35 | data = load_config(rainbow_config) |
36 | for d in data: | |
37 | c[d] = data[d] |