forget-cancellare-vecchi-toot/routes.py

100 lines
2.9 KiB
Python

from app import app
from flask import render_template, url_for, redirect, request, g, Response
from datetime import datetime
import lib.twitter
import lib
from lib import require_auth
from model import Account, Session, Post, TwitterArchive
from app import db
import tasks
@app.before_request
def load_viewer():
g.viewer = None
sid = request.cookies.get('forget_sid', None)
if sid:
g.viewer = Session.query.get(sid)
@app.after_request
def touch_viewer(resp):
if g.viewer:
g.viewer.touch()
db.session.commit()
return resp
@app.route('/')
def index():
if g.viewer:
posts = Post.query.filter_by(author_id = g.viewer.account_id).order_by(db.desc(Post.created_at)).limit(30)
return render_template('index.html', posts=posts)
else:
return render_template('index.html')
@app.route('/login/twitter')
def twitter_login_step1():
return redirect(lib.twitter.get_login_url(
callback = url_for('twitter_login_step2', _external=True),
**app.config.get_namespace("TWITTER_")
))
@app.route('/login/twitter/callback')
def twitter_login_step2():
oauth_token = request.args['oauth_token']
oauth_verifier = request.args['oauth_verifier']
token = lib.twitter.receive_verifier(oauth_token, oauth_verifier, **app.config.get_namespace("TWITTER_"))
session = Session(account_id = token.account_id)
db.session.add(session)
db.session.commit()
tasks.fetch_acc.s(token.account_id).apply_async(routing_key='high')
resp = Response(status=302, headers={"location": url_for('index')})
resp.set_cookie('forget_sid', session.id,
max_age=60*60*48,
httponly=True,
secure=app.config.get("HTTPS"))
return resp
@app.route('/upload_tweet_archive', methods=('POST',))
@require_auth
def upload_tweet_archive():
ta = TwitterArchive(account = g.viewer.account,
body = request.files['file'].read())
db.session.add(ta)
db.session.commit()
tasks.import_twitter_archive.s(ta.id).apply_async(routing_key='high')
return render_template('upload_tweet_archive.html')
@app.route('/settings', methods=('GET', 'POST'))
@require_auth
def settings():
if request.method == 'POST':
for attr in ('policy_enabled',
'policy_ignore_favourites',
'policy_keep_latest',
'policy_delete_every_significand',
'policy_delete_every_scale',
'policy_keep_younger_significand',
'policy_keep_younger_scale',
):
if attr in request.form:
setattr(g.viewer.account, attr, request.form[attr])
db.session.commit()
return render_template('settings.html', scales=lib.interval_scales)
@app.route('/logout')
@require_auth
def logout():
if(g.viewer):
db.session.delete(g.viewer)
db.session.commit()
g.viewer = None
return redirect(url_for('index'))