server.py

import json
import time
import re
import oauth2 as oauth
import os
from secret import keys
from flask import (Flask, jsonify, render_template)
from model import (connect_to_db, db, Tweet)
from sqlalchemy import desc

key = keys.key()


app = Flask(__name__)
#  key generated by OS
app.secret_key = key['Flask_Key']

link = re.compile(r'(http(s)?://\w+(\.\w+)+(/\w+)*|@(\w+)|#(\w+))')

DEFAULT_DB_URI = "postgresql:///newb"
DB_URI = os.environ.get(
    'NEWBIE_TWEETS_DB_URI',
    DEFAULT_DB_URI,
)
DEFAULT_LISTEN_HOST = '127.0.0.1:5000'
LISTEN_HOST = os.environ.get(
    'NEWBIE_TWEETS_LISTEN_HOST',
    DEFAULT_LISTEN_HOST,
)
DEFAULT_LISTEN_PORT = '5000'
LISTEN_PORT = int(os.environ.get(
    'NEWBIE_TWEETS_LISTEN_PORT',
    DEFAULT_LISTEN_PORT,
))

def authorize():
    """authorize w/ twitter api and fetch recent codenewbie tweets, return a json"""

    consumer = oauth.Consumer(key=key["consumer_key"], secret=key["consumer_secret"])
    access_token = oauth.Token(key=key["access_token"], secret=key["access_secret"])
    client = oauth.Client(consumer, access_token)
    test_url = "https://api.twitter.com/1.1/search/tweets.json?q=%23codenewbie&result_type=mixed&count=100&include_entities=false"
    response, data = client.request(test_url)

    return json.loads(data)


def format_tweets():
    """return a list of tuples of recent codenewbie tweets"""

    tweet = None
    time_created = None
    retweets = None
    results = authorize()["statuses"]
    output = []

    for result in results:
        if result['text'][0:2] != 'RT':
            tweet = result['text']
            time_created = time.strftime('%Y-%m-%d %H:%M:%S', time.strptime(result['created_at'],
                                                                            '%a %b %d %H:%M:%S +0000 %Y'))
            handle = result['user']['screen_name']
            if 'retweeted_status' in result:
                retweets = result['retweeted_status']['retweet_count']
            output.append((handle, time_created, tweet, retweets))

    return output


def tweet_to_db():
    """Add tweets into db"""

    output = format_tweets()

    text_list = [a.text for a in Tweet.query.all()]

    for tweet in output:
        if tweet[2] not in text_list:  # need to edit this
            tweet = Tweet(handle=tweet[0],
                          time_created=tweet[1],
                          text=tweet[2],
                          retweets=tweet[3])
            db.session.add(tweet)

    db.session.commit()


def linkyfy(text, is_name=False):
    """ Embeds links in anchor tag"""
    if is_name:
        text = u"<a href='https://twitter.com/{0}'>{0}</a>".format(text)
        return text
    links = link.findall(text)
    for l in links:
        if l[0].startswith('@'):
            text = text.replace(l[0], r"<a href='https://twitter.com/%s'>%s</a>" % (l[4], l[0]))
        elif l[0].startswith('#'):
            text = text.replace(l[0], r"<a href='https://twitter.com/search?q=%%23%s'>%s</a>"%(l[5], l[0]))
        else:
            text = text.replace(l[0], r"<a href='%s'>%s</a>" % (l[0], l[0]))
    return text


@app.route("/")
def homepage():
    """Display tweets"""

    tweet_to_db()
    output = [a for a in Tweet.query.order_by(desc('time_created')).all()]

    # to display as hyper links
    for tweet in output:
        tweet.handle = linkyfy(tweet.handle, is_name=True)
        tweet.text = linkyfy(tweet.text)

    return render_template("home.html", output=output)


@app.route("/about")
def display_about():
    """Diplay about page"""

    return render_template("about.html")


@app.route("/friends")
def display_friends():
    """Display list of contributors to open source code practice project"""

    return render_template("friends.html")


@app.route("/api/tweets")
def create_api_endpoint():
    """Using ingested dsta from twitter create an API endpoint"""

    tweedict = {}
    tweets = Tweet.query.all()

    for tweet in tweets:
        tweedict[tweet.handle] = tweet.text

    return jsonify(tweedict)


@app.route("/archives")
def archives():
    """ Displays previous tweets """

    return render_template("archives.html")


if __name__ == "__main__":
    app.debug = True
    connect_to_db(app, DB_URI)
    app.run(host=LISTEN_HOST, port=LISTEN_PORT)