Merge branch 'use_arrow'
[rainbowstream.git] / rainbowstream / draw.py
index 78c3a26..11fa000 100644 (file)
@@ -3,6 +3,8 @@ import itertools
 import requests
 import datetime
 import time
+import locale
+import arrow
 import re
 
 from twitter.util import printNicely
@@ -12,12 +14,10 @@ from dateutil import parser
 from .c_image import *
 from .colors import *
 from .config import *
-from .db import *
 from .py3patch import *
 
-
-db = RainbowDB()
-g = {}
+# Draw global variables
+dg = {}
 
 
 def init_cycle():
@@ -35,8 +35,8 @@ def start_cycle():
     """
     Notify from rainbow
     """
-    g['cyc'] = init_cycle()
-    g['cache'] = {}
+    dg['cyc'] = init_cycle()
+    dg['cache'] = {}
 
 
 def order_rainbow(s):
@@ -69,9 +69,9 @@ def Memoize(func):
     """
     @wraps(func)
     def wrapper(*args):
-        if args not in g['cache']:
-            g['cache'][args] = func(*args)
-        return g['cache'][args]
+        if args not in dg['cache']:
+            dg['cache'][args] = func(*args)
+        return dg['cache'][args]
     return wrapper
 
 
@@ -80,7 +80,7 @@ def cycle_color(s):
     """
     Cycle the colors_shuffle
     """
-    return next(g['cyc'])(s)
+    return next(dg['cyc'])(s)
 
 
 def ascii_art(text):
@@ -89,7 +89,7 @@ def ascii_art(text):
     """
     fi = figlet_format(text, font='doom')
     print('\n'.join(
-        [next(g['cyc'])(i) for i in fi.split('\n')]
+        [next(dg['cyc'])(i) for i in fi.split('\n')]
     ))
 
 
@@ -117,24 +117,23 @@ def validate_theme(theme):
     return theme in themes
 
 
-def reload_theme(current_config):
+def reload_theme(value, prev):
     """
     Check current theme and update if necessary
     """
-    exists = db.theme_query()
-    themes = [t.theme_name for t in exists]
-    if current_config != themes[0]:
+    if value != prev:
         config = os.path.dirname(
-            __file__) + '/colorset/' + current_config + '.json'
+            __file__) + '/colorset/' + value + '.json'
         # Load new config
         data = load_config(config)
         if data:
             for d in data:
                 c[d] = data[d]
-        # Restart color cycle and update db/config
+        # Restart color cycle and update config
         start_cycle()
-        db.theme_update(current_config)
-        set_config('THEME', current_config)
+        set_config('THEME', value)
+        return value
+    return prev
 
 
 def color_func(func_name):
@@ -146,21 +145,20 @@ def color_func(func_name):
     return globals()[func_name]
 
 
-def draw(t, keyword=None, check_semaphore=False, fil=[], ig=[]):
+def draw(t, keyword=None, humanize=True, check_semaphore=False, fil=[], ig=[]):
     """
     Draw the rainbow
     """
 
-    # Check the semaphore lock (stream process only)
+    # Check the semaphore pause and lock (stream process only)
     if check_semaphore:
-        if db.semaphore_query_pause():
+        if c['pause']:
             return
-        while db.semaphore_query_lock():
+        while c['lock']:
             time.sleep(0.5)
 
-    # Check config and theme
+    # Check config
     check_config()
-    reload_theme(c['THEME'])
 
     # Retrieve tweet
     tid = t['id']
@@ -169,14 +167,19 @@ def draw(t, keyword=None, check_semaphore=False, fil=[], ig=[]):
     name = t['user']['name']
     created_at = t['created_at']
     favorited = t['favorited']
+    retweet_count = t['retweet_count']
+    favorite_count = t['favorite_count']
     date = parser.parse(created_at)
-    date = date - datetime.timedelta(seconds=time.timezone)
-    clock_format = '%Y/%m/%d %H:%M:%S'
-    try:
-        clock_format = c['FORMAT']['TWEET']['CLOCK_FORMAT']
-    except:
-        pass
-    clock = date.strftime(clock_format)
+    date = arrow.get(date).to('local')
+    if humanize:
+        lang, encode = locale.getdefaultlocale()
+        clock = arrow.get(date).to('local').humanize(locale=lang)
+    else:
+        try:
+            clock_format = c['FORMAT']['TWEET']['CLOCK_FORMAT']
+        except:
+            clock_format = '%Y/%m/%d %H:%M:%S'
+        clock = date.datetime.strftime(clock_format)
 
     # Pull extended retweet text
     try:
