Merge branch 'use_arrow'

[rainbowstream.git] / rainbowstream / draw.py

diff --git a/rainbowstream/draw.py b/rainbowstream/draw.py
index 8bd0c50cd759cb11c0fb57affa2ed1c44ff7246f..11fa0000fb2fbf06e929a72d160fc1ea5ebd8758 100644 (file)
--- a/rainbowstream/draw.py
+++ b/rainbowstream/draw.py
@@ -3,6 +3,8 @@ import itertools
 import requests
 import datetime
 import time
 import requests
 import datetime
 import time

+import locale
+import arrow

 import re
 
 from twitter.util import printNicely
 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 .c_image import *
 from .colors import *
 from .config import *

-from .db import *

 from .py3patch import *
 
 from .py3patch import *
 

-
-db = RainbowDB()
-g = {}
+# Draw global variables
+dg = {}

 
 
 def init_cycle():
 
 
 def init_cycle():


@@ -35,8 +35,8 @@ def start_cycle():
     """
     Notify from rainbow
     """
     """
     Notify from rainbow
     """

-    g['cyc'] = init_cycle()
-    g['cache'] = {}
+    dg['cyc'] = init_cycle()
+    dg['cache'] = {}

 
 
 def order_rainbow(s):
 
 
 def order_rainbow(s):


@@ -69,9 +69,9 @@ def Memoize(func):
     """
     @wraps(func)
     def wrapper(*args):
     """
     @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
 
 
     return wrapper
 
 


@@ -80,7 +80,7 @@ def cycle_color(s):
     """
     Cycle the colors_shuffle
     """
     """
     Cycle the colors_shuffle
     """

-    return next(g['cyc'])(s)
+    return next(dg['cyc'])(s)

 
 
 def ascii_art(text):
 
 
 def ascii_art(text):


