X-Git-Url: https://vcs.fsf.org/?p=rainbowstream.git;a=blobdiff_plain;f=rainbowstream%2Fdraw.py;h=11fa0000fb2fbf06e929a72d160fc1ea5ebd8758;hp=bdeafdfc77ca8ce01d308c05d94b070d39156a2a;hb=b80f059ad8d12bd4f5d0665db06974f0f610606a;hpb=2a0cabee7d64c340ffa722472098c00aff2c1361

diff --git a/rainbowstream/draw.py b/rainbowstream/draw.py
index bdeafdf..11fa000 100644
--- a/rainbowstream/draw.py
+++ b/rainbowstream/draw.py
@@ -3,67 +3,75 @@ import itertools
 import requests
 import datetime
 import time
+import locale
+import arrow
+import re
 
 from twitter.util import printNicely
 from functools import wraps
 from pyfiglet import figlet_format
-from functools import reduce
-from StringIO import StringIO
 from dateutil import parser
 from .c_image import *
 from .colors import *
 from .config import *
-from .db import *
+from .py3patch import *
+
+# Draw global variables
+dg = {}
 
-db = RainbowDB()
-g = {}
 
 def init_cycle():
     """
     Init the cycle
     """
     colors_shuffle = [globals()[i.encode('utf8')]
-        if not i.startswith('term_')
-        else term_color(int(i[5:]))
-        for i in c['CYCLE_COLOR']]
-    return colors_shuffle, itertools.cycle(colors_shuffle)
-g['colors_shuffle'], g['cyc'] = init_cycle()
+                      if not str(i).isdigit()
+                      else term_color(int(i))
+                      for i in c['CYCLE_COLOR']]
+    return itertools.cycle(colors_shuffle)
 
 
-def notify_cycle():
+def start_cycle():
     """
     Notify from rainbow
     """
-    g['colors_shuffle'], g['cyc'] = init_cycle()
+    dg['cyc'] = init_cycle()
+    dg['cache'] = {}
 
 
 def order_rainbow(s):
     """
     Print a string with ordered color with each character
     """
-    c = [g['colors_shuffle'][i % 7](s[i]) for i in xrange(len(s))]
-    return reduce(lambda x, y: x + y, c)
+    colors_shuffle = [globals()[i.encode('utf8')]
+                      if not str(i).isdigit()
+                      else term_color(int(i))
+                      for i in c['CYCLE_COLOR']]
+    colored = [colors_shuffle[i % 7](s[i]) for i in xrange(len(s))]
+    return ''.join(colored)
 
 
 def random_rainbow(s):
     """
     Print a string with random color with each character
     """
-    c = [random.choice(g['colors_shuffle'])(i) for i in s]
-    return reduce(lambda x, y: x + y, c)
+    colors_shuffle = [globals()[i.encode('utf8')]
+                      if not str(i).isdigit()
+                      else term_color(int(i))
+                      for i in c['CYCLE_COLOR']]
+    colored = [random.choice(colors_shuffle)(i) for i in s]
+    return ''.join(colored)
 
 
 def Memoize(func):
     """
     Memoize decorator
     """
-    cache = {}
-
     @wraps(func)
     def wrapper(*args):
-        if args not in cache:
-            cache[args] = func(*args)
-        return cache[args]
+        if args not in dg['cache']:
+            dg['cache'][args] = func(*args)
+        return dg['cache'][args]
     return wrapper
 
 
@@ -72,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):
@@ -81,70 +89,77 @@ 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')]
     ))
 
 
-def show_calendar(month, date, rel):
+def check_config():
     """
-    Show the calendar in rainbow mode
+    Check if config is changed
     """
