fetch account on login

This commit is contained in:
codl 2017-07-30 13:53:14 +02:00
parent bc3def5152
commit cf286918f9
No known key found for this signature in database
GPG Key ID: 6CD7C8891ED1233A
1 changed files with 5 additions and 0 deletions

View File

@ -4,6 +4,7 @@ from datetime import datetime
import lib.twitter
from model import Account, Session, Post
from app import db
import tasks
@app.before_request
def load_viewer():
@ -39,9 +40,13 @@ 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).delay()
resp = Response(status=301, headers={"location": url_for('index')})
resp.set_cookie('forget_sid', session.id)
return resp