pbr-ayct-backend/ayct_backend/campaign/__init__.py

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('/campaign', methods=['GET'])
def get_twitter_campaigns():
    twitter_campaigns = TwitterCampaign.query.all()

    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
        if not json['campaign_name'] or not json['user_input'] or not json['twitter_account_id']:
            return "Invalid request!", 400
        else:
            consumer_key = current_app.config["TWITTER_CONSUMER_KEY"]
            consumer_secret = current_app.config["TWITTER_CONSUMER_SECERT"]

            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},
            ).json()

            # 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!", 400