import re
import os
import os.path
-from twitter.util import printNicely
+
+from .db import *
+
+db = RainbowDB()
# Regular expression for comments
comment_re = re.compile(
re.DOTALL | re.MULTILINE
)
+
def load_config(filepath):
- try:
- with open(filepath) as f:
- content = ''.join(f.readlines())
+ """
+ Load config from filepath
+ """
+ with open(filepath) as f:
+ content = ''.join(f.readlines())
+ match = comment_re.search(content)
+ while match:
+ content = content[:match.start()] + content[match.end():]
match = comment_re.search(content)
- while match:
- content = content[:match.start()] + content[match.end():]
- match = comment_re.search(content)
- data = json.loads(content)
- for d in data:
- globals()[d] = data[d]
- except:
- pass
-
-# Image config
-IMAGE_SHIFT = 10
-IMAGE_MAX_HEIGHT = 40
+ return json.loads(content)
-# Load colorset
+# Load default colorset
+c = {}
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)
+data = load_config(default_config)
+for d in data:
+ c[d] = data[d]
+c['theme'] = 'default'
+# 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