import pkg_resources
import socks
import socket
+import re
from io import BytesIO
from twitter.stream import TwitterStream, Timeout, HeartbeatTimeout, Hangup
from twitter.oauth_dance import oauth_dance
from twitter.util import printNicely
+from pocket import Pocket
+
from .draw import *
from .colors import *
from .config import *
parser.add_argument(
'-s',
'--stream',
- default="mine",
+ default="mine",
help='Default stream after program start. (Default: mine)')
parser.add_argument(
'-to',
CONSUMER_SECRET)
+def pckt_authen():
+ """
+ Authenticate with Pocket OAuth
+ """
+ pocket_credential = os.environ.get(
+ 'HOME',
+ os.environ.get(
+ 'USERPROFILE',
+ '')) + os.sep + '.rainbow_pckt_oauth'
+
+ if not os.path.exists(pocket_credential):
+ request_token = Pocket.get_request_token(consumer_key=PCKT_CONSUMER_KEY)
+ auth_url = Pocket.get_auth_url(code=request_token, redirect_uri="/")
+ webbrowser.open(auth_url)
+ printNicely(green("*** Press [ENTER] after authorization ***"))
+ raw_input()
+ user_credentials = Pocket.get_credentials(consumer_key=PCKT_CONSUMER_KEY, code=request_token)
+ access_token = user_credentials['access_token']
+ f = open(pocket_credential, 'w')
+ f.write(access_token)
+ f.close()
+ else:
+ with open(pocket_credential, 'r') as f:
+ access_token = str(f.readlines()[0])
+ f.close()
+
+ return Pocket(PCKT_CONSUMER_KEY, access_token)
+
+
def build_mute_dict(dict_data=False):
"""
Build muting list
notice += light_yellow(current) + '\n'
notice += light_magenta('You should upgrade with ')
notice += light_green('pip install -U rainbowstream')
- printNicely(notice)
else:
notice = light_yellow('You are running latest version (')
notice += light_green(current)
notice += light_yellow(')')
notice += '\n'
+ printNicely(notice)
except:
pass
c['message_dict'] = []
# Image on term
c['IMAGE_ON_TERM'] = args.image_on_term
- set_config('IMAGE_ON_TERM', str(c['IMAGE_ON_TERM']))
# Use 24 bit color
c['24BIT'] = args.color_24bit
# Check type of ONLY_LIST and IGNORE_LIST
c['IGNORE_LIST'] = []
# Mute dict
c['IGNORE_LIST'] += build_mute_dict()
+ # Pocket init
+ pckt = pckt_authen() if c['POCKET_SUPPORT'] else None
def trend():
printNicely(red('A name should begin with a \'@\''))
+def view_my_tweets():
+ """
+ Display user's recent tweets.
+ """
+ t = Twitter(auth=authen())
+ try:
+ num = int(g['stuff'])
+ except:
+ num = c['HOME_TWEET_NUM']
+ for tweet in reversed(
+ t.statuses.user_timeline(count=num, screen_name=g['original_name'])):
+ draw(t=tweet)
+ printNicely('')
+
+
def search():
"""
Search
t.statuses.update(status=g['stuff'])
+def pocket():
+ """
+ Add new link to Pocket along with tweet id
+ """
+ if not c['POCKET_SUPPORT']:
+ printNicely(yellow('Pocket isn\'t enabled.'))
+ printNicely(yellow('You need to "config POCKET_SUPPORT = true"'))
+ return
+
+ # Get tweet infos
+ p = pckt_authen()
+
+ t = Twitter(auth=authen())
+ try:
+ id = int(g['stuff'].split()[0])
+ tid = c['tweet_dict'][id]
+ except:
+ printNicely(red('Sorry I can\'t understand.'))
+ return
+
+ tweet = t.statuses.show(id=tid)
+
+ if len(tweet['entities']['urls']) > 0:
+ url = tweet['entities']['urls'][0]['expanded_url']
+ else:
+ url = "https://twitter.com/" + \
+ tweet['user']['screen_name'] + '/status/' + str(tid)
+
+ # Add link to pocket
+ try:
+ p.add(title=re.sub(r'(http:\/\/\S+)', r'', tweet['text']),
+ url=url,
+ tweet_id=tid)
+ except:
+ printNicely(red('Something is wrong about your Pocket account,'+ \
+ ' please restart Rainbowstream.'))
+ pocket_credential = os.environ.get(
+ 'HOME',
+ os.environ.get(
+ 'USERPROFILE',
+ '')) + os.sep + '.rainbow_pckt_oauth'
+ if os.path.exists(pocket_credential):
+ os.remove(pocket_credential)
+ return
+
+ printNicely(green('Pocketed !'))
+ printNicely('')
+
+
def retweet():
"""
ReTweet
return
tid = c['tweet_dict'][int(g['stuff'])]
tweet = t.statuses.show(id=tid)
- link_prefix = ('http://', 'https://')
- link_ary = [u for u in tweet['text'].split()
- if u.startswith(link_prefix)]
- if not link_ary:
+ urls = tweet['entities']['urls']
+ if not urls:
printNicely(light_magenta('No url here @.@!'))
return
- for link in link_ary:
- webbrowser.open(link)
+ else:
+ for url in urls:
+ expanded_url = url['expanded_url']
+ webbrowser.open(expanded_url)
except:
debug_option()
printNicely(red('Sorry I can\'t open url in this tweet.'))
rel = t.mutes.users.create(screen_name=screen_name[1:])
if isinstance(rel, dict):
printNicely(green(screen_name + ' is muted.'))
- c['IGNORE_LIST'] += [unc(screen_name)]
+ c['IGNORE_LIST'] += [screen_name]
c['IGNORE_LIST'] = list(set(c['IGNORE_LIST']))
else:
printNicely(red(rel))
printNicely(light_magenta(guide))
only = raw_input('Only nicks [Ex: @xxx,@yy]: ')
ignore = raw_input('Ignore nicks [Ex: @xxx,@yy]: ')
- args.filter = filter(None, only.split(','))
- args.ignore = filter(None, ignore.split(','))
+ args.filter = list(filter(None, only.split(',')))
+ args.ignore = list(filter(None, ignore.split(',')))
except:
printNicely(red('Sorry, wrong format.'))
return
debug_option()
printNicely(red('Just can not set the key.'))
else:
- printNicely(light_magenta('Sorry I can\'s understand.'))
+ printNicely(light_magenta('Sorry I can\'t understand.'))
def help_discover():
light_green('trend JP Tokyo') + '.\n'
usage += s * 2 + light_green('home') + ' will show your timeline. ' + \
light_green('home 7') + ' will show 7 tweets.\n'
+ usage += s * 2 + light_green('me') + ' will show your latest tweets. ' + \
+ light_green('me 2') + ' will show your last 2 tweets.\n'
usage += s * 2 + \
light_green('notification') + ' will show your recent notification.\n'
usage += s * 2 + light_green('mentions') + ' will show mentions timeline. ' + \
light_yellow('[id=12]') + ' in your OS\'s image viewer.\n'
usage += s * 2 + light_green('open 12') + ' will open url in tweet with ' + \
light_yellow('[id=12]') + ' in your OS\'s default browser.\n'
+ usage += s * 2 + light_green('pt 12') + ' will add tweet with ' + \
+ light_yellow('[id=12]') + ' in your Pocket list.\n'
printNicely(usage)
sys.exit()
-def changelog_notify():
- # For v1.2.8. Hardcoded here but will improve later
- notice = light_yellow('Hey! RS just ')
- notice += light_green('doubled ')
- notice += light_yellow('pixels for higher image resolution. Upgrade and try')
- notice += light_green(' -iot ')
- notice += light_yellow('and you will like it for sure :)')
- printNicely(notice)
-
-
def reset():
"""
Reset prefix of line
printNicely(red('Your ~/.rainbow_config.json is messed up:'))
printNicely(red('>>> ' + c['USER_JSON_ERROR']))
printNicely('')
- if not g['using_latest']:
- changelog_notify()
printNicely(magenta('Need tips ? Type "h" and hit Enter key!'))
g['reset'] = False
try:
't',
'rt',
'quote',
+ 'me',
'allrt',
'conversation',
'fav',
'c',
'v',
'q',
+ 'pt',
]
# Handle function set
tweet,
retweet,
quote,
+ view_my_tweets,
allretweet,
conversation,
favorite,
clear,
upgrade_center,
quit,
+ pocket,
]
[], # tweet
[], # retweet
[], # quote
+ [], # view_my_tweets
[], # allretweet
[], # conversation
[], # favorite
[], # clear
[], # version
[], # quit
+ [], # pocket
]
))
init_interactive_shell(d)
# Save cmd to global variable and call process
g['stuff'] = ' '.join(line.split()[1:])
# Check tweet length
- if check_tweet_length():
- # Process the command
- process(cmd)()
- # Not re-display
- if cmd in ['switch', 't', 'rt', 'rep']:
- g['prefix'] = False
- else:
- g['prefix'] = True
- # Release the semaphore lock
- c['lock'] = False
+ # Process the command
+ process(cmd)()
+ # Not re-display
+ if cmd in ['switch', 't', 'rt', 'rep']:
+ g['prefix'] = False
+ else:
+ g['prefix'] = True
except EOFError:
printNicely('')
+ except TwitterHTTPError as e:
+ detail_twitter_error(e)
except Exception:
debug_option()
- printNicely(red('OMG something is wrong with Twitter right now.'))
-
-
-def check_tweet_length():
- """
- Check tweet length (should be <= 140 chars)
- """
- length = len(g['stuff'])
- if length <= 140:
- return True
-
- printNicely(red("Message is too long: %s chars" % length))
- return False
+ printNicely(red('OMG something is wrong with Twitter API right now.'))
+ finally:
+ # Release the semaphore lock
+ c['lock'] = False
def reconn_notice():
# Spawn stream thread
target = args.stream.split()[0]
- if target == 'mine' :
+ if target == 'mine':
spawn_personal_stream(args)
else:
try: