fix json syntax

[rainbowstream.git] / rainbowstream / config.py

diff --git a/rainbowstream/config.py b/rainbowstream/config.py
index 79458991e41467bfccbb852721c820f96790fd21..409f1605c192aad5413ca3d94b79cb079e034121 100644 (file)
--- a/rainbowstream/config.py
+++ b/rainbowstream/config.py
@@ -1,25 +1,52 @@
-# Max Search record
+from .colors import *
+import json
+import os
+import os.path
+
+# 'search': max search record
 SEARCH_MAX_RECORD = 5
-# Default home tweet
+# 'home': default number of home's tweets
 HOME_TWEET_NUM = 5
-# Default direct message's number
+# 'allrt': default number of retweets
+RETWEETS_SHOW_NUM = 5
+# 'inbox','sent': default number of direct message
 MESSAGES_DISPLAY = 5
-# Max trending topics display
+# 'trend': max trending topics
 TREND_MAX = 10
-# Autocomplete history
+# 'switch': Filter and Ignore list ex: ['@fat','@mdo']
+ONLY_LIST = []
+IGNORE_LIST = []
+
+# Autocomplete history file name
 HISTORY_FILENAME = 'completer.hist'
 
-# Stream Domain
 USER_DOMAIN = 'userstream.twitter.com'
 PUBLIC_DOMAIN = 'stream.twitter.com'
 SITE_DOMAIN = 'sitestream.twitter.com'
-# Actually called
 DOMAIN = USER_DOMAIN
 
-# Filter and Ignore list ex: ['@fat','@mdo']
-ONLY_LIST = []
-IGNORE_LIST = []
-
-# Image shift and size
+# Image config
 IMAGE_SHIFT = 10
 IMAGE_MAX_HEIGHT = 40
+
+# Load colorset
+default_colorset = 'rainbowstream/colorset/default.json'
+try:
+    if os.path.exists(default_colorset):
+        data = json.load(open(default_colorset))
+        for d in data:
+            locals()[d] = local()[data[d]]
+except:
+    pass
+
+# Load json config
+rainbow_config = os.environ.get('HOME', os.environ.get('USERPROFILE','')) + os.sep + '.rainbow_config.json'
+try:
+    if os.path.exists(rainbow_config):
+        data = json.load(open(rainbow_config))
+        for d in data:
+            locals()[d] = local()[data[d]]
+except:
+    pass
+
+