forked from ticketfrei/ticketfrei
91 lines
3.2 KiB
Python
Executable file
91 lines
3.2 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import logging
|
|
import tweepy
|
|
import re
|
|
import requests
|
|
from time import time
|
|
import report
|
|
from bot import Bot
|
|
import tfglobals
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class TwitterBot(Bot):
|
|
|
|
def get_api(self, user):
|
|
keys = user.get_twitter_credentials()
|
|
auth = tweepy.OAuthHandler(consumer_key=keys[0],
|
|
consumer_secret=keys[1])
|
|
auth.set_access_token(keys[2], # access_token_key
|
|
keys[3]) # access_token_secret
|
|
return tweepy.API(auth, wait_on_rate_limit=True)
|
|
|
|
def crawl(self, user):
|
|
"""
|
|
crawls all Tweets which mention the bot from the twitter rest API.
|
|
|
|
:return: reports: (list of report.Report objects)
|
|
"""
|
|
reports = []
|
|
#global last_twitter_request
|
|
if tfglobals.last_twitter_request + 60 > time():
|
|
return reports
|
|
try:
|
|
api = self.get_api(user)
|
|
except TypeError:
|
|
# When there is no twitter account for this bot, we want to
|
|
# seamlessly continue.
|
|
#logger.error("Error Authenticating Twitter", exc_info=True)
|
|
return reports
|
|
last_mention = user.get_seen_tweet()
|
|
try:
|
|
if last_mention == 0:
|
|
mentions = api.mentions_timeline()
|
|
else:
|
|
mentions = api.mentions_timeline(since_id=last_mention)
|
|
tfglobals.last_twitter_request = time()
|
|
for status in mentions:
|
|
text = re.sub(
|
|
"(?<=^|(?<=[^a-zA-Z0-9-_\.]))@([A-Za-z]+[A-Za-z0-9-_]+)",
|
|
"", status.text)
|
|
username = "@" + api.me().screen_name
|
|
if username in status.text:
|
|
reports.append(report.Report(status.author.screen_name,
|
|
self,
|
|
text,
|
|
status.id,
|
|
status.created_at))
|
|
user.save_seen_tweet(status.id)
|
|
return reports
|
|
except tweepy.RateLimitError:
|
|
logger.error("Twitter API Error: Rate Limit Exceeded",
|
|
exc_info=True)
|
|
# :todo implement rate limiting
|
|
except requests.exceptions.ConnectionError:
|
|
logger.error("Twitter API Error: Bad Connection", exc_info=True)
|
|
except tweepy.TweepError:
|
|
logger.error("Twitter API Error: General Error", exc_info=True)
|
|
return []
|
|
|
|
def post(self, user, report):
|
|
try:
|
|
api = self.get_api(user)
|
|
except IndexError:
|
|
return # no twitter account for this user.
|
|
try:
|
|
if report.source == self:
|
|
api.retweet(report.id)
|
|
else:
|
|
text = report.text
|
|
if len(text) > 280:
|
|
text = text[:280 - 4] + u' ...'
|
|
api.update_status(status=text)
|
|
except requests.exceptions.ConnectionError:
|
|
logger.error("Twitter API Error: Bad Connection",
|
|
exc_info=True)
|
|
except tweepy.error.TweepError:
|
|
logger.error("Twitter API Error", exc_info=True)
|