do nothing when meet other event in stream

[rainbowstream.git] / rainbowstream / draw.py

diff --git a/rainbowstream/draw.py b/rainbowstream/draw.py
index 5bde76c4c8074741ea8a948c05c93cea7134c089..e8cbcaf031745e445f85c115f2583fea4c01aa22 100644 (file)
--- a/rainbowstream/draw.py
+++ b/rainbowstream/draw.py
@@ -145,7 +145,7 @@ def color_func(func_name):
     return globals()[func_name]
 
 
-def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
+def draw(t, keyword=None, humanize=True, noti=False, fil=[], ig=[]):
     """
     Draw the rainbow
     """
@@ -177,6 +177,14 @@ def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
     try:
         text = 'RT @' + t['retweeted_status']['user']['screen_name'] + ': ' +\
             t['retweeted_status']['text']
+        # Display as a notification
+        target = t['retweeted_status']['user']['screen_name']
+        if all([target == c['original_name'], not noti]):
+            # Add to evens for 'notification' command
+            t['event'] = 'retweet'
+            c['events'].append(t)
+            notify_retweet(t)
+            return
     except:
         pass
 
@@ -223,7 +231,10 @@ def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
 
     # Format info
     name = cycle_color(name)
-    nick = color_func(c['TWEET']['nick'])(screen_name)
+    if mytweet:
+        nick = color_func(c['TWEET']['mynick'])(screen_name)
+    else:
+        nick = color_func(c['TWEET']['nick'])(screen_name)
     clock = clock
     id = str(rid)
     fav = ''
@@ -308,6 +319,13 @@ def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
     except:
         pass
 
+    # Add spaces in begining of line if this is inside a notification
+    if noti:
+        formater = '\n  '.join(formater.split('\n'))
+        # Reformat
+        if formater.startswith('\n'):
+            formater = formater[1:]
+
     # Draw
     printNicely(formater)
 
@@ -579,6 +597,279 @@ def print_message(m):
     printNicely(formater)
 
 