@@ -211,17 +214,19 @@ def draw(t, keyword=None, check_semaphore=False, fil=[], ig=[]):
 
     # Filter and ignore
     screen_name = '@' + screen_name
+    fil = list(set((fil or []) + c['ONLY_LIST']))
+    ig = list(set((ig or []) + c['IGNORE_LIST']))
     if fil and screen_name not in fil:
         return
     if ig and screen_name in ig:
         return
 
     # Get rainbow id
-    res = db.tweet_to_rainbow_query(tid)
-    if not res:
-        db.tweet_store(tid)
-        res = db.tweet_to_rainbow_query(tid)
-    rid = res[0].rainbow_id
+    if tid not in c['tweet_dict']:
+        c['tweet_dict'].append(tid)
+        rid = len(c['tweet_dict']) - 1
+    else:
+        rid = c['tweet_dict'].index(tid)
 
     # Format info
     name = cycle_color(name)
@@ -235,7 +240,7 @@ def draw(t, keyword=None, check_semaphore=False, fil=[], ig=[]):
     tweet = text.split()
     # Replace url
     if expanded_url:
-        for index in range(len(expanded_url)):
+        for index in xrange(len(expanded_url)):
             tweet = lmap(
                 lambda x: expanded_url[index]
                 if x == url[index]
@@ -272,12 +277,13 @@ def draw(t, keyword=None, check_semaphore=False, fil=[], ig=[]):
             tweet = delimiter.join(ary)
 
     # Load config formater
+    formater = ''
     try:
         formater = c['FORMAT']['TWEET']['DISPLAY']
-        formater = name.join(formater.split("#name"))
-        formater = nick.join(formater.split("#nick"))
-        formater = fav.join(formater.split("#fav"))
-        formater = tweet.join(formater.split("#tweet"))
+        formater = name.join(formater.split('#name'))
+        formater = nick.join(formater.split('#nick'))
+        formater = fav.join(formater.split('#fav'))
+        formater = tweet.join(formater.split('#tweet'))
         # Change clock word
         word = [w for w in formater.split() if '#clock' in w][0]
         delimiter = color_func(c['TWEET']['clock'])(
@@ -287,9 +293,18 @@ def draw(t, keyword=None, check_semaphore=False, fil=[], ig=[]):
         word = [w for w in formater.split() if '#id' in w][0]
         delimiter = color_func(c['TWEET']['id'])(id.join(word.split('#id')))
         formater = delimiter.join(formater.split(word))
+        # Change retweet count word
+        word = [w for w in formater.split() if '#rt_count' in w][0]
+        delimiter = color_func(c['TWEET']['retweet_count'])(
+            str(retweet_count).join(word.split('#rt_count')))
+        formater = delimiter.join(formater.split(word))
+        # Change favorites count word
+        word = [w for w in formater.split() if '#fa_count' in w][0]
+        delimiter = color_func(c['TWEET']['favorite_count'])(
+            str(favorite_count).join(word.split('#fa_count')))
+        formater = delimiter.join(formater.split(word))
     except:
-        printNicely(red('Wrong format in config.'))
-        return
+        pass
 
     # Draw
     printNicely(formater)
@@ -309,11 +324,11 @@ def print_message(m, check_semaphore=False):
     Print direct message
     """
 
-    # Check the semaphore lock (stream process only)
+    # Check the semaphore pause and lock (stream process only)
     if check_semaphore:
-        if db.semaphore_query_pause():
+        if c['pause']:
             return
-        while db.semaphore_query_lock():
+        while c['lock']:
             time.sleep(0.5)
 
     # Retrieve message
@@ -324,7 +339,7 @@ def print_message(m, check_semaphore=False):
     recipient_name = m['recipient']['name']
     mid = m['id']
     date = parser.parse(m['created_at'])
-    date = date - datetime.timedelta(seconds=time.timezone)
+    date = arrow.get(date).to('local').datetime
     clock_format = '%Y/%m/%d %H:%M:%S'
     try:
         clock_format = c['FORMAT']['MESSAGE']['CLOCK_FORMAT']
@@ -333,11 +348,11 @@ def print_message(m, check_semaphore=False):
     clock = date.strftime(clock_format)
 
     # Get rainbow id
-    res = db.message_to_rainbow_query(mid)
-    if not res:
-        db.message_store(mid)
-        res = db.message_to_rainbow_query(mid)
-    rid = res[0].rainbow_id
+    if mid not in c['message_dict']:
+        c['message_dict'].append(mid)
+        rid = len(c['message_dict']) - 1
+    else:
+        rid = c['message_dict'].index(mid)
 
     # Draw
     sender_name = cycle_color(sender_name)
@@ -417,8 +432,8 @@ def show_profile(u):
     location = 'Location : ' + color_func(c['PROFILE']['location'])(location)
     url = 'URL : ' + (color_func(c['PROFILE']['url'])(url) if url else '')
     date = parser.parse(created_at)
-    date = date - datetime.timedelta(seconds=time.timezone)
-    clock = date.strftime('%Y/%m/%d %H:%M:%S')
+    lang, encode = locale.getdefaultlocale()
+    clock = arrow.get(date).to('local').humanize(locale=lang)
     clock = 'Join at ' + color_func(c['PROFILE']['clock'])(clock)
 
     # Format
@@ -479,25 +494,25 @@ def print_list(group):
     """
     Display a list
     """
-    for g in group:
+    for grp in group:
         # Format
-        name = g['full_name']
+        name = grp['full_name']
         name = color_func(c['GROUP']['name'])(name + ' : ')
-        member = str(g['member_count'])
+        member = str(grp['member_count'])
         member = color_func(c['GROUP']['member'])(member + ' member')
-        subscriber = str(g['subscriber_count'])
+        subscriber = str(grp['subscriber_count'])
         subscriber = color_func(
             c['GROUP']['subscriber'])(
             subscriber +
             ' subscriber')
-        description = g['description'].strip()
+        description = grp['description'].strip()
         description = color_func(c['GROUP']['description'])(description)
-        mode = g['mode']
+        mode = grp['mode']
         mode = color_func(c['GROUP']['mode'])('Type: ' + mode)
-        created_at = g['created_at']
+        created_at = grp['created_at']
         date = parser.parse(created_at)
-        date = date - datetime.timedelta(seconds=time.timezone)
-        clock = date.strftime('%Y/%m/%d %H:%M:%S')
+        lang, encode = locale.getdefaultlocale()
+        clock = arrow.get(date).to('local').humanize(locale=lang)
         clock = 'Created at ' + color_func(c['GROUP']['clock'])(clock)
 
         # Create lines
@@ -536,5 +551,58 @@ def show_calendar(month, date, rel):
         printNicely(' '.join(ary))
 
 
+def format_quote(tweet):
+    """
+    Quoting format
+    """
+    # Retrieve info
+    screen_name = '@' + tweet['user']['screen_name']
+    text = tweet['text']
+    # Validate quote format
+    if '#owner' not in c['QUOTE_FORMAT']:
+        printNicely(light_magenta('Quote should contains #owner'))
+        return False
+    if '#comment' not in c['QUOTE_FORMAT']:
+        printNicely(light_magenta('Quote format should have #comment'))
+        return False
+    # Build formater
+    formater = ''
+    try:
+        formater = c['QUOTE_FORMAT']
+        formater = screen_name.join(formater.split('#owner'))
+        formater = text.join(formater.split('#tweet'))
+        formater = u2str(formater)
+    except:
+        pass
+    # Highlight like a tweet
+    formater = formater.split()
+    formater = lmap(
+        lambda x: light_green(x)
+        if x == '#comment'
+        else x,
+        formater)
+    formater = lmap(
+        lambda x: color_func(c['TWEET']['rt'])(x)
+        if x == 'RT'
+        else x,
+        formater)
+    formater = lmap(lambda x: cycle_color(x) if x[0] == '@' else x, formater)
+    formater = lmap(
+        lambda x: color_func(c['TWEET']['link'])(x)
+        if x[0:4] == 'http'
+        else x,
+        formater)
+    formater = lmap(
+        lambda x: color_func(c['TWEET']['hashtag'])(x)
+        if x.startswith('#')
+        else x,
+        formater)
+    formater = ' '.join(formater)
+    # Notice
+    notice = light_magenta('Quoting: "') + formater + light_magenta('"')
+    printNicely(notice)
+    return formater
+
+
 # Start the color cycle
 start_cycle()