X-Git-Url: https://vcs.fsf.org/?a=blobdiff_plain;ds=inline;f=rainbowstream%2Frainbow.py;h=f31fa23308f284988d5a918b9880c5eefbe4c694;hb=632c6fa515fe36cf88ef2bf39378464bd24701d8;hp=60e9890e4ab4076f858398dd37ff5487a55dc336;hpb=1f24a05aaef004ff9fada95cc80984eaab3d2601;p=rainbowstream.git diff --git a/rainbowstream/rainbow.py b/rainbowstream/rainbow.py index 60e9890..f31fa23 100644 --- a/rainbowstream/rainbow.py +++ b/rainbowstream/rainbow.py @@ -2,7 +2,6 @@ Colorful user's timeline stream """ from multiprocessing import Process -from dateutil import parser import os import os.path @@ -10,7 +9,6 @@ import sys import signal import argparse import time -import datetime import requests from twitter.stream import TwitterStream, Timeout, HeartbeatTimeout, Hangup @@ -20,6 +18,7 @@ from twitter.oauth_dance import oauth_dance from twitter.util import printNicely from StringIO import StringIO +from .draw import * from .colors import * from .config import * from .consumer import * @@ -59,241 +58,12 @@ cmdset = [ 'unblock', 'report', 'cal', + 'theme', 'h', 'c', 'q' ] - -def draw(t, iot=False, keyword=None, fil=[], ig=[]): - """ - Draw the rainbow - """ - - # Retrieve tweet - tid = t['id'] - text = t['text'] - screen_name = t['user']['screen_name'] - name = t['user']['name'] - created_at = t['created_at'] - favorited = t['favorited'] - date = parser.parse(created_at) - date = date - datetime.timedelta(seconds=time.timezone) - clock = date.strftime('%Y/%m/%d %H:%M:%S') - - # Get expanded url - try: - expanded_url = [] - url = [] - urls = t['entities']['urls'] - for u in urls: - expanded_url.append(u['expanded_url']) - url.append(u['url']) - except: - expanded_url = None - url = None - - # Get media - try: - media_url = [] - media = t['entities']['media'] - for m in media: - media_url.append(m['media_url']) - except: - media_url = None - - # Filter and ignore - screen_name = '@' + screen_name - 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 - - # Format info - user = cycle_color(name) + grey(' ' + screen_name + ' ') - meta = grey('[' + clock + '] [id=' + str(rid) + '] ') - if favorited: - meta = meta + green(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, - tweet) - # Highlight RT - tweet = map(lambda x: grey(x) if x == 'RT' else x, tweet) - # Highlight screen_name - tweet = map(lambda x: cycle_color(x) if x[0] == '@' else x, tweet) - # Highlight link - tweet = map(lambda x: cyan(x) if x[0:4] == 'http' else x, tweet) - # Highlight search keyword - if keyword: - tweet = map( - lambda x: on_yellow(x) if - ''.join(c for c in x if c.isalnum()).lower() == keyword.lower() - else x, - tweet - ) - # Recreate tweet - tweet = ' '.join(tweet) - - # 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) - - # Display Image - if iot and media_url: - for mu in media_url: - response = requests.get(mu) - image_to_display(StringIO(response.content)) - - -def print_message(m): - """ - Print direct message - """ - sender_screen_name = '@' + m['sender_screen_name'] - sender_name = m['sender']['name'] - text = 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') - - # 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 - - sender = cycle_color(sender_name) + grey(' ' + sender_screen_name + ' ') - recipient = cycle_color( - recipient_name) + grey(' ' + recipient_screen_name + ' ') - user = sender + magenta(' >>> ') + recipient - meta = grey('[' + clock + '] [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 - - printNicely('') - printNicely(line1) - printNicely(line2) - printNicely(line3) - - -def show_profile(u): - """ - Show a profile - """ - # Retrieve info - name = u['name'] - screen_name = u['screen_name'] - description = u['description'] - profile_image_url = u['profile_image_url'] - location = u['location'] - url = u['url'] - created_at = u['created_at'] - statuses_count = u['statuses_count'] - friends_count = u['friends_count'] - followers_count = u['followers_count'] - # Create content - statuses_count = green(str(statuses_count) + ' tweets') - friends_count = green(str(friends_count) + ' following') - followers_count = green(str(followers_count) + ' followers') - count = statuses_count + ' ' + friends_count + ' ' + followers_count - user = cycle_color(name) + grey(' @' + screen_name + ' : ') + count - profile_image_raw_url = 'Profile photo: ' + cyan(profile_image_url) - description = ''.join( - map(lambda x: x + ' ' * 4 if x == '\n' else x, description)) - description = yellow(description) - location = 'Location : ' + magenta(location) - url = 'URL : ' + (cyan(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') - clock = 'Join at ' + white(clock) - # Format - line1 = u"{u:>{uw}}".format( - u=user, - uw=len(user) + 2, - ) - line2 = u"{p:>{pw}}".format( - p=profile_image_raw_url, - pw=len(profile_image_raw_url) + 4, - ) - line3 = u"{d:>{dw}}".format( - d=description, - dw=len(description) + 4, - ) - line4 = u"{l:>{lw}}".format( - l=location, - lw=len(location) + 4, - ) - line5 = u"{u:>{uw}}".format( - u=url, - uw=len(url) + 4, - ) - line6 = u"{c:>{cw}}".format( - c=clock, - cw=len(clock) + 4, - ) - # Display - printNicely('') - printNicely(line1) - if g['iot']: - response = requests.get(profile_image_url) - image_to_display(StringIO(response.content), 2, 20) - else: - printNicely(line2) - for line in [line3, line4, line5, line6]: - printNicely(line) - printNicely('') - - -def print_trends(trends): - """ - Display topics - """ - for topic in trends[:TREND_MAX]: - name = topic['name'] - url = topic['url'] - line = cycle_color(name) + ': ' + cyan(url) - printNicely(line) - printNicely('') - - def parse_arguments(): """ Parse the arguments @@ -365,6 +135,10 @@ def get_decorated_name(): g['original_name'] = name[1:] g['decorated_name'] = grey('[') + grey(name) + grey(']: ') + files = os.listdir('rainbowstream/colorset') + themes = [f.split('.')[0] for f in files if f.split('.')[-1]=='json'] + g['themes'] = themes + def switch(): """ @@ -382,8 +156,8 @@ def switch(): args.filter = filter(None, only.split(',')) args.ignore = filter(None, ignore.split(',')) elif g['stuff'].split()[-1] == '-d': - args.filter = ONLY_LIST - args.ignore = IGNORE_LIST + args.filter = c['ONLY_LIST'] + args.ignore = c['IGNORE_LIST'] except: printNicely(red('Sorry, wrong format.')) return @@ -400,7 +174,7 @@ def switch(): p = Process( target=stream, args=( - PUBLIC_DOMAIN, + c['PUBLIC_DOMAIN'], args)) p.start() g['stream_pid'] = p.pid @@ -413,7 +187,7 @@ def switch(): p = Process( target=stream, args=( - USER_DOMAIN, + c['USER_DOMAIN'], args, g['original_name'])) p.start() @@ -471,7 +245,7 @@ def home(): Home """ t = Twitter(auth=authen()) - num = HOME_TWEET_NUM + num = c['HOME_TWEET_NUM'] if g['stuff'].isdigit(): num = int(g['stuff']) for tweet in reversed(t.statuses.home_timeline(count=num)): @@ -489,7 +263,7 @@ def view(): try: num = int(g['stuff'].split()[1]) except: - num = HOME_TWEET_NUM + num = c['HOME_TWEET_NUM'] for tweet in reversed(t.statuses.user_timeline(count=num, screen_name=user[1:])): draw(t=tweet, iot=g['iot']) printNicely('') @@ -502,7 +276,7 @@ def mentions(): Mentions timeline """ t = Twitter(auth=authen()) - num = HOME_TWEET_NUM + num = c['HOME_TWEET_NUM'] if g['stuff'].isdigit(): num = int(g['stuff']) for tweet in reversed(t.statuses.mentions_timeline(count=num)): @@ -548,9 +322,9 @@ def allretweet(): try: num = int(g['stuff'].split()[1]) except: - num = RETWEETS_SHOW_NUM + num = c['RETWEETS_SHOW_NUM'] # Get result and display - rt_ary = t.statuses.retweets(id=tid,count=num) + rt_ary = t.statuses.retweets(id=tid, count=num) if not rt_ary: printNicely(magenta('This tweet has no retweet.')) return @@ -634,7 +408,7 @@ def search(): rel = t.search.tweets(q=g['stuff'])['statuses'] if rel: printNicely('Newest tweets:') - for i in reversed(xrange(SEARCH_MAX_RECORD)): + for i in reversed(xrange(c['SEARCH_MAX_RECORD'])): draw(t=rel[i], iot=g['iot'], keyword=g['stuff'].strip()[1:]) @@ -733,7 +507,7 @@ def inbox(): Inbox direct messages """ t = Twitter(auth=authen()) - num = MESSAGES_DISPLAY + num = c['MESSAGES_DISPLAY'] rel = [] if g['stuff'].isdigit(): num = g['stuff'] @@ -766,7 +540,7 @@ def sent(): Sent direct messages """ t = Twitter(auth=authen()) - num = MESSAGES_DISPLAY + num = c['MESSAGES_DISPLAY'] rel = [] if g['stuff'].isdigit(): num = int(g['stuff']) @@ -819,7 +593,7 @@ def whois(): user = t.users.show( screen_name=screen_name[1:], include_entities=False) - show_profile(user) + show_profile(user, g['iot']) except: printNicely(red('Omg no user.')) else: @@ -900,7 +674,6 @@ def muting(): """ t = Twitter(auth=authen()) # Init cursor - d = {'fl': 'followers', 'fr': 'friends'} next_cursor = -1 rel = {} # Cursor loop @@ -977,14 +750,35 @@ def cal(): month = random_rainbow(month) date = rel.pop(0) date = ' '.join([cycle_color(i) for i in date.split(' ')]) - today = os.popen('date +\'%d\'').read().strip() + today = str(int(os.popen('date +\'%d\'').read().strip())) # Display - print month - print date + printNicely(month) + printNicely(date) for line in rel: ary = line.split(' ') - ary = map(lambda x:on_grey(x) if x==today else grey(x),ary) - print ' '.join(ary) + ary = map(lambda x: on_grey(x) if x == today else grey(x), ary) + printNicely(' '.join(ary)) + + +def theme(): + """ + List and change theme + """ + if not g['stuff']: + # List themes + for theme in g['themes']: + line = ' '*2 + magenta('* ' + theme) + printNicely(line) + else: + # Change theme + try : + new_config = 'rainbowstream/colorset/' + g['stuff'] + '.json' + new_config = load_config(new_config) + for nc in new_config: + c[nc] = new_config[nc] + printNicely(green('Theme changed.')) + except : + printNicely(red('Sorry, config file is broken!')) def help(): @@ -1029,7 +823,7 @@ def help(): green('rt 12 ') + ' will retweet to tweet with ' + \ yellow('[id=12]') + '.\n' usage += s * 2 + \ - green('allrt 12 20 ') + ' list 20 newest retweet of the tweet with ' + \ + green('allrt 12 20 ') + ' will list 20 newest retweet of the tweet with ' + \ yellow('[id=12]') + '.\n' usage += s * 2 + green('rep 12 oops') + ' will reply "' + \ yellow('oops') + '" to tweet with ' + yellow('[id=12]') + '.\n' @@ -1098,19 +892,21 @@ def help(): ' filter will decide nicks will be EXCLUDE.\n' usage += s * 2 + green('switch mine -d') + \ ' will use the config\'s ONLY_LIST and IGNORE_LIST.\n' - usage += s * 3 + '(see ' + grey('rainbowstream/config.py') + ').\n' # Smart shell usage += '\n' usage += s + grey(u'\u266A' + ' Smart shell\n') - usage += s*2 + green('111111 * 9 / 7') + ' or any math expression ' + \ + usage += s * 2 + green('111111 * 9 / 7') + ' or any math expression ' + \ 'will be evaluate by Python interpreter.\n' - usage += s*2 + 'Even ' + green('cal') + ' will show the calendar' + \ + usage += s * 2 + 'Even ' + green('cal') + ' will show the calendar' + \ ' for current month.\n' # Screening usage += '\n' usage += s + grey(u'\u266A' + ' Screening \n') + usage += s * 2 + green('theme') + ' will list available theme.' + \ + green('theme monokai') + ' will apply '+ yellow('monokai') + \ + ' theme immediately.\n' usage += s * 2 + green('h') + ' will show this help again.\n' usage += s * 2 + green('c') + ' will clear the screen.\n' usage += s * 2 + green('q') + ' will quit.\n' @@ -1147,7 +943,7 @@ def reset(): printNicely(magenta('Need tips ? Type "h" and hit Enter key!')) g['reset'] = False try: - print eval(g['cmd']) + printNicely(eval(g['cmd'])) except: pass @@ -1188,6 +984,7 @@ def process(cmd): unblock, report, cal, + theme, help, clear, quit @@ -1231,6 +1028,7 @@ def listen(): ['@'], # unblock ['@'], # report [], # cal + g['themes'], # theme [], # help [], # clear [], # quit @@ -1269,9 +1067,9 @@ def stream(domain, args, name='Rainbow Stream'): # The Logo art_dict = { - USER_DOMAIN: name, - PUBLIC_DOMAIN: args.track_keywords, - SITE_DOMAIN: 'Site Stream', + c['USER_DOMAIN']: name, + c['PUBLIC_DOMAIN']: args.track_keywords, + c['SITE_DOMAIN']: 'Site Stream', } ascii_art(art_dict[domain]) @@ -1292,9 +1090,9 @@ def stream(domain, args, name='Rainbow Stream'): domain=domain, **stream_args) - if domain == USER_DOMAIN: + if domain == c['USER_DOMAIN']: tweet_iter = stream.user(**query_args) - elif domain == SITE_DOMAIN: + elif domain == c['SITE_DOMAIN']: tweet_iter = stream.site(**query_args) else: if args.track_keywords: @@ -1303,23 +1101,27 @@ def stream(domain, args, name='Rainbow Stream'): tweet_iter = stream.statuses.sample() # Iterate over the stream. - for tweet in tweet_iter: - if tweet is None: - printNicely("-- None --") - elif tweet is Timeout: - printNicely("-- Timeout --") - elif tweet is HeartbeatTimeout: - printNicely("-- Heartbeat Timeout --") - elif tweet is Hangup: - printNicely("-- Hangup --") - elif tweet.get('text'): - draw( - t=tweet, - iot=args.image_on_term, - keyword=args.track_keywords, - fil=args.filter, - ig=args.ignore, - ) + try: + for tweet in tweet_iter: + if tweet is None: + printNicely("-- None --") + elif tweet is Timeout: + printNicely("-- Timeout --") + elif tweet is HeartbeatTimeout: + printNicely("-- Heartbeat Timeout --") + elif tweet is Hangup: + printNicely("-- Hangup --") + elif tweet.get('text'): + draw( + t=tweet, + iot=args.image_on_term, + keyword=args.track_keywords, + fil=args.filter, + ig=args.ignore, + ) + except: + printNicely(magenta("I'm afraid we have problem with twitter'S maximum connection.")) + printNicely(magenta("Let's try again later.")) def fly(): @@ -1329,7 +1131,7 @@ def fly(): # Spawn stream process args = parse_arguments() get_decorated_name() - p = Process(target=stream, args=(USER_DOMAIN, args, g['original_name'])) + p = Process(target=stream, args=(c['USER_DOMAIN'], args, g['original_name'])) p.start() # Start listen process