diff options
author | sudan <sudan@douban.com> | 2016-01-20 19:11:15 +0800 |
---|---|---|
committer | sudan <sudan@douban.com> | 2016-01-20 19:11:15 +0800 |
commit | 084442616ffcceb309c71a62b123f7a11606576a (patch) | |
tree | 3ee7c3b2b7b9a454b8f1b3e8309c3edf25611541 /pastebin.py | |
parent | some change (diff) | |
parent | misc updates (diff) | |
download | stackbin-084442616ffcceb309c71a62b123f7a11606576a.tar.gz stackbin-084442616ffcceb309c71a62b123f7a11606576a.tar.bz2 stackbin-084442616ffcceb309c71a62b123f7a11606576a.zip |
merge from jiawei
Diffstat (limited to 'pastebin.py')
-rw-r--r-- | pastebin.py | 116 |
1 files changed, 10 insertions, 106 deletions
diff --git a/pastebin.py b/pastebin.py index 3824285..03ab9b8 100644 --- a/pastebin.py +++ b/pastebin.py @@ -1,26 +1,12 @@ from datetime import datetime -from flask import Flask, request, url_for, redirect, g, session, flash, \ - abort, render_template +from flask import (Flask, request, url_for, redirect, g, + render_template, session) from flask.ext.sqlalchemy import SQLAlchemy -from flask.ext.oauth import OAuth -from juggernaut import Juggernaut app = Flask(__name__) app.config.from_pyfile('config.cfg') db = SQLAlchemy(app) -oauth = OAuth() -jug = Juggernaut() - -facebook = oauth.remote_app('facebook', - base_url='https://graph.facebook.com/', - request_token_url=None, - access_token_url='/oauth/access_token', - authorize_url='https://www.facebook.com/dialog/oauth', - consumer_key='188477911223606', - consumer_secret='621413ddea2bcc5b2e83d42fc40495de', - request_token_params={'scope': 'email'} -) def url_for_other_page(page): @@ -30,6 +16,13 @@ def url_for_other_page(page): app.jinja_env.globals['url_for_other_page'] = url_for_other_page +@app.before_request +def check_user_status(): + g.user = None + if 'user_id' in session: + g.user = User.query.get(session['user_id']) + + class Paste(db.Model): id = db.Column(db.Integer, primary_key=True) code = db.Column(db.Text) @@ -53,26 +46,6 @@ class User(db.Model): pastes = db.relationship(Paste, lazy='dynamic', backref='user') -def send_new_paste_notifications(paste, reply): - """Notifies clients about new pastes.""" - user = None - user_id = None - if paste.user: - user = paste.user.display_name - user_id = paste.user.id - data = {'paste_id': paste.id, 'reply_id': reply.id, 'user': user} - jug.publish('paste-replies:%d' % paste.id, data) - if user_id is not None: - jug.publish('user-replies:%d' % user_id, data) - - -@app.before_request -def check_user_status(): - g.user = None - if 'user_id' in session: - g.user = User.query.get(session['user_id']) - - @app.route('/', methods=['GET', 'POST']) def new_paste(): parent = None @@ -83,82 +56,13 @@ def new_paste(): paste = Paste(g.user, request.form['code'], parent=parent) db.session.add(paste) db.session.commit() - if parent is not None: - send_new_paste_notifications(parent, paste) return redirect(url_for('show_paste', paste_id=paste.id)) return render_template('new_paste.html', parent=parent) +@app.route('/<int:paste_id>/') @app.route('/<int:paste_id>') def show_paste(paste_id): paste = Paste.query.options(db.eagerload('children')).get_or_404(paste_id) return render_template('show_paste.html', paste=paste) - -@app.route('/<int:paste_id>/delete', methods=['GET', 'POST']) -def delete_paste(paste_id): - paste = Paste.query.get_or_404(paste_id) - if g.user is None or g.user != paste.user: - abort(401) - if request.method == 'POST': - if 'yes' in request.form: - db.session.delete(paste) - db.session.commit() - flash('Paste was deleted') - return redirect(url_for('new_paste')) - else: - return redirect(url_for('show_paste', paste_id=paste.id)) - return render_template('delete_paste.html', paste=paste) - - -@app.route('/my-pastes/', defaults={'page': 1}) -@app.route('/my-pastes/page/<int:page>') -def my_pastes(page): - if g.user is None: - return redirect(url_for('login', next=request.url)) - pagination = Paste.query.filter_by(user=g.user).paginate(page) - return render_template('my_pastes.html', pagination=pagination) - - -@app.route('/login') -def login(): - return facebook.authorize(callback=url_for('facebook_authorized', - next=request.args.get('next') or request.referrer or None, - _external=True)) - - -@app.route('/logout') -def logout(): - session.clear() - flash('You were logged out') - return redirect(url_for('new_paste')) - - -@app.route('/login/authorized') -@facebook.authorized_handler -def facebook_authorized(resp): - next_url = request.args.get('next') or url_for('new_paste') - if resp is None: - flash('You denied the login') - return redirect(next_url) - - session['fb_access_token'] = (resp['access_token'], '') - - me = facebook.get('/me') - user = User.query.filter_by(fb_id=me.data['id']).first() - if user is None: - user = User() - user.fb_id = me.data['id'] - db.session.add(user) - - user.display_name = me.data['name'] - db.session.commit() - session['user_id'] = user.id - - flash('You are now logged in as %s' % user.display_name) - return redirect(next_url) - - -@facebook.tokengetter -def get_facebook_oauth_token(): - return session.get('fb_access_token') |