Color the text block splited by space only before newline

orakaro · May 18, 2017 · a8884f4 · a8884f4
1 parent 24a80d7
commit a8884f4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/rainbowstream/draw.py b/rainbowstream/draw.py
@@ -273,8 +273,9 @@ def draw(t, keyword=None, humanize=True, noti=False, fil=[], ig=[]):
     if favorited:
         fav = color_func(c['TWEET']['favorited'])(u'\u2605')
 
-    tweet = text.split(' ')
-    tweet = [x for x in tweet if x != '']
+    tweet = list()
+    for s in [x.replace('\n', '\n\0').split('\0') for x in text.split(' ')]:
+        tweet.extend(filter(lambda x: x != '', s))
     # Replace url
     if expanded_url:
         for index in xrange(len(expanded_url)):
@@ -315,7 +316,7 @@ def draw(t, keyword=None, humanize=True, noti=False, fil=[], ig=[]):
                  ]
     # Highlight keyword
     tweet = ' '.join(tweet)
-    tweet = '\n  '.join(tweet.split('\n'))
+    tweet = tweet.replace('\n', '\n ')
     if keyword:
         roj = re.search(keyword, tweet, re.IGNORECASE)
         if roj: