231 lines
7.2 KiB
Python
231 lines
7.2 KiB
Python
from twitter import Twitter, OAuth, TwitterHTTPError, TwitterError
|
|
from werkzeug.urls import url_decode
|
|
from model import OAuthToken, Account, Post, TwitterArchive
|
|
from app import db, app, sentry
|
|
from math import inf
|
|
from datetime import datetime
|
|
import locale
|
|
from zipfile import ZipFile
|
|
from io import BytesIO
|
|
from libforget.exceptions import PermanentError, TemporaryError
|
|
from urllib.error import URLError
|
|
|
|
|
|
def get_login_url(callback='oob', consumer_key=None, consumer_secret=None):
|
|
twitter = Twitter(
|
|
auth=OAuth('', '', consumer_key, consumer_secret),
|
|
format='', api_version=None)
|
|
resp = url_decode(twitter.oauth.request_token(oauth_callback=callback))
|
|
oauth_token = resp['oauth_token']
|
|
oauth_token_secret = resp['oauth_token_secret']
|
|
|
|
token = OAuthToken(token=oauth_token, token_secret=oauth_token_secret)
|
|
db.session.merge(token)
|
|
db.session.commit()
|
|
|
|
return (
|
|
"https://api.twitter.com/oauth/authenticate?oauth_token=%s"
|
|
% (oauth_token,))
|
|
|
|
|
|
def account_from_api_user_object(obj):
|
|
return Account(
|
|
twitter_id=obj['id_str'],
|
|
display_name=obj['name'],
|
|
screen_name=obj['screen_name'],
|
|
avatar_url=obj['profile_image_url_https'],
|
|
reported_post_count=obj['statuses_count'])
|
|
|
|
|
|
def receive_verifier(oauth_token, oauth_verifier,
|
|
consumer_key=None, consumer_secret=None):
|
|
temp_token = OAuthToken.query.get(oauth_token)
|
|
if not temp_token:
|
|
raise Exception("OAuth token has expired")
|
|
twitter = Twitter(
|
|
auth=OAuth(temp_token.token, temp_token.token_secret,
|
|
consumer_key, consumer_secret),
|
|
format='', api_version=None)
|
|
resp = url_decode(
|
|
twitter.oauth.access_token(oauth_verifier=oauth_verifier))
|
|
db.session.delete(temp_token)
|
|
new_token = OAuthToken(token=resp['oauth_token'],
|
|
token_secret=resp['oauth_token_secret'])
|
|
new_token = db.session.merge(new_token)
|
|
new_twitter = Twitter(
|
|
auth=OAuth(new_token.token, new_token.token_secret,
|
|
consumer_key, consumer_secret))
|
|
remote_acct = new_twitter.account.verify_credentials()
|
|
acct = account_from_api_user_object(remote_acct)
|
|
acct = db.session.merge(acct)
|
|
|
|
new_token.account = acct
|
|
db.session.commit()
|
|
|
|
return new_token
|
|
|
|
|
|
def get_twitter_for_acc(account):
|
|
consumer_key = app.config['TWITTER_CONSUMER_KEY']
|
|
consumer_secret = app.config['TWITTER_CONSUMER_SECRET']
|
|
|
|
tokens = (OAuthToken.query.with_parent(account)
|
|
.order_by(db.desc(OAuthToken.created_at)).all())
|
|
for token in tokens:
|
|
t = Twitter(
|
|
auth=OAuth(token.token, token.token_secret,
|
|
consumer_key, consumer_secret))
|
|
try:
|
|
t.account.verify_credentials()
|
|
return t
|
|
except TwitterHTTPError as e:
|
|
if e.e.code == 401:
|
|
# token revoked
|
|
|
|
if sentry:
|
|
sentry.captureMessage(
|
|
'Twitter auth revoked', extra=locals())
|
|
db.session.delete(token)
|
|
db.session.commit()
|
|
else:
|
|
raise TemporaryError(e)
|
|
except URLError as e:
|
|
raise TemporaryError(e)
|
|
|
|
return None
|
|
|
|
|
|
locale.setlocale(locale.LC_TIME, 'C')
|
|
|
|
|
|
def post_from_api_tweet_object(tweet, post=None):
|
|
if not post:
|
|
post = Post()
|
|
post.twitter_id = tweet['id_str']
|
|
try:
|
|
post.created_at = datetime.strptime(
|
|
tweet['created_at'], '%a %b %d %H:%M:%S %z %Y')
|
|
except ValueError:
|
|
post.created_at = datetime.strptime(
|
|
tweet['created_at'], '%Y-%m-%d %H:%M:%S %z')
|
|
# whyyy
|
|
post.author_id = 'twitter:{}'.format(tweet['user']['id_str'])
|
|
if 'favorited' in tweet:
|
|
post.favourite = tweet['favorited']
|
|
if 'entities' in tweet:
|
|
post.has_media = bool(
|
|
'media' in tweet['entities'] and tweet['entities']['media'])
|
|
if 'favorite_count' in tweet:
|
|
post.favourites = tweet['favorite_count']
|
|
if 'retweet_count' in tweet:
|
|
post.reblogs = tweet['retweet_count']
|
|
post.is_reblog = 'retweeted_status' in tweet
|
|
return post
|
|
|
|
|
|
def fetch_acc(account, cursor):
|
|
t = get_twitter_for_acc(account)
|
|
if not t:
|
|
print("no twitter access, aborting")
|
|
return
|
|
|
|
try:
|
|
user = t.account.verify_credentials()
|
|
db.session.merge(account_from_api_user_object(user))
|
|
|
|
kwargs = {
|
|
'user_id': account.twitter_id,
|
|
'count': 200,
|
|
'trim_user': True,
|
|
'tweet_mode': 'extended',
|
|
}
|
|
if cursor:
|
|
kwargs.update(cursor)
|
|
|
|
if 'max_id' not in kwargs:
|
|
most_recent_post = (
|
|
Post.query.order_by(db.desc(Post.created_at))
|
|
.filter(Post.author_id == account.id).first())
|
|
if most_recent_post:
|
|
kwargs['since_id'] = most_recent_post.twitter_id
|
|
|
|
tweets = t.statuses.user_timeline(**kwargs)
|
|
except (TwitterError, URLError) as e:
|
|
handle_error(e)
|
|
|
|
print("processing {} tweets for {acc}".format(len(tweets), acc=account))
|
|
|
|
if len(tweets) > 0:
|
|
|
|
kwargs['max_id'] = +inf
|
|
|
|
for tweet in tweets:
|
|
db.session.merge(post_from_api_tweet_object(tweet))
|
|
kwargs['max_id'] = min(tweet['id'] - 1, kwargs['max_id'])
|
|
|
|
else:
|
|
kwargs = None
|
|
|
|
db.session.commit()
|
|
|
|
return kwargs
|
|
|
|
|
|
def refresh_posts(posts):
|
|
if not posts:
|
|
return posts
|
|
|
|
t = get_twitter_for_acc(posts[0].author)
|
|
if not t:
|
|
return
|
|
try:
|
|
tweets = t.statuses.lookup(
|
|
_id=",".join((post.twitter_id for post in posts)),
|
|
trim_user=True, tweet_mode='extended')
|
|
except (URLError, TwitterError) as e:
|
|
handle_error(e)
|
|
refreshed_posts = list()
|
|
for post in posts:
|
|
tweet = next(
|
|
(tweet for tweet in tweets if tweet['id_str'] == post.twitter_id),
|
|
None)
|
|
if not tweet:
|
|
db.session.delete(post)
|
|
else:
|
|
post = db.session.merge(post_from_api_tweet_object(tweet))
|
|
refreshed_posts.append(post)
|
|
|
|
return refreshed_posts
|
|
|
|
|
|
def delete(post):
|
|
t = get_twitter_for_acc(post.author)
|
|
t.statuses.destroy(id=post.twitter_id)
|
|
db.session.delete(post)
|
|
|
|
|
|
def chunk_twitter_archive(archive_id):
|
|
ta = TwitterArchive.query.get(archive_id)
|
|
|
|
with ZipFile(BytesIO(ta.body), 'r') as zipfile:
|
|
files = [filename for filename in zipfile.namelist()
|
|
if filename.startswith('data/js/tweets/')
|
|
and filename.endswith('.js')]
|
|
|
|
files.sort()
|
|
|
|
return files
|
|
|
|
|
|
def handle_error(e):
|
|
if isinstance(e, TwitterHTTPError):
|
|
data = e.response_data
|
|
if isinstance(data, dict) and 'errors' in data.keys():
|
|
for error in data['errors']:
|
|
if error.get('code',0) == 326:
|
|
# account locked lol rip
|
|
# although this is a temporary error in twitter terms
|
|
# it's best not to waste api calls on locked accounts
|
|
raise PermanentError(e)
|
|
raise TemporaryError(e)
|