colorful usage
[rainbowstream.git] / rainbowstream / rainbow.py
index 61bb97b958c90a11ccd6b8763b7cdf1e758d9d7d..83a86b2cd21c641a84e3b61c1ea2e140eeab231c 100644 (file)
@@ -1,7 +1,6 @@
 """
 Colorful user's timeline stream
 """
-from __future__ import print_function
 from multiprocessing import Process
 from dateutil import parser
 
@@ -188,8 +187,8 @@ def switch():
             if g['stuff'].split()[-1] == '-f':
                 only = raw_input('Only nicks: ')
                 ignore = raw_input('Ignore nicks: ')
-                args.filter = only.split(',')
-                args.ignore = ignore.split(',')
+                args.filter = filter(None,only.split(','))
+                args.ignore = filter(None,ignore.split(','))
             elif g['stuff'].split()[-1] == '-d':
                 args.filter = ONLY_LIST
                 args.ignore = IGNORE_LIST
@@ -348,10 +347,10 @@ def friend():
     t = Twitter(auth=authen())
     g['friends'] = t.friends.ids()['ids']
     for i in g['friends']:
-        screen_name = t.users.lookup(user_id=i)[0]['screen_name']
-        user = cycle_color('@' + screen_name)
-        print(user, end=' ')
-    printNicely('')
+        name = t.users.lookup(user_id=i)[0]['name']
+        screen_name = '@' + t.users.lookup(user_id=i)[0]['screen_name']
+        user = cycle_color(name) + grey(' ' + screen_name + ' ')
+        print user
 
 
 def follower():
@@ -361,42 +360,47 @@ def follower():
     t = Twitter(auth=authen())
     g['followers'] = t.followers.ids()['ids']
     for i in g['followers']:
-        screen_name = t.users.lookup(user_id=i)[0]['screen_name']
-        user = cycle_color('@' + screen_name)
-        print(user, end=' ')
-    printNicely('')
+        name = t.users.lookup(user_id=i)[0]['name']
+        screen_name = '@' + t.users.lookup(user_id=i)[0]['screen_name']
+        user = cycle_color(name) + grey(' ' + screen_name + ' ')
+        print user
 
 
 def help():
     """
     Help
     """
-    usage = '''
-    Hi boss! I'm ready to serve you right now!
-    -------------------------------------------------------------
-    You are already on your personal stream:
-      "switch public #AKB" will switch to public stream and follow "AKB" keyword.
-      "switch mine" will switch back to your personal stream.
-      "switch mine -f" will prompt to enter the filter.
-        "Only nicks" filter will decide nicks will be INCLUDE ONLY.
-        "Ignore nicks" filter will decide nicks will be EXCLUDE.
-      "switch mine -d" will use the config's ONLY_LIST and IGNORE_LIST(see config.py).
-    For more action:
-      "home" will show your timeline. "home 7" will show 7 tweet.
-      "view @bob" will show your friend @bob's home.
-      "t oops" will tweet "oops" immediately.
-      "rt 12345" will retweet to tweet with id "12345".
-      "rep 12345 oops" will reply "oops" to tweet with id "12345".
-      "del 12345" will delete tweet with id "12345".
-      "s #AKB48" will search for "AKB48" and return 5 newest tweet.
-      "fr" will list out your following people.
-      "fl" will list out your followers.
-      "h" will show this help again.
-      "c" will clear the terminal.
-      "q" will exit.
-    -------------------------------------------------------------
-    Have fun and hang tight!
-    '''
+    s = ' '*2
+
+    usage = '\n'
+    usage += s + 'Hi boss! I\'m ready to serve you right now!\n'
+    usage += s + '-'*70 + '\n'
+
+    usage += s + 'You are ' + yellow('already') +' on your personal stream.\n'
+    usage += s*2 + green('switch public #AKB') +  ' will switch to public stream and follow "' + yellow('AKB') + '" keyword.\n'
+    usage += s*2 + green('switch mine') + ' will switch to your personal stream.\n'
+    usage += s*2 + green('switch mine -f ') + ' will prompt to enter the filter.\n'
+    usage += s*3 + yellow('Only nicks') + ' filter will decide nicks will be INCLUDE ONLY.\n'
+    usage += s*3 + yellow('Ignore nicks') + ' filter will decide nicks will be EXCLUDE.\n'
+    usage += s*2 + green('switch mine -d') + ' will use the config\'s ONLY_LIST and IGNORE_LIST\n'
+    usage += s*3 + '(see ' + grey('rainbowstream/config.py') + ').\n'
+
+    usage += s + 'For more action: \n'
+    usage += s*2 + green('home') + ' will show your timeline. ' + green('home 7') + ' will show 7 tweet.\n'
+    usage += s*2 + green('view @mdo') + ' will show @mdo\'s home.\n'
+    usage += s*2 + green('t oops ') + 'will tweet "' + yellow('oops') + '"" immediately.\n'
+    usage += s*2 + green('rt 12 ') + ' will retweet to tweet with '+ yellow('[id=12]') + '.\n'
+    usage += s*2 + green('rep 12 oops') + ' will reply "' + yellow('oops') + '" to tweet with ' + yellow('[id=12]') + '.\n'
+    usage += s*2 + green('del 12 ') + ' will delete tweet with ' + yellow('[id=12]') + '.\n'
+    usage += s*2 + green('s #AKB48') + ' will search for "'+yellow('AKB48') + '" and return 5 newest tweet.\n'
+    usage += s*2 + green('fr') + ' will list out your following people.\n'
+    usage += s*2 + green('fl') + ' will list out your follower.\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 + '-'*70 + '\n'
+    usage += s + 'Have fun and hang tight!\n'
     printNicely(usage)
 
 
@@ -421,7 +425,7 @@ def reset():
     Reset prefix of line
     """
     if g['reset']:
-        printNicely(green('Need tips ? Type "h" and hit Enter key!'))
+        printNicely(magenta('Need tips ? Type "h" and hit Enter key!'))
     g['reset'] = False
 
 
@@ -474,7 +478,10 @@ def listen():
     init_interactive_shell(d)
     reset()
     while True:
-        line = raw_input(g['decorated_name'])
+        if g['prefix']:
+            line = raw_input(g['decorated_name'])
+        else:
+            line = raw_input()
         try:
             cmd = line.split()[0]
         except:
@@ -482,6 +489,10 @@ def listen():
         # Save cmd to global variable and call process
         g['stuff'] = ' '.join(line.split()[1:])
         process(cmd)()
+        if cmd in ['switch','t','rt','rep']:
+            g['prefix'] = False
+        else:
+            g['prefix'] = True
 
 
 def stream(domain, args, name='Rainbow Stream'):
@@ -551,6 +562,7 @@ def fly():
     # Start listen process
     time.sleep(0.5)
     g['reset'] = True
+    g['prefix'] = True
     g['stream_pid'] = p.pid
     listen()