load config

[rainbowstream.git] / rainbowstream / config.py

diff --git a/rainbowstream/config.py b/rainbowstream/config.py
index 12dffb29e7ea81d756fcba1fee2257e909c7a0de..837aec0d4c19a59076ff3b0ff3947aeaff29c1eb 100644 (file)
--- a/rainbowstream/config.py
+++ b/rainbowstream/config.py
@@ -1,21 +1,48 @@
-# 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
-# Autocomplete history
+# 'allrt': default number of retweets
+RETWEETS_SHOW_NUM = 5
+# 'inbox','sent': default number of direct message
+MESSAGES_DISPLAY = 5
+# 'trend': max trending topics
+TREND_MAX = 10
+# '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
\ No newline at end of file
+IMAGE_MAX_HEIGHT = 40
+
+# Load colorset
+COLOR_SET = ['colorset.default']
+modules = map(__import__, COLOR_SET)
+
+# 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] = data[d]
+except:
+    pass
+
+