-    month = random_rainbow(month)
-    date = ' '.join([cycle_color(i) for i in date.split(' ')])
-    today = str(int(os.popen('date +\'%d\'').read().strip()))
-    # Display
-    printNicely(month)
-    printNicely(date)
-    for line in rel:
-        ary = line.split(' ')
-        ary = map(lambda x: color_func(c['CAL']['today'])(x)
-            if x == today
-            else color_func(c['CAL']['days'])(x)
-            , ary)
-        printNicely(' '.join(ary))
+    changed = False
+    data = get_all_config()
+    for key in c:
+        if key in data:
+            if data[key] != c[key]:
+                changed = True
+    if changed:
+        reload_config()
 
 
-def check_theme():
+def validate_theme(theme):
+    """
+    Validate a theme exists or not
+    """
+    # Theme changed check
+    files = os.listdir(os.path.dirname(__file__) + '/colorset')
+    themes = [f.split('.')[0] for f in files if f.split('.')[-1] == 'json']
+    return theme in themes
+
+
+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 c['theme'] != themes[0]:
-        c['theme'] = themes[0]
-        # Determine path
-        if c['theme'] == 'custom':
-            config = os.environ.get(
-                'HOME',
-                os.environ.get('USERPROFILE',
-                '')) + os.sep + '.rainbow_config.json'
-        else:
-            config = os.path.dirname(__file__) + '/colorset/'+c['theme']+'.json'
+    if value != prev:
+        config = os.path.dirname(
+            __file__) + '/colorset/' + value + '.json'
         # Load new config
         data = load_config(config)
         if data:
             for d in data:
                 c[d] = data[d]
-        # Re-init color cycle
-        g['cyc'] = init_cycle()
+        # Restart color cycle and update config
+        start_cycle()
+        set_config('THEME', value)
+        return value
+    return prev
 
 
 def color_func(func_name):
     """
     Call color function base on name
     """
-    pure = func_name.encode('utf8')
-    if pure.startswith('term_') and pure[5:].isdigit():
-        return term_color(int(pure[5:]))
-    return globals()[pure]
+    if str(func_name).isdigit():
+        return term_color(int(func_name))
+    return globals()[func_name]
 
 
-def draw(t, iot=False, keyword=None, fil=[], ig=[]):
+def draw(t, keyword=None, humanize=True, check_semaphore=False, fil=[], ig=[]):
     """
     Draw the rainbow
     """
 
-    check_theme()
+    # Check the semaphore pause and lock (stream process only)
+    if check_semaphore:
+        if c['pause']:
+            return
+        while c['lock']:
+            time.sleep(0.5)
+
+    # Check config
+    check_config()
+
     # Retrieve tweet
     tid = t['id']
     text = t['text']
@@ -152,9 +167,29 @@ def draw(t, iot=False, keyword=None, 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 = date.strftime('%Y/%m/%d %H:%M:%S')
+    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:
+        text = 'RT @' + t['retweeted_status']['user']['screen_name'] + ': ' +\
+            t['retweeted_status']['text']
+    except:
+        pass
+
+    # Unescape HTML character
+    text = unescape(text)
 
     # Get expanded url
     try:
@@ -179,132 +214,185 @@ def draw(t, iot=False, keyword=None, 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
-    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) + '] ')
+    name = cycle_color(name)
+    nick = color_func(c['TWEET']['nick'])(screen_name)
+    clock = clock
+    id = str(rid)
+    fav = ''
     if favorited:
-        meta = meta + color_func(c['TWEET']['favorite'])(u'\u2605')
+        fav = color_func(c['TWEET']['favorited'])(u'\u2605')
+
     tweet = text.split()
     # Replace url
     if expanded_url:
