Update README.md

[rainbowstream.git] / rainbowstream / rainbow.py

diff --git a/rainbowstream/rainbow.py b/rainbowstream/rainbow.py
index b4005299b6448189edce668a2f6bfb8ce9ec7bd9..888b261f99fdbf52a06b012e1f5b093fc1926615 100644 (file)
--- a/rainbowstream/rainbow.py
+++ b/rainbowstream/rainbow.py
@@ -49,7 +49,7 @@ cmdset = [
 ]
 
 
-def draw(t, imgflg = 0, keyword=None, fil=[], ig=[]):
+def draw(t, iot=False, keyword=None, fil=[], ig=[]):
     """
     Draw the rainbow
     """
@@ -145,7 +145,7 @@ def draw(t, imgflg = 0, keyword=None, fil=[], ig=[]):
     printNicely(line3)
 
     # Display Image
-    if imgflg and media_url:
+    if iot and media_url:
         for mu in media_url:
             response = requests.get(mu)
             image_to_display(StringIO(response.content))
@@ -183,9 +183,10 @@ def parse_arguments():
         '--ignore',
         help='Ignore specific screen_name.')
     parser.add_argument(
-        '-img',
-        '--image',
-        help='Display all photo on terminal.')
+        '-iot',
+        '--image-on-term',
+        action='store_true',
+        help='Display all image on terminal.')
     return parser.parse_args()
 
 
@@ -294,7 +295,7 @@ def home():
     if g['stuff'].isdigit():
         num = g['stuff']
     for tweet in reversed(t.statuses.home_timeline(count=num)):
-        draw(t=tweet, imgflg=g['image'])
+        draw(t=tweet, iot=g['iot'])
     printNicely('')
 
 
@@ -310,7 +311,7 @@ def view():
         except:
             num = HOME_TWEET_NUM
         for tweet in reversed(t.statuses.user_timeline(count=num, screen_name=user[1:])):
-            draw(t=tweet, imgflg=g['image'])
+            draw(t=tweet, iot=g['iot'])
         printNicely('')
     else:
         printNicely(red('A name should begin with a \'@\''))
@@ -347,7 +348,7 @@ def favorite():
         tid = db.rainbow_query(id)[0].tweet_id
         t.favorites.create(_id=tid, include_entities=False)
         printNicely(green('Favorited.'))
-        draw(t.statuses.show(id=tid), imgflg=g['image'])
+        draw(t.statuses.show(id=tid), iot=g['iot'])
     except:
         printNicely(red('Omg some syntax is wrong.'))
 
@@ -392,7 +393,7 @@ def unfavorite():
         tid = db.rainbow_query(id)[0].tweet_id
         t.favorites.destroy(_id=tid)
         printNicely(green('Okay it\'s unfavorited.'))
-        draw(t.statuses.show(id=tid), imgflg=g['image'])
+        draw(t.statuses.show(id=tid), iot=g['iot'])
     except:
         printNicely(red('Sorry I can\'t unfavorite this tweet for you.'))
 
@@ -409,7 +410,7 @@ def search():
                 printNicely('Newest tweets:')
                 for i in reversed(xrange(SEARCH_MAX_RECORD)):
                     draw(t=rel[i],
-                        imgflg=g['image'],
+                        iot=g['iot'],
                         keyword=g['stuff'].strip()[1:])
                 printNicely('')
             else:
@@ -523,8 +524,8 @@ def help():
         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('show image 12') + ' will show image in ' + \
-        yellow('[id=12]') + 'in your OS\'s image viewer.\n'
+    usage += s * 2 + green('show image 12') + ' will show image in tweet with ' + \
+        yellow('[id=12]') + ' in your OS\'s image viewer.\n'
     usage += s * 2 + green('fl @dtvd88') + ' will follow ' + \
         yellow('@dtvd88') + '.\n'
     usage += s * 2 + green('ufl @dtvd88') + ' will unfollow ' + \
@@ -690,7 +691,7 @@ def stream(domain, args, name='Rainbow Stream'):
         elif tweet.get('text'):
             draw(
                 t=tweet,
-                imgflg=args.image,
+                iot=args.image_on_term,
                 keyword=args.track_keywords,
                 fil=args.filter,
                 ig=args.ignore,
@@ -712,5 +713,5 @@ def fly():
     g['reset'] = True
     g['prefix'] = True
     g['stream_pid'] = p.pid
-    g['image'] = args.image
+    g['iot'] = args.image_on_term
     listen()
\ No newline at end of file