@@ -89,58 +89,51 @@ def ascii_art(text):
     """
     fi = figlet_format(text, font='doom')
     print('\n'.join(
     """
     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 = lmap(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_config():
-   """
-   Check if config is changed
-   """
-   changed = False
-   data = get_all_config()
-   for key in data:
-       if data[key] != c[key]:
-           changed = True
-   if changed:
-       reload_config()
+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 check_theme():
+def reload_theme(value, prev):

     """
     Check current theme and update if necessary
     """
     """
     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]
+    if value != prev:

         config = os.path.dirname(
         config = os.path.dirname(

-            __file__) + '/colorset/' + c['THEME'] + '.json'
+            __file__) + '/colorset/' + value + '.json'

         # Load new config
         data = load_config(config)
         if data:
             for d in data:
                 c[d] = data[d]
         # 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):
 
 
 def color_func(func_name):


@@ -152,13 +145,21 @@ def color_func(func_name):
     return globals()[func_name]
 
 
     return globals()[func_name]
 
 

-def draw(t, iot=False, keyword=None, check_semaphore=False, fil=[], ig=[]):
+def draw(t, keyword=None, humanize=True, check_semaphore=False, fil=[], ig=[]):

     """
     Draw the rainbow
     """
 
     """
     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()
     check_config()

+

     # Retrieve tweet
     tid = t['id']
     text = t['text']
     # Retrieve tweet
     tid = t['id']
     text = t['text']


@@ -166,9 +167,19 @@ def draw(t, iot=False, keyword=None, check_semaphore=False, fil=[], ig=[]):
     name = t['user']['name']
     created_at = t['created_at']
     favorited = t['favorited']
     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 = 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:
 
     # Pull extended retweet text
     try:


@@ -203,46 +214,58 @@ def draw(t, iot=False, keyword=None, check_semaphore=False, fil=[], ig=[]):
 
     # Filter and ignore
     screen_name = '@' + screen_name
 
     # 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
     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
 
     # 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:
     if favorited:

-        meta = meta + color_func(c['TWEET']['favorited'])(u'\u2605')
+        fav = color_func(c['TWEET']['favorited'])(u'\u2605')
+

     tweet = text.split()
     # Replace url
     if expanded_url:
     tweet = text.split()
     # Replace url
     if expanded_url:

-        for index in range(len(expanded_url)):
+        for index in xrange(len(expanded_url)):

             tweet = lmap(
             tweet = lmap(

-                lambda x: expanded_url[index] if x == url[index] else x,
+                lambda x: expanded_url[index]
+                if x == url[index]
+                else x,

                 tweet)
     # Highlight RT
     tweet = lmap(
                 tweet)
     # Highlight RT
     tweet = lmap(

-        lambda x: color_func(
-            c['TWEET']['rt'])(x) if x == 'RT' else x,
+        lambda x: color_func(c['TWEET']['rt'])(x)
+        if x == 'RT'
+        else x,

         tweet)
     # Highlight screen_name
     tweet = lmap(lambda x: cycle_color(x) if x[0] == '@' else x, tweet)
     # Highlight link
     tweet = lmap(
         tweet)
     # Highlight screen_name
     tweet = lmap(lambda x: cycle_color(x) if x[0] == '@' else x, tweet)
     # Highlight link
     tweet = lmap(

-        lambda x: color_func(
-            c['TWEET']['link'])(x) if x[
-            0:4] == 'http' else x,
+        lambda x: color_func(c['TWEET']['link'])(x)
+        if x[0:4] == 'http'
+        else x,
+        tweet)
+    # Highlight hashtag
+    tweet = lmap(
+        lambda x: color_func(c['TWEET']['hashtag'])(x)
+        if x.startswith('#')
+        else x,

         tweet)
         tweet)

-

     # Highlight keyword
     tweet = ' '.join(tweet)
     if keyword:
     # Highlight keyword
     tweet = ' '.join(tweet)
     if keyword:


@@ -250,33 +273,44 @@ def draw(t, iot=False, keyword=None, check_semaphore=False, fil=[], ig=[]):
         if roj:
             occur = roj.group()
             ary = tweet.split(occur)
         if roj:
             occur = roj.group()
             ary = tweet.split(occur)

-            delimeter = color_func(c['TWEET']['keyword'])(occur)
-            tweet = delimeter.join(ary)
+            delimiter = color_func(c['TWEET']['keyword'])(occur)
+            tweet = delimiter.join(ary)

 
 

-    # 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
-
-    # Check the semaphore lock
-    if check_semaphore:
-        while db.semaphore_query():
-            time.sleep(0.5)
+    # 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

 
 

-    # Output
-    printNicely('')
-    printNicely(line1)
-    printNicely(line2)
-    printNicely(line3)
+    # Draw
+    printNicely(formater)

 
     # Display Image
 
     # Display Image

-    if iot and media_url:
+    if c['IMAGE_ON_TERM'] and media_url:

         for mu in media_url:
             try:
                 response = requests.get(mu)
         for mu in media_url:
             try:
                 response = requests.get(mu)


@@ -285,10 +319,19 @@ def draw(t, iot=False, keyword=None, check_semaphore=False, fil=[], ig=[]):
                 printNicely(red('Sorry, image link is broken'))
 
 
                 printNicely(red('Sorry, image link is broken'))
 
 

-def print_message(m):
+def print_message(m, check_semaphore=False):

     """
     Print direct message
     """
     """
     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 = unescape(m['text'])
     sender_screen_name = '@' + m['sender_screen_name']
     sender_name = m['sender']['name']
     text = unescape(m['text'])


@@ -296,48 +339,60 @@ def print_message(m):
     recipient_name = m['recipient']['name']
     mid = m['id']
     date = parser.parse(m['created_at'])
     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
 
     # 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
 
     # 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(lmap(lambda x: x + '  ' if x == '\n' else x, 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)

 
 

-    line1 = u"{u:>{uw}}:".format(
-        u=user,
-        uw=len(user) + 2,
-    )
-    line2 = u"{c:>{cw}}".format(
-        c=meta,
-        cw=len(meta) + 2,
-    )
+    text = ''.join(lmap(lambda x: x + '  ' if x == '\n' else x, text))

 
 

-    line3 = '  ' + 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
     """
     """
     Show a profile
     """


@@ -377,8 +432,8 @@ def show_profile(u, iot=False):
     location = 'Location : ' + color_func(c['PROFILE']['location'])(location)
     url = 'URL : ' + (color_func(c['PROFILE']['url'])(url) if url else '')
     date = parser.parse(created_at)
     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
     clock = 'Join at ' + color_func(c['PROFILE']['clock'])(clock)
 
     # Format


@@ -410,10 +465,10 @@ def show_profile(u, iot=False):
     # Display
     printNicely('')
     printNicely(line1)
     # Display
     printNicely('')
     printNicely(line1)

-    if iot:
+    if c['IMAGE_ON_TERM']:

         try:
             response = requests.get(profile_image_url)
         try:
             response = requests.get(profile_image_url)

-            image_to_display(BytesIO(response.content), 2, 20)
+            image_to_display(BytesIO(response.content))

         except:
             pass
     else:
         except:
             pass
     else:


@@ -439,25 +494,25 @@ def print_list(group):
     """
     Display a list
     """
     """
     Display a list
     """

-    for g in group:
+    for grp in group:

         # Format
         # Format

-        name = g['full_name']
+        name = grp['full_name']

         name = color_func(c['GROUP']['name'])(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')
         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')
         subscriber = color_func(
             c['GROUP']['subscriber'])(
             subscriber +
             ' subscriber')

-        description = g['description'].strip()
+        description = grp['description'].strip()

         description = color_func(c['GROUP']['description'])(description)
         description = color_func(c['GROUP']['description'])(description)

-        mode = g['mode']
+        mode = grp['mode']

         mode = color_func(c['GROUP']['mode'])('Type: ' + 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 = 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
         clock = 'Created at ' + color_func(c['GROUP']['clock'])(clock)
 
         # Create lines


@@ -476,5 +531,78 @@ def print_list(group):
     printNicely('')
 
 
     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()
 # Start the color cycle
 start_cycle()