def main():
for user in getCollUsers().find():
if "access_token" in user and "twitter_credentials" in user:
#if user["screen_name"]=="rabshakeh":
print(user["screen_name"])
access_token = dict(urllib.parse.parse_qsl(user["access_token"]))
twitter = OAuthApi(consumer_key, consumer_secret,
access_token["oauth_token"],
access_token["oauth_token_secret"])
#user["twitter_credentials"] = twitter.VerifyCredentials()
user_timeline = twitter.GetHomeTimeline({"count": 5})
ut = getCollUnprocessedTweets()
for tweet in user_timeline:
tweet["inprocess"] = 0
tweet["for_user"] = int(user["id_str"])
tweet["created_at_utc"] = str(tweet["created_at"])
tweet["created_at"] = parseDatetime(tweet["created_at"])
tweet["newsrivr_userid_md5"] = [user["newsrivr_userid_md5"]]
insertTweet(tweet["id_str"], user["newsrivr_userid_md5"],
tweet)
friends = twitter.GetFriendsIDs()
friend_list = []
for fp in friends:
friend_list.append(fp)
user["twitter_friend_list"] = friend_list
getCollUsers().save(user)
curses.init_pair(3, curses.COLOR_MAGENTA, -1) # whitespace
curses.init_pair(4, curses.COLOR_GREEN, -1) # whitespace
last_check = 0
lines = []
addTweetsToLines(tweets, lines, maxNameLength)
tweet_ids = map(lambda x: x['id_str'], tweets)
cursor = len(lines) - 1 if lines else 0
link_cursor = 0
while working:
# Load new tweets
new_tweets = []
if time() - last_check >= 60 * 3:
timeline = twitter.GetHomeTimeline()
timeline.reverse()
for t in timeline:
if not t['id_str'] in tweet_ids:
t['__mst_readed__'] = False
new_tweets.append(t)
tweets.append(t)
tweet_ids.append(t['id_str'])
last_check = time()
# Build lines
if len(new_tweets) > 0:
maxNameLength = reduce(
lambda acc, i: len(i['user']['screen_name'])
if len(i['user']['screen_name']) > acc else acc, new_tweets,
class CliTwi(object):
consumer_key = config.consumer_key
consumer_secret = config.consumer_secret
cli_twi_path = os.path.expanduser('~') + '/.clitwi'
token = None
secret = None
config_ok = False
def __init__(self):
os.chdir(self.cli_twi_path)
# Figure out the action
self.read_action()
# Verify we have all of the necessary config files
self.check_config()
self.read_config()
# And the power button...
self.run()
def read_action(self):
l = len(sys.argv)
if l == 1:
self.action = 'list'
elif l == 2:
arg = sys.argv[1]
if arg == 'setup':
self.action = 'setup'
elif arg == '-m':
self.action = 'mentions'
elif arg == '--help':
self.action = 'help'
else:
self.action = 'update'
def check_config(self):
self.config_ok = True
if not os.path.isfile('_config'):
print 'Missing _config'
self.config_ok = False
def read_config(self):
if self.action == 'setup' or self.action == 'help':
return
if not self.config_ok:
return
# Read the config file
file = open('_config', 'r')
v = file.read()
config = json.loads(v)
file.close()
self.latest = config['latest']
self.mention = config['mention']
self.token = config['token']
self.secret = config['secret']
def setup(self):
os.system('clear')
twitter = OAuthApi(self.consumer_key, self.consumer_secret)
# Get the temporary credentials for our next few calls
temp_credentials = twitter.getRequestToken()
# User pastes this into their browser to bring back a pin number
print(twitter.getAuthorizationURL(temp_credentials))
# Get the pin # from the user and get our permanent credentials
oauth_verifier = raw_input('What is the PIN? ')
access_token = twitter.getAccessToken(temp_credentials, oauth_verifier)
self.token = access_token['oauth_token']
self.secret = access_token['oauth_token_secret']
self.latest = 1
self.mention = 1
print 'Clitwi was successfully set up.'
def setup_api(self):
self.twitter = OAuthApi(self.consumer_key,
self.consumer_secret,
self.token,
self.secret)
def list_tweets(self, type='default'):
os.system('clear')
if type == 'default':
user_timeline = self.twitter.GetHomeTimeline({
'since_id': self.latest
})
else:
user_timeline = self.twitter.GetMentions({
'since_id': self.mention
})
if user_timeline == []:
print 'No new tweets.'
return
first = True
for tweet in user_timeline:
print tweet['user']['screen_name'] + ':'
self.print_text(tweet['text'])
if first:
if type == 'default':
self.latest = tweet['id']
else:
self.mention = tweet['id']
first = False
def print_text(self, text):
l = len(text)
if l > 77:
print ' ' + text[:77]
print ' ' + text[77:]
else:
print ' ' + text
def send_tweet(self):
tweet = sys.argv[1]
if len(tweet) > 140:
print 'Too long'
sys.exit(1)
return
else:
self.twitter.UpdateStatus(str(tweet))
print 'Sent!'
def show_help(self):
print 'Cli Twi - version %s' % __version__
print 'main.py [options]'
print 'main.py - will print latest tweets'
print 'main.py "Some message" - will update your status'
print 'main.py -m - will print latest mentions'
print 'main.py setup - will run the OAuth process'
print 'main.py --help - will display this help'
def run(self):
# Actions that don't require any config (help, setup)
if self.action == 'setup':
self.setup()
elif self.action == 'help':
self.show_help()
# Check if we have correct config settings
if not self.config_ok:
return
self.setup_api()
if self.action == 'list':
self.list_tweets()
elif self.action == 'update':
self.send_tweet()
elif self.action == 'mentions':
self.list_tweets('mentions')
def finish(self):
d = {
'token': self.token,
'secret': self.secret,
'latest': self.latest,
'mention': self.mention
}
f = open(os.getcwd() + '/_config', 'w')
f.write(json.dumps(d))
f.close()
