rainbowstream/config.py

   1 import json
   2 import re
   3 import os
   4 import os.path
   5
   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):
  13     """
  14     Load config from filepath
  15     """
  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)
  23         return json.loads(content)
  24     except:
  25         pass
  26
  27 # Load default colorset
  28 c = {}
  29 default_config = 'rainbowstream/colorset/default.json'
  30 data = load_config(default_config)
  31 for d in data:
  32     c[d] = data[d]
  33 # Load user's colorset
  34 rainbow_config = os.environ.get('HOME', os.environ.get('USERPROFILE','')) + os.sep + '.rainbow_config.json'
  35 data = load_config(rainbow_config)
  36 for d in data:
  37     c[d] = data[d]