s434708/pbr-ayct-backend
1
2
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

campaign api v1

Browse Source
This commit is contained in:
Łukasz Jędyk 2022-01-24 23:20:49 +01:00
parent a41a7dc052
commit a295309398
5 changed files with 71 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ayct_backend/__init__.py
View File

@ -23,7 +23,8 @@ def create_app():
twitter_db.drop_all(app=app)
twitter_db.create_all(app=app)
#campaign_db.create_all(app=app)
campaign_db.drop_all(app=app)
campaign_db.create_all(app=app)
app.register_blueprint(twitter, url_prefix='/twitter')
app.register_blueprint(campaign, url_prefix='/campaign')

76
ayct_backend/campaign/__init__.py
View File

@ -1,31 +1,71 @@
import os
from requests_oauthlib import OAuth1Session
from flask import Blueprint, current_app, request, jsonify
from ayct_backend.twitter.models import *
from ayct_backend.campaign.models import *
campaign = Blueprint('campaign', __name__)
@campaign.route('/account', methods=['GET'])
def get_twitter_accounts():
content_type = request.headers.get('Content-Type')
if (content_type == 'application/json'):
json = request.json
return json
else:
return 'Content-Type not supported!'
@campaign.route('/campaign', methods=['GET'])
def get_twitter_campaigns():
twitter_campaigns = TwitterCampaign.query.all()
@campaign.route('/account', methods=['POST'])
campaigns = []
for campaign in twitter_campaigns:
campaign_data = {}
campaign_data['campaign_name'] = campaign.campaign_name
campaign_data['twitter_account_id'] = campaign.twitter_account_id
campaign_data['user_input'] = campaign.user_input
campaign_data['generated_content'] = campaign.generated_content
campaign_data['twitter_post_id'] = campaign.twitter_post_id
campaigns.append(campaign_data)
return jsonify({
"twitter_campaigns": campaigns
}), 200
@campaign.route('/campaign', methods=['POST'])
def add_twitter_account():
content_type = request.headers.get('Content-Type')
if (content_type == 'application/json'):
json = request.json
return json
if not json['campaign_name'] or not json['user_input'] or not json['twitter_account_id']:
return "Invalid request!", 400
else:
return 'Content-Type not supported!'
consumer_key = current_app.config["TWITTER_CONSUMER_KEY"]
consumer_secret = current_app.config["TWITTER_CONSUMER_SECERT"]
@campaign.route('/account', methods=['DELETE'])
def delete_twitter_account():
content_type = request.headers.get('Content-Type')
if (content_type == 'application/json'):
json = request.json
return json
twitter_account = TwitterAccount.query.filter_by(twitter_account_id=json['twitter_account_id']).first()
# generate campaign content, change after core integration
generated_content = "This is wonderful post content generated by our very smart core API!"
# 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": generated_content},
)
# save campaign to database
new_twitter_campaign = TwitterCampaign(
campaign_name = json['campaign_name'],
twitter_account_id = json['twitter_account_id'],
user_input = json['user_input'],
generated_content = generated_content,
twitter_post_id = response['id']
)
campaign_db.session.add(new_twitter_campaign)
campaign_db.session.commit()
return "Campaign succesfully created.", 201
else:
return 'Content-Type not supported!'
return "Content-type not supported!", 400

5
ayct_backend/campaign/models.py
View File

@ -6,3 +6,8 @@ class TwitterCampaign(campaign_db.Model):
__tablename__ = 'twitter_campaign'
campaign_id = campaign_db.Column(campaign_db.Integer, primary_key=True)
campaign_name = campaign_db.Column(campaign_db.String(64), nullable=False)
twitter_account_id = campaign_db.Column(campaign_db.String(32), unique=True, nullable=False)
user_input = campaign_db.Column(campaign_db.String(100), nullable=False)
generated_content = campaign_db.Column(campaign_db.String(300), nullable=False)
twitter_post_id = campaign_db.Column(campaign_db.String(32), unique=True, nullable=False)

9
ayct_backend/twitter/__init__.py
View File

@ -1,5 +1,4 @@
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 *
@ -44,10 +43,10 @@ def add_twitter_account():
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_account_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)

2
ayct_backend/twitter/models.py
View File

@ -7,7 +7,7 @@ class TwitterAccount(twitter_db.Model):
__tablename__ = 'twitter_account'
account_id = twitter_db.Column(twitter_db.Integer, primary_key=True)
twitter_user_id = twitter_db.Column(twitter_db.String(32), unique=True, nullable=False)
twitter_account_id = twitter_db.Column(twitter_db.String(32), unique=True, nullable=False)
username = twitter_db.Column(twitter_db.String(16), unique=True, nullable=False)
access_token = twitter_db.Column(twitter_db.String(256), nullable=False)
access_token_secret = twitter_db.Column(twitter_db.String(256), nullable=False)