+def notify_retweet(t):
+    """
+    Notify a retweet
+    """
+    source = t['user']
+    created_at = t['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'retweeted your tweet')
+    date = parser.parse(created_at)
+    date = arrow.get(date).to('local')
+    lang, encode = locale.getdefaultlocale()
+    clock = arrow.get(date).to('local').humanize(locale=lang)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    # Output
+    printNicely('')
+    printNicely(meta)
+    draw(t=t['retweeted_status'], noti=True)
+
+
+def notify_favorite(e):
+    """
+    Notify a favorite event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = e['target_object']
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'favorited your tweet')
+    date = parser.parse(created_at)
+    date = arrow.get(date).to('local')
+    lang, encode = locale.getdefaultlocale()
+    clock = arrow.get(date).to('local').humanize(locale=lang)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    # Output
+    printNicely('')
+    printNicely(meta)
+    draw(t=target_object, noti=True)
+
+
+def notify_unfavorite(e):
+    """
+    Notify a unfavorite event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = e['target_object']
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'unfavorited your tweet')
+    date = parser.parse(created_at)
+    date = arrow.get(date).to('local')
+    lang, encode = locale.getdefaultlocale()
+    clock = arrow.get(date).to('local').humanize(locale=lang)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    # Output
+    printNicely('')
+    printNicely(meta)
+    draw(t=target_object, noti=True)
+
+
+def notify_follow(e):
+    """
+    Notify a follow event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'followed you')
+    date = parser.parse(created_at)
+    date = arrow.get(date).to('local')
+    lang, encode = locale.getdefaultlocale()
+    clock = arrow.get(date).to('local').humanize(locale=lang)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    # Output
+    printNicely('')
+    printNicely(meta)
+
+
+def notify_list_member_added(e):
+    """
+    Notify a list_member_added event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = [e['target_object']]  # list of Twitter list
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'added you to a list')
+    date = parser.parse(created_at)
+    date = arrow.get(date).to('local')
+    lang, encode = locale.getdefaultlocale()
+    clock = arrow.get(date).to('local').humanize(locale=lang)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    # Output
+    printNicely('')
+    printNicely(meta)
+    print_list(target_object, noti=True)
+
+
+def notify_list_member_removed(e):
+    """
+    Notify a list_member_removed event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = [e['target_object']]  # list of Twitter list
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'removed you from a list')
+    date = parser.parse(created_at)
+    date = arrow.get(date).to('local')
+    lang, encode = locale.getdefaultlocale()
+    clock = arrow.get(date).to('local').humanize(locale=lang)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    # Output
+    printNicely('')
+    printNicely(meta)
+    print_list(target_object, noti=True)
+
+
+def notify_list_user_subscribed(e):
+    """
+    Notify a list_user_subscribed event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = [e['target_object']]  # list of Twitter list
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'subscribed to your list')
+    date = parser.parse(created_at)
+    date = arrow.get(date).to('local')
+    lang, encode = locale.getdefaultlocale()
+    clock = arrow.get(date).to('local').humanize(locale=lang)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    # Output
+    printNicely('')
+    printNicely(meta)
+    print_list(target_object, noti=True)
+
+
+def notify_list_user_unsubscribed(e):
+    """
+    Notify a list_user_unsubscribed event
+    """
+    # Retrieve info
+    target = e['target']
+    if target['screen_name'] != c['original_name']:
+        return
+    source = e['source']
+    target_object = [e['target_object']]  # list of Twitter list
+    created_at = e['created_at']
+    # Format
+    source_user = cycle_color(source['name']) + \
+        color_func(c['NOTIFICATION']['source_nick'])(
+        ' @' + source['screen_name'])
+    notify = color_func(c['NOTIFICATION']['notify'])(
+        'unsubscribed from your list')
+    date = parser.parse(created_at)
+    date = arrow.get(date).to('local')
+    lang, encode = locale.getdefaultlocale()
+    clock = arrow.get(date).to('local').humanize(locale=lang)
+    clock = color_func(c['NOTIFICATION']['clock'])(clock)
+    meta = c['NOTIFY_FORMAT']
+    meta = source_user.join(meta.split('#source_user'))
+    meta = notify.join(meta.split('#notify'))
+    meta = clock.join(meta.split('#clock'))
+    # Output
+    printNicely('')
+    printNicely(meta)
+    print_list(target_object, noti=True)
+
+
+def nothing(e):
+    """
+    Do nothing for other event
+    """
+    return
+
+
+def print_event(e):
+    """
+    Notify an event
+    """
+    event_dict = {
+        'retweet': notify_retweet,
+        'favorite': notify_favorite,
+        'unfavorite': notify_unfavorite,
+        'follow': notify_follow,
+        'list_member_added': notify_list_member_added,
+        'list_member_removed': notify_list_member_removed,
+        'list_user_subscribed': notify_list_user_subscribed,
+        'list_user_unsubscribed': notify_list_user_unsubscribed,
+    }
+    event_dict.get(e['event'],nothing)(e)
+
+
 def show_profile(u):
     """
     Show a profile
@@ -677,7 +968,7 @@ def print_trends(trends):
     printNicely('')
 
 
-def print_list(group):
+def print_list(group, noti=False):
     """
     Display a list
     """
@@ -702,11 +993,15 @@ def print_list(group):
         clock = arrow.get(date).to('local').humanize(locale=lang)
         clock = 'Created at ' + color_func(c['GROUP']['clock'])(clock)
 
+        prefix = ' ' * 2
+        # Add spaces in begining of line if this is inside a notification
+        if noti:
+            prefix = ' ' * 2 + prefix
         # Create lines
-        line1 = ' ' * 2 + name + member + '  ' + subscriber
-        line2 = ' ' * 4 + description
-        line3 = ' ' * 4 + mode
-        line4 = ' ' * 4 + clock
+        line1 = prefix + name + member + '  ' + subscriber
+        line2 = prefix + ' ' * 2 + description
+        line3 = prefix + ' ' * 2 + mode
+        line4 = prefix + ' ' * 2 + clock
 
         # Display
         printNicely('')
@@ -715,7 +1010,8 @@ def print_list(group):
         printNicely(line3)
         printNicely(line4)
 
-    printNicely('')
+    if not noti:
+        printNicely('')
 
 
 def show_calendar(month, date, rel):