twitter_bot.py

#!/usr/bin/env python
# pylint: disable=C0103

"""Twitter Bot. Listens for mentions and replies to them."""

#
# IMPORTS
#

# Allow using print as a function with parenthesis: print()
from __future__ import print_function
# basic operating system interactions
import os
import sys
# import the code that connects to Twitter
from twython import Twython, TwythonError
# import all functions from tweet_text.py
from tweet_text import *
# import all functions from helper.py
from helper import *

# Try to import the variables defined in credentials.py
# If that does not exist (e.g. on Heroku), fall back to environment variables
try:
    from credentials import CONSUMER_KEY, CONSUMER_SECRET, ACCESS_TOKEN, ACCESS_TOKEN_SECRET
except ImportError as error:
    print('Info: {e}'.format(e=error))
    print('Info: Cannot load credentials.py. Will use environment variables.')
    try:
        CONSUMER_KEY = os.environ['CONSUMER_KEY']
        CONSUMER_SECRET = os.environ['CONSUMER_SECRET']
        ACCESS_TOKEN = os.environ['ACCESS_TOKEN']
        ACCESS_TOKEN_SECRET = os.environ['ACCESS_TOKEN_SECRET']
    except KeyError as error:
        print('Error: {e} not found in environment variables'.format(e=error))
        print('Error: Could not retrieve credentials from either credentials.py or environment variables. Make sure either is set.')
        # can't do anything without credentials, so quit
        sys.exit()


#
# BOT CODE
#

INTERVAL_MINUTES = 10

def setup():
    # Login to Twitter
    account = Twython(CONSUMER_KEY, CONSUMER_SECRET, ACCESS_TOKEN, ACCESS_TOKEN_SECRET)

    # Check the supplied credentials, get some general info on the account
    # https://dev.twitter.com/rest/reference/get/account/verify_credentials
    info = account.verify_credentials(include_entities=False, skip_status=True, include_email=False)
    print('user:', info['screen_name'])
    print('tweet count:', info['statuses_count'])
    print('favourite count:', info['favourites_count'])
    print('friends count:', info['friends_count'])
    return account

def tweet(account):
    """check for mentions and answer, otherwise tweet idle tweet"""
    replied = False
    mentions = account.get_mentions_timeline()
    rate_limit_remaining = account.get_lastfunction_header('x-rate-limit-remaining')
    print('rate limit remaining', rate_limit_remaining)
    # for each mention
    for tweet in mentions:
        # if the tweet was sent after the last time we checked mentions
        if tweet_minutes_ago(tweet) < INTERVAL_MINUTES:
            # get reply from tweet_text.py
            reply_text = reply(tweet)
            if reply_text is not None:
                replied = True
                try:
                    print('Replying to https://twitter.com/statuses/{id}'.format(id=tweet['id']))
                    sent_tweet = account.update_status(status=reply_text, in_reply_to_status_id=tweet['id'])
                    print('https://twitter.com/statuses/{id}'.format(id=sent_tweet['id']))
                except TwythonError as e:
                    print(e)
    if not replied:
        # from tweet_text.py
        text = idle_text()
        # Send the tweet!
        tweet = account.update_status(status=text)
        # Print some info on the sent tweet
        print('https://twitter.com/statuses/{id}'.format(id=tweet['id']))

if __name__ == "__main__":
    account = setup()
    tweet(account)