autopep8
authorOrakaro <nhatminh_179@hotmail.com>
Wed, 2 Jul 2014 16:41:05 +0000 (01:41 +0900)
committerOrakaro <nhatminh_179@hotmail.com>
Wed, 2 Jul 2014 16:41:05 +0000 (01:41 +0900)
rainbowstream/colors.py
rainbowstream/colorset/default.json
rainbowstream/colorset/monokai.json
rainbowstream/config.py
rainbowstream/draw.py
rainbowstream/rainbow.py

index bcf48ed..909a85a 100644 (file)
@@ -3,6 +3,7 @@ import itertools
 from functools import wraps
 from pyfiglet import figlet_format
 from .config import *
+from functools import reduce
 
 
 def basic_color(code):
@@ -27,41 +28,42 @@ def RGB(code):
     return inner
 
 
-default          = basic_color('39')
-black            = basic_color('30')
-red              = basic_color('31')
-green            = basic_color('32')
-yellow           = basic_color('33')
-blue             = basic_color('34')
-magenta          = basic_color('35')
-cyan             = basic_color('36')
-grey             = basic_color('90')
-light_red        = basic_color('91')
-light_green      = basic_color('92')
-light_yellow     = basic_color('93')
-light_blue       = basic_color('94')
-light_magenta    = basic_color('95')
-light_cyan       = basic_color('96')
-white            = basic_color('97')
-
-on_default       = basic_color('49')
-on_black         = basic_color('40')
-on_red           = basic_color('41')
-on_green         = basic_color('42')
-on_yellow        = basic_color('43')
-on_blue          = basic_color('44')
-on_magenta       = basic_color('45')
-on_cyan          = basic_color('46')
-on_grey          = basic_color('100')
-on_light_red     = basic_color('101')
-on_light_green   = basic_color('102')
-on_light_yellow  = basic_color('103')
-on_light_blue    = basic_color('104')
+default = basic_color('39')
+black = basic_color('30')
+red = basic_color('31')
+green = basic_color('32')
+yellow = basic_color('33')
+blue = basic_color('34')
+magenta = basic_color('35')
+cyan = basic_color('36')
+grey = basic_color('90')
+light_red = basic_color('91')
+light_green = basic_color('92')
+light_yellow = basic_color('93')
+light_blue = basic_color('94')
+light_magenta = basic_color('95')
+light_cyan = basic_color('96')
+white = basic_color('97')
+
+on_default = basic_color('49')
+on_black = basic_color('40')
+on_red = basic_color('41')
+on_green = basic_color('42')
+on_yellow = basic_color('43')
+on_blue = basic_color('44')
+on_magenta = basic_color('45')
+on_cyan = basic_color('46')
+on_grey = basic_color('100')
+on_light_red = basic_color('101')
+on_light_green = basic_color('102')
+on_light_yellow = basic_color('103')
+on_light_blue = basic_color('104')
 on_light_magenta = basic_color('105')
-on_light_cyan    = basic_color('106')
-on_white         = basic_color('107')
+on_light_cyan = basic_color('106')
+on_white = basic_color('107')
 
