import random
+import textwrap
import itertools
import requests
import locale
import re
import os
+from io import BytesIO
from twitter.util import printNicely
from functools import wraps
from pyfiglet import figlet_format
from .colors import *
from .config import *
from .py3patch import *
+from .emoji import *
# Draw global variables
dg = {}
"""
dg['cyc'] = init_cycle()
dg['cache'] = {}
+ dg['humanize_unsupported'] = False
def order_rainbow(s):
return globals()[func_name]
-def draw(t, keyword=None, humanize=True, fil=[], ig=[]):
+def fallback_humanize(date, fallback_format=None, use_fallback=False):
+ """
+ Format date with arrow and a fallback format
+ """
+ # Convert to local timezone
+ date = arrow.get(date).to('local')
+ # Set default fallback format
+ if not fallback_format:
+ fallback_format = '%Y/%m/%d %H:%M:%S'
+ # Determine using fallback format or not by a variable
+ if use_fallback:
+ return date.datetime.strftime(fallback_format)
+ try:
+ # Use Arrow's humanize function
+ lang, encode = locale.getdefaultlocale()
+ clock = date.humanize(locale=lang)
+ except:
+ # Notice at the 1st time only
+ if not dg['humanize_unsupported']:
+ dg['humanize_unsupported'] = True
+ printNicely(
+ light_magenta('Humanized date display method does not support your $LC_ALL.'))
+ # Fallback when LC_ALL is not supported
+ clock = date.datetime.strftime(fallback_format)
+ return clock
+
+
+def draw(t, keyword=None, humanize=True, noti=False, fil=[], ig=[]):
"""
Draw the rainbow
"""
favorited = t['favorited']
retweet_count = t['retweet_count']
favorite_count = t['favorite_count']
+ client = t['source']
date = parser.parse(created_at)
- 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)
+ try:
+ clock_format = c['FORMAT']['TWEET']['CLOCK_FORMAT']
+ except:
+ clock_format = '%Y/%m/%d %H:%M:%S'
+ clock = fallback_humanize(date, clock_format, not humanize)
# Pull extended retweet text
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
# Unescape HTML character
text = unescape(text)
+ # Get client name
+ try:
+ client = client.split('>')[-2].split('<')[0]
+ except:
+ client = None
+
# Get expanded url
try:
expanded_url = []
media_url = None
# Filter and ignore
+ mytweet = screen_name == c['original_name']
screen_name = '@' + screen_name
fil = list(set((fil or []) + c['ONLY_LIST']))
ig = list(set((ig or []) + c['IGNORE_LIST']))
# 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 = ''
if favorited:
fav = color_func(c['TWEET']['favorited'])(u'\u2605')
- tweet = text.split()
+ tweet = text.split(' ')
+ tweet = [x for x in tweet if x != '']
# Replace url
if expanded_url:
for index in xrange(len(expanded_url)):
else x,
tweet)
# Highlight screen_name
- tweet = lmap(lambda x: cycle_color(x) if x[0] == '@' else x, tweet)
+ tweet = lmap(
+ lambda x: cycle_color(x) if x.lstrip().startswith('@') else x, tweet)
# Highlight link
tweet = lmap(
lambda x: color_func(c['TWEET']['link'])(x)
- if x[0:4] == 'http'
+ if x.lstrip().startswith('http')
else x,
tweet)
# Highlight hashtag
tweet = lmap(
lambda x: color_func(c['TWEET']['hashtag'])(x)
- if x.startswith('#')
+ if x.lstrip().startswith('#')
else x,
tweet)
+ # Highlight my tweet
+ if mytweet:
+ tweet = [color_func(c['TWEET']['mytweet'])(x)
+ for x in tweet
+ if not any([
+ x == 'RT',
+ x.lstrip().startswith('http'),
+ x.lstrip().startswith('#')])
+ ]
# Highlight keyword
tweet = ' '.join(tweet)
+ tweet = '\n '.join(tweet.split('\n'))
if keyword:
roj = re.search(keyword, tweet, re.IGNORECASE)
if roj:
formater = nick.join(formater.split('#nick'))
formater = fav.join(formater.split('#fav'))
formater = tweet.join(formater.split('#tweet'))
+ formater = emojize(formater)
# Change clock word
word = [wo for wo in formater.split() if '#clock' in wo][0]
delimiter = color_func(c['TWEET']['clock'])(
delimiter = color_func(c['TWEET']['favorite_count'])(
str(favorite_count).join(word.split('#fa_count')))
formater = delimiter.join(formater.split(word))
+ # Change client word
+ word = [wo for wo in formater.split() if '#client' in wo][0]
+ delimiter = color_func(c['TWEET']['client'])(
+ client.join(word.split('#client')))
+ formater = delimiter.join(formater.split(word))
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)
messages = dg['thread'][partner]
messages.sort(key=lambda x: parser.parse(x['created_at']))
# Use legacy display on non-ascii text message
- text_ary = [m['text'] for m in messages]
- not_ascii_text_ary = [t for t in text_ary if not is_ascii(t)]
- if not_ascii_text_ary:
+ ms = [m['text'] for m in messages]
+ ums = [m for m in ms if not all(ord(c) < 128 for c in m)]
+ if ums:
for m in messages:
print_message(m)
printNicely('')
h, w = os.popen('stty size', 'r').read().split()
w = int(w)
frame_width = w // 3 - dg['frame_margin']
+ frame_width = max(c['THREAD_MIN_WIDTH'], frame_width)
step = frame_width - 2 * dg['frame_margin']
- slicing = [m['text'][i:i + step] for i in range(0, len(m['text']), step)]
+ slicing = textwrap.wrap(m['text'], step)
spaces = w - frame_width - dg['frame_margin']
dotline = ' ' * spaces + '-' * frame_width
dotline = color_func(c['MESSAGE']['me_frame'])(dotline)
word = [wo for wo in formater.split() if '#id' in wo][0]
delimiter = color_func(c['MESSAGE']['id'])(id.join(word.split('#id')))
formater = delimiter.join(formater.split(word))
+ formater = emojize(formater)
except Exception:
printNicely(red('Wrong format in config.'))
return
h, w = os.popen('stty size', 'r').read().split()
w = int(w)
frame_width = w // 3 - dg['frame_margin']
+ frame_width = max(c['THREAD_MIN_WIDTH'], frame_width)
step = frame_width - 2 * dg['frame_margin']
- slicing = [m['text'][i:i + step] for i in range(0, len(m['text']), step)]
+ slicing = textwrap.wrap(m['text'], step)
spaces = dg['frame_margin']
dotline = ' ' * spaces + '-' * frame_width
dotline = color_func(c['MESSAGE']['partner_frame'])(dotline)
word = [wo for wo in formater.split() if '#id' in wo][0]
delimiter = color_func(c['MESSAGE']['id'])(id.join(word.split('#id')))
formater = delimiter.join(formater.split(word))
+ formater = emojize(formater)
except Exception:
printNicely(red('Wrong format in config.'))
return
word = [wo for wo in formater.split() if '#id' in wo][0]
delimiter = color_func(c['MESSAGE']['id'])(id.join(word.split('#id')))
formater = delimiter.join(formater.split(word))
+ formater = emojize(formater)
except:
printNicely(red('Wrong format in config.'))
return
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)
+ clock = fallback_humanize(date)
+ 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'))
+ meta = emojize(meta)
+ # 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)
+ clock = fallback_humanize(date)
+ 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'))
+ meta = emojize(meta)
+ # 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)
+ clock = fallback_humanize(date)
+ 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'))
+ meta = emojize(meta)
+ # 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)
+ clock = fallback_humanize(date)
+ 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'))
+ meta = emojize(meta)
+ # 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)
+ clock = fallback_humanize(date)
+ 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'))
+ meta = emojize(meta)
+ # 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)
+ clock = fallback_humanize(date)
+ 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'))
+ meta = emojize(meta)
+ # 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)
+ clock = fallback_humanize(date)
+ 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'))
+ meta = emojize(meta)
+ # 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)
+ clock = fallback_humanize(date)
+ 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'))
+ meta = emojize(meta)
+ # Output
+ printNicely('')
+ printNicely(meta)
+ print_list(target_object, noti=True)
+
+
+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'], lambda *args: None)(e)
+
+
def show_profile(u):
"""
Show a profile
location = 'Location : ' + color_func(c['PROFILE']['location'])(location)
url = 'URL : ' + (color_func(c['PROFILE']['url'])(url) if url else '')
date = parser.parse(created_at)
- lang, encode = locale.getdefaultlocale()
- clock = arrow.get(date).to('local').humanize(locale=lang)
- clock = 'Join at ' + color_func(c['PROFILE']['clock'])(clock)
+ clock = fallback_humanize(date)
+ clock = 'Joined ' + color_func(c['PROFILE']['clock'])(clock)
# Format
line1 = u"{u:>{uw}}".format(
printNicely('')
-def print_list(group):
+def print_list(group, noti=False):
"""
Display a list
"""
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 = fallback_humanize(date)
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('')
printNicely(line3)
printNicely(line4)
- printNicely('')
+ if not noti:
+ printNicely('')
def show_calendar(month, date, rel):
Quoting format
"""
# Retrieve info
- screen_name = '@' + tweet['user']['screen_name']
- text = tweet['text']
+ screen_name = str( tweet['user']['screen_name'] )
+ text = str( tweet['text'] )
+ tid = str( tweet['id'] )
+
# Validate quote format
if '#owner' not in c['QUOTE_FORMAT']:
printNicely(light_magenta('Quote should contains #owner'))
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)
+
+ formater = formater.replace('#owner', screen_name)
+ formater = formater.replace('#tweet', text)
+ formater = formater.replace('#tid', tid)
+
+ formater = emojize(formater)
except:
pass
# Highlight like a tweet