-        for index in range(len(expanded_url)):
-            tweet = map(
-                lambda x: expanded_url[index] if x == url[index] else x,
+        for index in xrange(len(expanded_url)):
+            tweet = lmap(
+                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 = lmap(
+        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)
+    tweet = lmap(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 = lmap(
+        lambda x: color_func(c['TWEET']['link'])(x)
+        if x[0:4] == 'http'
+        else x,
         tweet)
-    # Highlight search keyword
-    if keyword:
-        tweet = map(
-            lambda x: color_func(c['TWEET']['keyword'])(x) if
-            ''.join(c for c in x if c.isalnum()).lower() == keyword.lower()
-            else x,
-            tweet
-        )
-    # Recreate tweet
+    # Highlight hashtag
+    tweet = lmap(
+        lambda x: color_func(c['TWEET']['hashtag'])(x)
+        if x.startswith('#')
+        else x,
+        tweet)
+    # Highlight keyword
     tweet = ' '.join(tweet)
+    if keyword:
+        roj = re.search(keyword, tweet, re.IGNORECASE)
+        if roj:
+            occur = roj.group()
+            ary = tweet.split(occur)
+            delimiter = color_func(c['TWEET']['keyword'])(occur)
+            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'))
+        # Change clock word
+        word = [w for w in formater.split() if '#clock' in w][0]
+        delimiter = color_func(c['TWEET']['clock'])(
+            clock.join(word.split('#clock')))
+        formater = delimiter.join(formater.split(word))
+        # Change id word
+        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:
+        pass
 
-    # Draw rainbow
-    line1 = u"{u:>{uw}}:".format(
-        u=user,
-        uw=len(user) + 2,
-    )
-    line2 = u"{c:>{cw}}".format(
-        c=meta,
-        cw=len(meta) + 2,
-    )
-    line3 = '  ' + tweet
-
-    printNicely('')
-    printNicely(line1)
-    printNicely(line2)
-    printNicely(line3)
+    # Draw
+    printNicely(formater)
 
     # Display Image
-    if iot and media_url:
+    if c['IMAGE_ON_TERM'] and media_url:
         for mu in media_url:
-            response = requests.get(mu)
-            image_to_display(StringIO(response.content))
+            try:
+                response = requests.get(mu)
+                image_to_display(BytesIO(response.content))
+            except Exception:
+                printNicely(red('Sorry, image link is broken'))
 
 
-def print_message(m):
+def print_message(m, check_semaphore=False):
     """
     Print direct message
     """
+
+    # Check the semaphore pause and lock (stream process only)
+    if check_semaphore:
+        if c['pause']:
+            return
+        while c['lock']:
+            time.sleep(0.5)
+
+    # Retrieve message
     sender_screen_name = '@' + m['sender_screen_name']
     sender_name = m['sender']['name']
-    text = m['text']
+    text = unescape(m['text'])
     recipient_screen_name = '@' + m['recipient_screen_name']
     recipient_name = m['recipient']['name']
     mid = m['id']
     date = parser.parse(m['created_at'])
-    date = date - datetime.timedelta(seconds=time.timezone)
-    clock = date.strftime('%Y/%m/%d %H:%M:%S')
+    date = arrow.get(date).to('local').datetime
+    clock_format = '%Y/%m/%d %H:%M:%S'
+    try:
+        clock_format = c['FORMAT']['MESSAGE']['CLOCK_FORMAT']
+    except:
+        pass
+    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 = 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) + '] ')
-    text = ''.join(map(lambda x: x + '  ' if x == '\n' else x, text))
-
-    line1 = u"{u:>{uw}}:".format(
-        u=user,
-        uw=len(user) + 2,
-    )
-    line2 = u"{c:>{cw}}".format(
-        c=meta,
-        cw=len(meta) + 2,
-    )
-
-    line3 = '  ' + text
+    sender_name = cycle_color(sender_name)
+    sender_nick = color_func(c['MESSAGE']['sender'])(sender_screen_name)
+    recipient_name = cycle_color(recipient_name)
+    recipient_nick = color_func(
+        c['MESSAGE']['recipient'])(recipient_screen_name)
+    to = color_func(c['MESSAGE']['to'])('>>>')
+    clock = clock
+    id = str(rid)
+
+    text = ''.join(lmap(lambda x: x + '  ' if x == '\n' else x, text))
+
+    # Load config formater
+    try:
+        formater = c['FORMAT']['MESSAGE']['DISPLAY']
+        formater = sender_name.join(formater.split("#sender_name"))
+        formater = sender_nick.join(formater.split("#sender_nick"))
+        formater = to.join(formater.split("#to"))
+        formater = recipient_name.join(formater.split("#recipient_name"))
+        formater = recipient_nick.join(formater.split("#recipient_nick"))
+        formater = text.join(formater.split("#message"))
+        # Change clock word
+        word = [w for w in formater.split() if '#clock' in w][0]
+        delimiter = color_func(c['MESSAGE']['clock'])(
+            clock.join(word.split('#clock')))
+        formater = delimiter.join(formater.split(word))
+        # Change id word
+        word = [w for w in formater.split() if '#id' in w][0]
+        delimiter = color_func(c['MESSAGE']['id'])(id.join(word.split('#id')))
+        formater = delimiter.join(formater.split(word))
+    except:
+        printNicely(red('Wrong format in config.'))
+        return
 
-    printNicely('')
-    printNicely(line1)
-    printNicely(line2)
-    printNicely(line3)
+    # Draw
+    printNicely(formater)
 
 
-def show_profile(u, iot=False):
+def show_profile(u):
     """
     Show a profile
     """
@@ -339,13 +427,13 @@ def show_profile(u, iot=False):
     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))
+        lmap(lambda x: x + ' ' * 4 if x == '\n' else x, description))
     description = color_func(c['PROFILE']['description'])(description)
     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
@@ -377,9 +465,12 @@ def show_profile(u, iot=False):
     # Display
     printNicely('')
     printNicely(line1)
-    if iot:
-        response = requests.get(profile_image_url)
-        image_to_display(StringIO(response.content), 2, 20)
+    if c['IMAGE_ON_TERM']:
+        try:
+            response = requests.get(profile_image_url)
+            image_to_display(BytesIO(response.content))
+        except:
+            pass
     else:
         printNicely(line2)
     for line in [line3, line4, line5, line6]:
@@ -397,3 +488,121 @@ def print_trends(trends):
         line = cycle_color(name) + ': ' + color_func(c['TREND']['url'])(url)
         printNicely(line)
     printNicely('')
+
+
+def print_list(group):
+    """
+    Display a list
+    """
+    for grp in group:
+        # Format
+        name = grp['full_name']
+        name = color_func(c['GROUP']['name'])(name + ' : ')
+        member = str(grp['member_count'])
+        member = color_func(c['GROUP']['member'])(member + ' member')
+        subscriber = str(grp['subscriber_count'])
+        subscriber = color_func(
+            c['GROUP']['subscriber'])(
+            subscriber +
+            ' subscriber')
+        description = grp['description'].strip()
+        description = color_func(c['GROUP']['description'])(description)
+        mode = grp['mode']
+        mode = color_func(c['GROUP']['mode'])('Type: ' + mode)
+        created_at = grp['created_at']
+        date = parser.parse(created_at)
+        lang, encode = locale.getdefaultlocale()
+        clock = arrow.get(date).to('local').humanize(locale=lang)
+        clock = 'Created at ' + color_func(c['GROUP']['clock'])(clock)
+
+        # Create lines
+        line1 = ' ' * 2 + name + member + '  ' + subscriber
+        line2 = ' ' * 4 + description
+        line3 = ' ' * 4 + mode
+        line4 = ' ' * 4 + clock
+
+        # Display
+        printNicely('')
+        printNicely(line1)
+        printNicely(line2)
+        printNicely(line3)
+        printNicely(line4)
+
+    printNicely('')
+
+
+def show_calendar(month, date, rel):
+    """
+    Show the calendar in rainbow mode
+    """
+    month = random_rainbow(month)
+    date = ' '.join([cycle_color(i) for i in date.split(' ')])
+    today = str(int(os.popen('date +\'%d\'').read().strip()))
+    # Display
+    printNicely(month)
+    printNicely(date)
+    for line in rel:
+        ary = line.split(' ')
+        ary = lmap(
+            lambda x: color_func(c['CAL']['today'])(x)
+            if x == today
+            else color_func(c['CAL']['days'])(x),
+            ary)
+        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()