-colors_shuffle = [locals()[i.encode('utf8')] if not i.startswith('RGB_') else RGB(int(i[4:])) for i in c['CYCLE_COLOR']]
+colors_shuffle = [locals()[i.encode('utf8')] if not i.startswith(
+    'RGB_') else RGB(int(i[4:])) for i in c['CYCLE_COLOR']]
 
 background_shuffle = [
     on_grey,
index fea5e81..5938650 100644 (file)
@@ -45,6 +45,7 @@
     http://www.calmar.ws/vim/256-xterm-24bit-rgb-color-chart.html
      */
 
+    "DECORATED_NAME" : "grey",
     "CYCLE_COLOR" :["light_red","light_green","light_yellow","light_blue","light_magenta","light_cyan"],
     "TWEET" : {
         "nick"      : "grey",
index 4130427..09c6341 100644 (file)
@@ -22,6 +22,7 @@
     http://www.calmar.ws/vim/256-xterm-24bit-rgb-color-chart.html
      */
 
+    "DECORATED_NAME" : "RGB_166",
     "CYCLE_COLOR" :["RGB_198","RGB_57","RGB_166","RGB_50","RGB_179","RGB_74","RGB_112"],
     "TWEET" : {
         "nick"      : "RGB_198",
index 3f61553..418afbb 100644 (file)
@@ -9,6 +9,7 @@ comment_re = re.compile(
     re.DOTALL | re.MULTILINE
 )
 
+
 def load_config(filepath):
     """
     Load config from filepath
@@ -31,7 +32,11 @@ data = load_config(default_config)
 for d in data:
     c[d] = data[d]
 # Load user's colorset
-rainbow_config = os.environ.get('HOME', os.environ.get('USERPROFILE','')) + os.sep + '.rainbow_config.json'
+rainbow_config = os.environ.get(
+    'HOME',
+    os.environ.get(
+        'USERPROFILE',
+        '')) + os.sep + '.rainbow_config.json'
 data = load_config(rainbow_config)
 for d in data:
     c[d] = data[d]
index b35649c..9a8fbdd 100644 (file)
@@ -20,7 +20,6 @@ def color_func(func_name):
     """
     Call color function base on name
     """
-    print globals()['TWEET']
     pure = func_name.encode('utf8')
     if pure.startswith('RGB_') and pure[4:].isdigit():
         return RGB(int(pure[4:]))
@@ -79,8 +78,10 @@ def draw(t, iot=False, keyword=None, fil=[], ig=[]):
     rid = res[0].rainbow_id
 
     # Format info
-    user = cycle_color(name) + color_func(c['TWEET']['nick'])(' ' + screen_name + ' ')
-    meta = color_func(c['TWEET']['clock'])('[' + clock + '] ') + color_func(c['TWEET']['id'])('[id=' + str(rid) + '] ')
+    user = cycle_color(
+        name) + color_func(c['TWEET']['nick'])(' ' + screen_name + ' ')
+    meta = color_func(c['TWEET']['clock'])(
+        '[' + clock + '] ') + color_func(c['TWEET']['id'])('[id=' + str(rid) + '] ')
     if favorited:
         meta = meta + color_func(c['TWEET']['favorite'])(u'\u2605')
     tweet = text.split()
@@ -91,11 +92,18 @@ def draw(t, iot=False, keyword=None, fil=[], ig=[]):
                 lambda x: expanded_url[index] if x == url[index] else x,
                 tweet)
     # Highlight RT
-    tweet = map(lambda x: color_func(c['TWEET']['rt'])(x) if x == 'RT' else x, tweet)
+    tweet = map(
+        lambda x: color_func(
+            c['TWEET']['rt'])(x) if x == 'RT' else x,
+        tweet)
     # Highlight screen_name
     tweet = map(lambda x: cycle_color(x) if x[0] == '@' else x, tweet)
     # Highlight link
-    tweet = map(lambda x: color_func(c['TWEET']['link'])(x) if x[0:4] == 'http' else x, tweet)
+    tweet = map(
+        lambda x: color_func(
+            c['TWEET']['link'])(x) if x[
+            0:4] == 'http' else x,
+        tweet)
     # Highlight search keyword
     if keyword:
         tweet = map(
@@ -152,10 +160,17 @@ def print_message(m):
     rid = res[0].rainbow_id
 
     # Draw
-    sender = cycle_color(sender_name) + color_func(c['MESSAGE']['sender'])(' ' + sender_screen_name + ' ')
-    recipient = cycle_color(recipient_name) + color_func(c['MESSAGE']['recipient'])(' ' + recipient_screen_name + ' ')
+    sender = cycle_color(
+        sender_name) + color_func(c['MESSAGE']['sender'])(' ' + sender_screen_name + ' ')
+    recipient = cycle_color(recipient_name) + color_func(
+        c['MESSAGE']['recipient'])(
+        ' ' + recipient_screen_name + ' ')
     user = sender + color_func(c['MESSAGE']['to'])(' >>> ') + recipient
-    meta = color_func(c['MESSAGE']['clock'])('[' + clock + ']') + color_func(c['MESSAGE']['id'])(' [message_id=' + str(rid) + '] ')
+    meta = color_func(
+        c['MESSAGE']['clock'])(
+        '[' + clock + ']') + color_func(
+            c['MESSAGE']['id'])(
+                ' [message_id=' + str(rid) + '] ')
     text = ''.join(map(lambda x: x + '  ' if x == '\n' else x, text))
 
     line1 = u"{u:>{uw}}:".format(
@@ -192,12 +207,23 @@ def show_profile(u, iot=False):
     followers_count = u['followers_count']
 
     # Create content
-    statuses_count = color_func(c['PROFILE']['statuses_count'])(str(statuses_count) + ' tweets')
-    friends_count = color_func(c['PROFILE']['friends_count'])(str(friends_count) + ' following')
-    followers_count = color_func(c['PROFILE']['followers_count'])(str(followers_count) + ' followers')
+    statuses_count = color_func(
+        c['PROFILE']['statuses_count'])(
+        str(statuses_count) +
+        ' tweets')
+    friends_count = color_func(
+        c['PROFILE']['friends_count'])(
+        str(friends_count) +
+        ' following')
+    followers_count = color_func(
+        c['PROFILE']['followers_count'])(
+        str(followers_count) +
+        ' followers')
     count = statuses_count + '  ' + friends_count + '  ' + followers_count
-    user = cycle_color(name) + color_func(c['PROFILE']['nick'])(' @' + screen_name + ' : ') + count
-    profile_image_raw_url = 'Profile photo: ' + color_func(c['PROFILE']['profile_image_url'])(profile_image_url)
+    user = cycle_color(
+        name) + color_func(c['PROFILE']['nick'])(' @' + screen_name + ' : ') + count
+    profile_image_raw_url = 'Profile photo: ' + \
+        color_func(c['PROFILE']['profile_image_url'])(profile_image_url)
     description = ''.join(
         map(lambda x: x + ' ' * 4 if x == '\n' else x, description))
     description = color_func(c['PROFILE']['description'])(description)
@@ -257,4 +283,3 @@ def print_trends(trends):
         line = cycle_color(name) + ': ' + color_func(TREND['url'])(url)
         printNicely(line)
     printNicely('')
-
index f31fa23..ce5d998 100644 (file)
@@ -64,6 +64,7 @@ cmdset = [
     'q'
 ]
 
+
 def parse_arguments():
     """
     Parse the arguments
@@ -133,10 +134,11 @@ def get_decorated_name():
     t = Twitter(auth=authen())
     name = '@' + t.account.verify_credentials()['screen_name']
     g['original_name'] = name[1:]
-    g['decorated_name'] = grey('[') + grey(name) + grey(']: ')
+    g['decorated_name'] = color_func(c['DECORATED_NAME'])('[' + name + ']: ')
+    g['ascii_art'] = True
 
     files = os.listdir('rainbowstream/colorset')
-    themes = [f.split('.')[0] for f in files if f.split('.')[-1]=='json']
+    themes = [f.split('.')[0] for f in files if f.split('.')[-1] == 'json']
     g['themes'] = themes
 
 
@@ -193,12 +195,12 @@ def switch():
             p.start()
             g['stream_pid'] = p.pid
         printNicely('')
-        printNicely(green('Stream switched.'))
         if args.filter:
             printNicely(cyan('Only: ' + str(args.filter)))
         if args.ignore:
             printNicely(red('Ignore: ' + str(args.ignore)))
         printNicely('')
+        g['ascii_art'] = True
     except:
         printNicely(red('Sorry I can\'t understand.'))
 
@@ -767,17 +769,25 @@ def theme():
     if not g['stuff']:
         # List themes
         for theme in g['themes']:
-            line = ' '*2 + magenta('* ' + theme)
+            line = ' ' * 2 + light_yellow('* ' + theme)
             printNicely(line)
     else:
         # Change theme
-        try :
+        try:
+            # Load new config
             new_config = 'rainbowstream/colorset/' + g['stuff'] + '.json'
             new_config = load_config(new_config)
             for nc in new_config:
                 c[nc] = new_config[nc]
+            # Reset stream
+            g['stuff'] = 'mine'
+            g['ascii_art'] = False
+            switch()
+            g['decorated_name'] = color_func(
+                c['DECORATED_NAME'])(
+                '[@' + g['original_name'] + ']: ')
             printNicely(green('Theme changed.'))
-        except :
+        except:
             printNicely(red('Sorry, config file is broken!'))
 
 
@@ -792,124 +802,126 @@ def help():
     usage = '\n'
     usage += s + 'Hi boss! I\'m ready to serve you right now!\n'
     usage += s + '-' * (int(w) - 4) + '\n'
-    usage += s + 'You are ' + yellow('already') + ' on your personal stream.\n'
+    usage += s + 'You are ' + \
+        light_yellow('already') + ' on your personal stream.\n'
     usage += s + 'Any update from Twitter will show up ' + \
-        yellow('immediately') + '.\n'
+        light_yellow('immediately') + '.\n'
     usage += s + 'In addtion, following commands are available right now:\n'
 
     # Discover the world
     usage += '\n'
     usage += s + grey(u'\u266A' + ' Discover the world \n')
-    usage += s * 2 + green('trend') + ' will show global trending topics. ' + \
-        'You can try ' + green('trend US') + ' or ' + \
-        green('trend JP Tokyo') + '.\n'
-    usage += s * 2 + green('home') + ' will show your timeline. ' + \
-        green('home 7') + ' will show 7 tweets.\n'
-    usage += s * 2 + green('mentions') + ' will show mentions timeline. ' + \
-        green('mentions 7') + ' will show 7 mention tweets.\n'
-    usage += s * 2 + green('whois @mdo') + ' will show profile  of ' + \
+    usage += s * 2 + light_green('trend') + ' will show global trending topics. ' + \
+        'You can try ' + light_green('trend US') + ' or ' + \
+        light_green('trend JP Tokyo') + '.\n'
+    usage += s * 2 + light_green('home') + ' will show your timeline. ' + \
+        light_green('home 7') + ' will show 7 tweets.\n'
+    usage += s * 2 + light_green('mentions') + ' will show mentions timeline. ' + \
+        light_green('mentions 7') + ' will show 7 mention tweets.\n'
+    usage += s * 2 + light_green('whois @mdo') + ' will show profile  of ' + \
         magenta('@mdo') + '.\n'
-    usage += s * 2 + green('view @mdo') + \
+    usage += s * 2 + light_green('view @mdo') + \
         ' will show ' + magenta('@mdo') + '\'s home.\n'
-    usage += s * 2 + green('s #AKB48') + ' will search for "' + \
-        yellow('AKB48') + '" and return 5 newest tweet.\n'
+    usage += s * 2 + light_green('s #AKB48') + ' will search for "' + \
+        light_yellow('AKB48') + '" and return 5 newest tweet.\n'
 
     # Tweet
     usage += '\n'
     usage += s + grey(u'\u266A' + ' Tweets \n')
-    usage += s * 2 + green('t oops ') + \
-        'will tweet "' + yellow('oops') + '" immediately.\n'
+    usage += s * 2 + light_green('t oops ') + \
+        'will tweet "' + light_yellow('oops') + '" immediately.\n'
     usage += s * 2 + \
-        green('rt 12 ') + ' will retweet to tweet with ' + \
-        yellow('[id=12]') + '.\n'
+        light_green('rt 12 ') + ' will retweet to tweet with ' + \
+        light_yellow('[id=12]') + '.\n'
     usage += s * 2 + \
-        green('allrt 12 20 ') + ' will list 20 newest retweet of the tweet with ' + \
-        yellow('[id=12]') + '.\n'
-    usage += s * 2 + green('rep 12 oops') + ' will reply "' + \
-        yellow('oops') + '" to tweet with ' + yellow('[id=12]') + '.\n'
+        light_green('allrt 12 20 ') + ' will list 20 newest retweet of the tweet with ' + \
+        light_yellow('[id=12]') + '.\n'
+    usage += s * 2 + light_green('rep 12 oops') + ' will reply "' + \
+        light_yellow('oops') + '" to tweet with ' + \
+        light_yellow('[id=12]') + '.\n'
     usage += s * 2 + \
-        green('fav 12 ') + ' will favorite the tweet with ' + \
-        yellow('[id=12]') + '.\n'
+        light_green('fav 12 ') + ' will favorite the tweet with ' + \
+        light_yellow('[id=12]') + '.\n'
     usage += s * 2 + \
-        green('ufav 12 ') + ' will unfavorite tweet with ' + \
-        yellow('[id=12]') + '.\n'
+        light_green('ufav 12 ') + ' will unfavorite tweet with ' + \
+        light_yellow('[id=12]') + '.\n'
     usage += s * 2 + \
-        green('del 12 ') + ' will delete tweet with ' + \
-        yellow('[id=12]') + '.\n'
-    usage += s * 2 + green('show image 12') + ' will show image in tweet with ' + \
-        yellow('[id=12]') + ' in your OS\'s image viewer.\n'
+        light_green('del 12 ') + ' will delete tweet with ' + \
+        light_yellow('[id=12]') + '.\n'
+    usage += s * 2 + light_green('show image 12') + ' will show image in tweet with ' + \
+        light_yellow('[id=12]') + ' in your OS\'s image viewer.\n'
 
     # Direct message
     usage += '\n'
     usage += s + grey(u'\u266A' + ' Direct messages \n')
-    usage += s * 2 + green('inbox') + ' will show inbox messages. ' + \
-        green('inbox 7') + ' will show newest 7 messages.\n'
-    usage += s * 2 + green('sent') + ' will show sent messages. ' + \
-        green('sent 7') + ' will show newest 7 messages.\n'
-    usage += s * 2 + green('mes @dtvd88 hi') + ' will send a "hi" messege to ' + \
+    usage += s * 2 + light_green('inbox') + ' will show inbox messages. ' + \
+        light_green('inbox 7') + ' will show newest 7 messages.\n'
+    usage += s * 2 + light_green('sent') + ' will show sent messages. ' + \
+        light_green('sent 7') + ' will show newest 7 messages.\n'
+    usage += s * 2 + light_green('mes @dtvd88 hi') + ' will send a "hi" messege to ' + \
         magenta('@dtvd88') + '.\n'
-    usage += s * 2 + green('trash 5') + ' will remove message with ' + \
-        yellow('[message_id=5]') + '.\n'
+    usage += s * 2 + light_green('trash 5') + ' will remove message with ' + \
+        light_yellow('[message_id=5]') + '.\n'
 
     # Follower and following
     usage += '\n'
     usage += s + grey(u'\u266A' + ' Fiends and followers \n')
     usage += s * 2 + \
-        green('ls fl') + \
+        light_green('ls fl') + \
         ' will list all followers (people who are following you).\n'
     usage += s * 2 + \
-        green('ls fr') + \
+        light_green('ls fr') + \
         ' will list all friends (people who you are following).\n'
-    usage += s * 2 + green('fl @dtvd88') + ' will follow ' + \
+    usage += s * 2 + light_green('fl @dtvd88') + ' will follow ' + \
         magenta('@dtvd88') + '.\n'
-    usage += s * 2 + green('ufl @dtvd88') + ' will unfollow ' + \
+    usage += s * 2 + light_green('ufl @dtvd88') + ' will unfollow ' + \
         magenta('@dtvd88') + '.\n'
-    usage += s * 2 + green('mute @dtvd88') + ' will mute ' + \
+    usage += s * 2 + light_green('mute @dtvd88') + ' will mute ' + \
         magenta('@dtvd88') + '.\n'
-    usage += s * 2 + green('unmute @dtvd88') + ' will unmute ' + \
+    usage += s * 2 + light_green('unmute @dtvd88') + ' will unmute ' + \
         magenta('@dtvd88') + '.\n'
-    usage += s * 2 + green('muting') + ' will list muting users.\n'
-    usage += s * 2 + green('block @dtvd88') + ' will block ' + \
+    usage += s * 2 + light_green('muting') + ' will list muting users.\n'
+    usage += s * 2 + light_green('block @dtvd88') + ' will block ' + \
         magenta('@dtvd88') + '.\n'
-    usage += s * 2 + green('unblock @dtvd88') + ' will unblock ' + \
+    usage += s * 2 + light_green('unblock @dtvd88') + ' will unblock ' + \
         magenta('@dtvd88') + '.\n'
-    usage += s * 2 + green('report @dtvd88') + ' will report ' + \
+    usage += s * 2 + light_green('report @dtvd88') + ' will report ' + \
         magenta('@dtvd88') + ' as a spam account.\n'
 
     # Switch
     usage += '\n'
     usage += s + grey(u'\u266A' + ' Switching streams \n')
-    usage += s * 2 + green('switch public #AKB') + \
+    usage += s * 2 + light_green('switch public #AKB') + \
         ' will switch to public stream and follow "' + \
-        yellow('AKB') + '" keyword.\n'
-    usage += s * 2 + green('switch mine') + \
+        light_yellow('AKB') + '" keyword.\n'
+    usage += s * 2 + light_green('switch mine') + \
         ' will switch to your personal stream.\n'
-    usage += s * 2 + green('switch mine -f ') + \
+    usage += s * 2 + light_green('switch mine -f ') + \
         ' will prompt to enter the filter.\n'
-    usage += s * 3 + yellow('Only nicks') + \
+    usage += s * 3 + light_yellow('Only nicks') + \
         ' filter will decide nicks will be INCLUDE ONLY.\n'
-    usage += s * 3 + yellow('Ignore nicks') + \
+    usage += s * 3 + light_yellow('Ignore nicks') + \
         ' filter will decide nicks will be EXCLUDE.\n'
-    usage += s * 2 + green('switch mine -d') + \
+    usage += s * 2 + light_green('switch mine -d') + \
         ' will use the config\'s ONLY_LIST and IGNORE_LIST.\n'
 
     # Smart shell
     usage += '\n'
     usage += s + grey(u'\u266A' + ' Smart shell\n')
-    usage += s * 2 + green('111111 * 9 / 7') + ' or any math expression ' + \
+    usage += s * 2 + light_green('111111 * 9 / 7') + ' or any math expression ' + \
         'will be evaluate by Python interpreter.\n'
-    usage += s * 2 + 'Even ' + green('cal') + ' will show the calendar' + \
+    usage += s * 2 + 'Even ' + light_green('cal') + ' will show the calendar' + \
         ' for current month.\n'
 
     # Screening
     usage += '\n'
     usage += s + grey(u'\u266A' + ' Screening \n')
-    usage += s * 2 + green('theme') + ' will list available theme.' + \
-        green('theme monokai') + ' will apply '+ yellow('monokai') + \
+    usage += s * 2 + light_green('theme') + ' will list available theme.' + \
+        light_green('theme monokai') + ' will apply ' + light_yellow('monokai') + \
         ' theme immediately.\n'
-    usage += s * 2 + green('h') + ' will show this help again.\n'
-    usage += s * 2 + green('c') + ' will clear the screen.\n'
-    usage += s * 2 + green('q') + ' will quit.\n'
+    usage += s * 2 + light_green('h') + ' will show this help again.\n'
+    usage += s * 2 + light_green('c') + ' will clear the screen.\n'
+    usage += s * 2 + light_green('q') + ' will quit.\n'
 
     # End
     usage += '\n'
@@ -1051,7 +1063,8 @@ def listen():
         try:
             g['stuff'] = ' '.join(line.split()[1:])
             process(cmd)()
-        except Exception:
+        except Exception as e:
+            print e
             printNicely(red('OMG something is wrong with Twitter right now.'))
         # Not redisplay prefix
         if cmd in ['switch', 't', 'rt', 'rep']:
@@ -1071,7 +1084,8 @@ def stream(domain, args, name='Rainbow Stream'):
         c['PUBLIC_DOMAIN']: args.track_keywords,
         c['SITE_DOMAIN']: 'Site Stream',
     }
-    ascii_art(art_dict[domain])
+    if g['ascii_art']:
+        ascii_art(art_dict[domain])
 
     # These arguments are optional:
     stream_args = dict(
@@ -1120,7 +1134,8 @@ def stream(domain, args, name='Rainbow Stream'):
                     ig=args.ignore,
                 )
     except:
-        printNicely(magenta("I'm afraid we have problem with twitter'S maximum connection."))
+        printNicely(
+            magenta("I'm afraid we have problem with twitter'S maximum connection."))
         printNicely(magenta("Let's try again later."))
 
 
@@ -1131,7 +1146,12 @@ def fly():
     # Spawn stream process
     args = parse_arguments()
     get_decorated_name()
-    p = Process(target=stream, args=(c['USER_DOMAIN'], args, g['original_name']))
+    p = Process(
+        target=stream,
+        args=(
+            c['USER_DOMAIN'],
+            args,
+            g['original_name']))
     p.start()
 
     # Start listen process