209 lines
6.7 KiB
Python
209 lines
6.7 KiB
Python
import requests
|
|
import json
|
|
from uuid import uuid4
|
|
from requests_oauthlib import OAuth1Session
|
|
from flask import Blueprint, current_app, request, jsonify
|
|
from nltk import sent_tokenize
|
|
from ayct_backend.models import *
|
|
from ayct_backend.firebase import verify_token
|
|
|
|
campaign = Blueprint('campaign', __name__)
|
|
|
|
@campaign.route('/', methods=['GET'])
|
|
def get_twitter_campaigns():
|
|
decoded_token = verify_token(request.headers)
|
|
if not decoded_token:
|
|
return "Not authorised!", 401
|
|
|
|
user_id = decoded_token['sub']
|
|
|
|
twitter_campaigns = TwitterCampaign.query.filter_by(user_id=user_id).all()
|
|
|
|
campaigns = []
|
|
|
|
for campaign in twitter_campaigns:
|
|
campaign_data = {}
|
|
campaign_data['campaign_id'] = campaign.id
|
|
campaign_data['campaign_name'] = campaign.campaign_name
|
|
campaign_data['twitter_account_id'] = campaign.twitter_account_id
|
|
campaigns.append(campaign_data)
|
|
|
|
return jsonify({
|
|
"twitter_campaigns": campaigns
|
|
}), 200
|
|
|
|
@campaign.route('/', methods=['POST'])
|
|
def add_twitter_campaign():
|
|
decoded_token = verify_token(request.headers)
|
|
if not decoded_token:
|
|
return "Not authorised!", 401
|
|
|
|
user_id = decoded_token['sub']
|
|
|
|
content_type = request.headers.get('Content-Type')
|
|
if content_type != 'application/json':
|
|
return "Content-type not supported!", 400
|
|
|
|
request_json = request.json
|
|
if 'campaign_name' not in request_json or 'twitter_account_id' not in request_json:
|
|
return "Invalid request!", 400
|
|
|
|
# save campaign to database
|
|
twitter_campaign = TwitterCampaign(
|
|
id = str(uuid4()),
|
|
user_id = user_id,
|
|
campaign_name = request_json['campaign_name'],
|
|
twitter_account_id = request_json['twitter_account_id'],
|
|
)
|
|
|
|
db.session.add(twitter_campaign)
|
|
db.session.commit()
|
|
|
|
return "Campaign succesfully created.", 201
|
|
|
|
@campaign.route('/', methods=['DELETE'])
|
|
def delete_twitter_campaign():
|
|
decoded_token = verify_token(request.headers)
|
|
if not decoded_token:
|
|
return "Not authorised!", 401
|
|
|
|
user_id = decoded_token['sub']
|
|
|
|
content_type = request.headers.get('Content-Type')
|
|
if content_type != 'application/json':
|
|
return "Content-type not supported!", 400
|
|
|
|
request_json = request.json
|
|
if 'campaign_id' not in request_json:
|
|
return "Invalid request!", 400
|
|
|
|
twitter_campaign = TwitterCampaign.query.filter_by(user_id=user_id, id=request_json['campaign_id']).first()
|
|
campaign_posts = TwitterPost.query.filter_by(campaign_id=request_json['campaign_id'])
|
|
|
|
if not twitter_campaign:
|
|
return "Capmaign not found!", 404
|
|
|
|
db.session.delete(campaign_posts)
|
|
db.session.delete(twitter_campaign)
|
|
db.session.commit()
|
|
|
|
return "Twitter campaign succesfully deleted.", 200
|
|
|
|
@campaign.route('/<campaign_id>', methods=['GET'])
|
|
def get_twitter_campaign_details(campaign_id):
|
|
decoded_token = verify_token(request.headers)
|
|
if not decoded_token:
|
|
return "Not authorised!", 401
|
|
|
|
user_id = decoded_token['sub']
|
|
|
|
twitter_campaign = TwitterCampaign.query.filter_by(user_id=user_id, id=campaign_id).first()
|
|
campaign_posts = TwitterPost.query.filter_by(user_id=user_id, campaign_id=campaign_id).all()
|
|
|
|
posts = []
|
|
for post in campaign_posts:
|
|
post_data = {}
|
|
post_data['post_id'] = post.id
|
|
post_data['campaign_id'] = post.campaign_id
|
|
post_data['post_content'] = post.post_content
|
|
post_data['twitter_post_id'] = post.twitter_post_id
|
|
posts.append(post_data)
|
|
|
|
return jsonify({
|
|
"campaign_id": twitter_campaign.id,
|
|
"campaign_name": twitter_campaign.campaign_name,
|
|
"user_input": twitter_campaign.user_input,
|
|
"twitter_account_id": twitter_campaign.twitter_account_id,
|
|
"posts": posts
|
|
}), 200
|
|
|
|
@campaign.route('/generate', methods=['POST'])
|
|
def generate_content():
|
|
decoded_token = verify_token(request.headers)
|
|
if not decoded_token:
|
|
return "Not authorised!", 401
|
|
|
|
content_type = request.headers.get('Content-Type')
|
|
if content_type != 'application/json':
|
|
return "Content-type not supported!", 400
|
|
|
|
request_json = request.json
|
|
if 'user_input' not in request_json:
|
|
return "Invalid request!", 400
|
|
|
|
core_url = 'http://s415366.projektstudencki.pl/generate'
|
|
payload = {
|
|
"data": request_json['user_input'],
|
|
"length": 420
|
|
}
|
|
|
|
try:
|
|
response = requests.post(core_url, headers={"Content-Type":"application/json"}, data=json.dumps(payload))
|
|
except:
|
|
return 'Failed to generate content!', 400
|
|
|
|
generated_content = response.content.decode("utf-8").replace('<|endoftext|>', '').replace(' ', ' ')
|
|
content_sentences = sent_tokenize(generated_content)
|
|
trimmed_content = ''
|
|
|
|
for sentence in content_sentences:
|
|
if len(trimmed_content) + len(sentence) < 270:
|
|
trimmed_content += ' '
|
|
trimmed_content += sentence
|
|
|
|
if len(trimmed_content) < 1:
|
|
trimmed_content = generated_content[:270]
|
|
|
|
return jsonify({
|
|
"content": trimmed_content
|
|
}), 200
|
|
|
|
@campaign.route('/post', methods=['POST'])
|
|
def create_post():
|
|
decoded_token = verify_token(request.headers)
|
|
if not decoded_token:
|
|
return "Not authorised!", 401
|
|
|
|
user_id = decoded_token['sub']
|
|
|
|
content_type = request.headers.get('Content-Type')
|
|
if content_type != 'application/json':
|
|
return "Content-type not supported!", 400
|
|
|
|
request_json = request.json
|
|
if 'user_input' not in request_json or 'content' not in request_json or 'campaign_id' not in request_json:
|
|
return "Invalid request!", 400
|
|
|
|
consumer_key = current_app.config["TWITTER_CONSUMER_KEY"]
|
|
consumer_secret = current_app.config["TWITTER_CONSUMER_SECERT"]
|
|
|
|
twitter_campaign = TwitterCampaign.query.filter_by(id=request_json['campaign_id']).first()
|
|
twitter_account = TwitterAccount.query.filter_by(id=twitter_campaign.twitter_account_id).first()
|
|
|
|
# create post on twitter
|
|
oauth = OAuth1Session(
|
|
consumer_key,
|
|
client_secret=consumer_secret,
|
|
resource_owner_key=twitter_account.access_token,
|
|
resource_owner_secret=twitter_account.access_token_secret,
|
|
)
|
|
|
|
response = oauth.post(
|
|
"https://api.twitter.com/2/tweets",
|
|
json={"text": request_json['content']},
|
|
).json()['data']
|
|
|
|
twitter_campaign_post = TwitterPost(
|
|
id = str(uuid4()),
|
|
campaign_id = twitter_campaign.id,
|
|
user_id = user_id,
|
|
user_input = request_json['user_input'],
|
|
post_content = request_json['content'],
|
|
twitter_post_id = response['id']
|
|
)
|
|
|
|
db.session.add(twitter_campaign_post)
|
|
db.session.commit()
|
|
|
|
return "Campaign post succesfully created.", 201
|