mirror of
https://github.com/jlengrand/twitterboard.git
synced 2026-03-10 08:51:22 +00:00
3277ebdffe
Creating a table in javascript is not a good way, as it is not rendered correctly in the browser. Better format it server side or find a way to populate it correctly client side. Also has to create a remove_hashtag method that takes arguments in input
119 lines
2.7 KiB
Python
119 lines
2.7 KiB
Python
from flask import Flask, jsonify, render_template, request
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy import func
|
|
from sqlalchemy import desc
|
|
|
|
|
|
import data
|
|
from datamodel import TrendyHashtag
|
|
from datamodel import Tweet
|
|
from datamodel import Member
|
|
|
|
from streamer import HashtagLogger
|
|
|
|
|
|
app = Flask(__name__)
|
|
h = HashtagLogger(data.engine_url, oauth=data.oauth)
|
|
h.start()
|
|
|
|
|
|
## Utilities
|
|
def connect():
|
|
"""
|
|
Separated so that the method can be run in each created thread.
|
|
Initiates connexion to the database and starts a Session to be used to query it.
|
|
Returns the session used to communicate with the database
|
|
"""
|
|
# creates engine, tries to create all the tables needed later on
|
|
engine = create_engine(data.engine_url, echo=data.debug)
|
|
# initiates session to the database, tries to create proper session
|
|
Session = sessionmaker(bind=engine)
|
|
|
|
return Session(), engine # Bridges class to db
|
|
|
|
|
|
## Ajax queries
|
|
@app.route('/trendy')
|
|
def trendy():
|
|
|
|
session, engine = connect()
|
|
|
|
# requests active hashtags
|
|
h_query = session.query(TrendyHashtag).filter(TrendyHashtag.active == True).order_by(desc(TrendyHashtag.created))
|
|
hashtags = h_query.all()
|
|
|
|
trendy = []
|
|
for h in hashtags:
|
|
trendy.append([h.hashtag, 1])
|
|
|
|
session.close()
|
|
engine.dispose()
|
|
|
|
trendy = [[1, 2], [3, 4], [5, 6]]
|
|
#trendy = 'plop'
|
|
return jsonify(trendy=trendy)
|
|
|
|
|
|
@app.route('/nb_trendy')
|
|
def nb_trendy():
|
|
|
|
session, engine = connect()
|
|
|
|
# requests active hashtags
|
|
query = session.query(func.count(TrendyHashtag.id)).filter(TrendyHashtag.active == True)
|
|
hashs = query.first()[0]
|
|
|
|
#requests number of tweets
|
|
query = session.query(func.count(Tweet.id))
|
|
tweets = query.first()[0]
|
|
|
|
#requests number of members
|
|
query = session.query(func.count(Member.id))
|
|
members = query.first()[0]
|
|
|
|
session.close()
|
|
engine.dispose()
|
|
|
|
return jsonify(hashs=hashs, tweets=tweets, members=members)
|
|
|
|
|
|
@app.route('/_add_hashtag')
|
|
def add_hashtag():
|
|
new_hash = request.args.get('new_hash')
|
|
hashtag = h.add_hashtag(new_hash) # just to check we create the same hashtag
|
|
return jsonify(hash="Adding %s !" % (hashtag))
|
|
|
|
|
|
@app.route('/remove_hashtag')
|
|
def remove_hashtag():
|
|
hashtag = h.remove_hashtag('plop')
|
|
return hashtag
|
|
|
|
|
|
@app.route('/stop')
|
|
def stop():
|
|
h.stop()
|
|
return "Session closed"
|
|
|
|
|
|
## Static part
|
|
@app.route('/')
|
|
def index():
|
|
return render_template('index.html')
|
|
|
|
|
|
@app.route('/statistics')
|
|
def statistics():
|
|
return render_template('statistics.html')
|
|
|
|
|
|
@app.route('/about')
|
|
def about():
|
|
return render_template('about.html')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True)
|