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
  13 def load_config(filepath):
  14     """
  15     Load config from filepath
  16     """
  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
  25 # Config dictionary
  26 c = {}
  27 # Load default
  28 default_config = os.path.dirname(__file__) + '/colorset/default.json'
  29 data = load_config(default_config)
  30 for d in data:
  31     c[d] = data[d]
  32 c['theme'] = 'default'
  33 # Load init if exist
  34 try:
  35     path = os.path.dirname(__file__) + '/colorset/init'
  36     f = open(path)
  37     lines = f.readlines()
  38     if len(lines) > 1:
  39         raise Exception('More than 1 default theme')
  40     theme_name = lines[0].strip()
  41     default_config = os.path.dirname(
  42         __file__) + '/colorset/' + theme_name + '.json'
  43     data = load_config(default_config)
  44     for d in data:
  45         c[d] = data[d]
  46     c['theme'] = theme_name
  47     f.close()
  48 except:
  49     pass
  50 # Load user's colorset
  51 rainbow_config = os.environ.get(
  52     'HOME',
  53     os.environ.get(
  54         'USERPROFILE',
  55         '')) + os.sep + '.rainbow_config.json'
  56 try:
  57     data = load_config(rainbow_config)
  58     for d in data:
  59         c[d] = data[d]
  60     c['theme'] = 'custom'
  61 except:
  62     pass