81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
import os
|
|
from uuid import uuid4
|
|
from requests_oauthlib import OAuth1Session
|
|
from flask import Blueprint, Response, current_app, request, jsonify
|
|
from ayct_backend.twitter.models import *
|
|
|
|
twitter = Blueprint('twitter', __name__)
|
|
|
|
@twitter.route('/account', methods=['GET'])
|
|
def get_twitter_accounts():
|
|
twitter_accounts = TwitterAccount.query.all()
|
|
|
|
accounts = []
|
|
|
|
for account in twitter_accounts:
|
|
account_data = {}
|
|
account_data['twitter_user_id'] = account.twitter_user_id
|
|
account_data['username'] = account.username
|
|
accounts.append(account_data)
|
|
|
|
return Response(
|
|
jsonify({
|
|
"twitter_accounts": accounts
|
|
}),
|
|
status=200,
|
|
)
|
|
|
|
@twitter.route('/account', methods=['POST'])
|
|
def add_twitter_account():
|
|
content_type = request.headers.get('Content-Type')
|
|
if (content_type == 'application/json'):
|
|
json = request.json
|
|
if not json['veryfier']:
|
|
return Response(
|
|
"Invalid request!",
|
|
status=400,
|
|
)
|
|
else:
|
|
consumer_key = current_app.config["TWITTER_CONSUMER_KEY"]
|
|
consumer_secret = current_app.config["TWITTER_CONSUMER_SECERT"]
|
|
|
|
oauth = OAuth1Session(
|
|
consumer_key,
|
|
client_secret=consumer_secret
|
|
)
|
|
request_token_url = "https://api.twitter.com/oauth/request_token?oauth_callback=oob&x_auth_access_type=write"
|
|
fetch_response = oauth.fetch_request_token(request_token_url)
|
|
|
|
access_token_url = "https://api.twitter.com/oauth/access_token"
|
|
oauth = OAuth1Session(
|
|
consumer_key,
|
|
client_secret=consumer_secret,
|
|
resource_owner_key=fetch_response.get("oauth_token"),
|
|
resource_owner_secret=fetch_response.get("oauth_token_secret"),
|
|
verifier=json['veryfier'],
|
|
)
|
|
oauth_tokens = oauth.fetch_access_token(access_token_url)
|
|
|
|
new_twitter_account = TwitterAccount(
|
|
twitter_user_id = oauth_tokens["user_id"],
|
|
username = oauth_tokens["screen_name"],
|
|
access_token = oauth_tokens["oauth_token"],
|
|
access_token_secret = oauth_tokens["oauth_token_secret"]
|
|
)
|
|
|
|
twitter_db.session.add(new_twitter_account)
|
|
twitter_db.session.commit()
|
|
|
|
return Response(
|
|
jsonify({
|
|
"message": "Twitter account succesfully added."
|
|
}),
|
|
status=201,
|
|
)
|
|
else:
|
|
return Response(
|
|
"Content-type not supported!",
|
|
status=400,
|
|
)
|
|
|