From 803d20968fea1e25a298aa3b4e670d5670995e0d Mon Sep 17 00:00:00 2001 From: Armin Ronacher Date: Mon, 5 Dec 2011 22:38:43 -0500 Subject: Added as example app --- pastebin.py | 164 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 164 insertions(+) create mode 100644 pastebin.py (limited to 'pastebin.py') diff --git a/pastebin.py b/pastebin.py new file mode 100644 index 0000000..3824285 --- /dev/null +++ b/pastebin.py @@ -0,0 +1,164 @@ +from datetime import datetime +from flask import Flask, request, url_for, redirect, g, session, flash, \ + abort, render_template +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): + args = request.view_args.copy() + args['page'] = page + return url_for(request.endpoint, **args) +app.jinja_env.globals['url_for_other_page'] = url_for_other_page + + +class Paste(db.Model): + id = db.Column(db.Integer, primary_key=True) + code = db.Column(db.Text) + pub_date = db.Column(db.DateTime) + user_id = db.Column(db.Integer, db.ForeignKey('user.id')) + parent_id = db.Column(db.Integer, db.ForeignKey('paste.id')) + parent = db.relationship('Paste', lazy=True, backref='children', + uselist=False, remote_side=[id]) + + def __init__(self, user, code, parent=None): + self.user = user + self.code = code + self.pub_date = datetime.utcnow() + self.parent = parent + + +class User(db.Model): + id = db.Column(db.Integer, primary_key=True) + display_name = db.Column(db.String(120)) + fb_id = db.Column(db.String(30), unique=True) + 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 + reply_to = request.args.get('reply_to', type=int) + if reply_to is not None: + parent = Paste.query.get(reply_to) + if request.method == 'POST' and request.form['code']: + 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('/') +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('//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/') +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') -- cgit