campaign api v1

ayct_backend/__init__.py

@@ -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')

ayct_backend/campaign/__init__.py

@@ -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'])
+@campaign.route('/campaign', methods=['GET'])
-def get_twitter_accounts():
+def get_twitter_campaigns():
-    content_type = request.headers.get('Content-Type')
+    twitter_campaigns = TwitterCampaign.query.all()
-    if (content_type == 'application/json'):
-        json = request.json
-        return json
-    else:
-        return 'Content-Type not supported!'
 
-@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'])
+            twitter_account = TwitterAccount.query.filter_by(twitter_account_id=json['twitter_account_id']).first()
-def delete_twitter_account():
+
-    content_type = request.headers.get('Content-Type')
+            # generate campaign content, change after core integration
-    if (content_type == 'application/json'):
+            generated_content = "This is wonderful post content generated by our very smart core API!"
-        json = request.json
+
-        return json
+            # 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

ayct_backend/campaign/models.py

@@ -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)

ayct_backend/twitter/__init__.py

@@ -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"],
+                twitter_account_id = oauth_tokens['user_id'],
-                username = oauth_tokens["screen_name"],
+                username = oauth_tokens['screen_name'],
-                access_token = oauth_tokens["oauth_token"],
+                access_token = oauth_tokens['oauth_token'],
-                access_token_secret = oauth_tokens["oauth_token_secret"]
+                access_token_secret = oauth_tokens['oauth_token_secret']
             )
 
             twitter_db.session.add(new_twitter_account)

ayct_backend/twitter/models.py

@@ -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)