mirror of
https://github.com/kennethreitz/curated-nodes.git
synced 2026-06-05 23:20:20 +00:00
Kenneth Reitz
286 lines
7.6 KiB
Python
286 lines
7.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import re
|
|
import unidecode
|
|
from functools import wraps
|
|
from datetime import datetime
|
|
|
|
from flask import Flask, abort, request, render_template, Response, redirect, url_for
|
|
from flask.ext.sqlalchemy import SQLAlchemy
|
|
from sqlalchemy.dialects.postgresql import HSTORE
|
|
|
|
from flask.ext.markdown import Markdown
|
|
|
|
|
|
STYLES = ('prose', 'audio', 'photo', 'code', 'product')
|
|
VIEWS = ('expressions', 'exposures', 'experiments', 'pages')
|
|
|
|
SECRET = os.environ['SECRET']
|
|
|
|
app = Flask(__name__)
|
|
app.debug = os.environ.get('DEBUG')
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = os.environ['DATABASE_URL']
|
|
|
|
|
|
db = SQLAlchemy(app)
|
|
Markdown(app)
|
|
|
|
|
|
|
|
def slugify(str):
|
|
str = unidecode.unidecode(str).lower()
|
|
return re.sub(r'\W+','-',str)
|
|
|
|
|
|
def check_auth(username, password):
|
|
"""This function is called to check if a username /
|
|
password combination is valid.
|
|
"""
|
|
return password == SECRET
|
|
|
|
def authenticate():
|
|
"""Sends a 401 response that enables basic auth"""
|
|
return Response(
|
|
'Could not verify your access level for that URL.\n'
|
|
'You have to login with proper credentials', 401,
|
|
{'WWW-Authenticate': 'Basic realm="Login Required"'})
|
|
|
|
|
|
def requires_auth(f):
|
|
@wraps(f)
|
|
def decorated(*args, **kwargs):
|
|
auth = request.authorization
|
|
if not auth or not check_auth(auth.username, auth.password):
|
|
return authenticate()
|
|
return f(*args, **kwargs)
|
|
return decorated
|
|
|
|
|
|
class BaseModel(object):
|
|
def save(self):
|
|
db.session.add(self)
|
|
return db.session.commit()
|
|
|
|
class Page(db.Model, BaseModel):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
slug = db.Column(db.String(80), unique=True)
|
|
content = db.Column(db.Text())
|
|
title = db.Column(db.String(140))
|
|
draft = db.Column(db.Boolean, default=True)
|
|
|
|
def __repr__(self):
|
|
return '<Page %r>' % self.slug
|
|
|
|
class Expression(db.Model, BaseModel):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
slug = db.Column(db.String(80), unique=True)
|
|
content = db.Column(db.Text())
|
|
title = db.Column(db.String(140))
|
|
link = db.Column(db.String(280))
|
|
draft = db.Column(db.Boolean, default=True)
|
|
style = db.Column(db.String(80))
|
|
meta = db.Column(HSTORE())
|
|
created = db.Column(db.DateTime, default=datetime.now)
|
|
updated = db.Column(db.DateTime, default=datetime.now)
|
|
|
|
def __repr__(self):
|
|
return '<Expression %r>' % self.slug
|
|
|
|
class Experiment(db.Model, BaseModel):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
slug = db.Column(db.String(80), unique=True)
|
|
content = db.Column(db.Text())
|
|
title = db.Column(db.String(140))
|
|
link = db.Column(db.String(280))
|
|
draft = db.Column(db.Boolean, default=True)
|
|
style = db.Column(db.String(80))
|
|
status = db.Column(db.String(80))
|
|
meta = db.Column(HSTORE())
|
|
created = db.Column(db.DateTime, default=datetime.now)
|
|
updated = db.Column(db.DateTime, default=datetime.now)
|
|
graduated = db.Column(db.String(280))
|
|
|
|
def __repr__(self):
|
|
return '<Experiment %r>' % self.slug
|
|
|
|
class Exposure(db.Model, BaseModel):
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
slug = db.Column(db.String(80), unique=True)
|
|
content = db.Column(db.Text())
|
|
title = db.Column(db.String(140))
|
|
link = db.Column(db.String(280))
|
|
draft = db.Column(db.Boolean, default=True)
|
|
style = db.Column(db.String(80))
|
|
meta = db.Column(HSTORE())
|
|
created = db.Column(db.DateTime, default=datetime.now)
|
|
updated = db.Column(db.DateTime, default=datetime.now)
|
|
|
|
def __repr__(self):
|
|
return '<Exposure %r>' % self.slug
|
|
|
|
|
|
@app.errorhandler(404)
|
|
def page_not_found(error):
|
|
return render_template('404.html'), 404
|
|
|
|
@app.route('/')
|
|
def get_index():
|
|
return render_template('index.html')
|
|
|
|
|
|
|
|
@app.route('/<slug>')
|
|
def get_page(slug):
|
|
|
|
want_drafts = ('preview' in request.args)
|
|
|
|
p = Page.query.filter_by(slug=slug, draft=want_drafts).first() or abort(404)
|
|
|
|
return render_template('page.html', page=p)
|
|
|
|
@app.route('/exposures')
|
|
def get_exposures(filter=None):
|
|
|
|
f = {'draft': False}
|
|
if filter:
|
|
f['style'] = filter
|
|
|
|
q = Exposure.query.filter_by(**f).order_by(Exposure.id.desc()).all()
|
|
|
|
return render_template('list.html', items=q, view='exposures', filter=filter)
|
|
|
|
@app.route('/exposures/<slug>')
|
|
def get_exposure(slug):
|
|
|
|
want_drafts = ('preview' in request.args)
|
|
|
|
if slug in STYLES:
|
|
return get_exposures(filter=slug)
|
|
|
|
e = Exposure.query.filter_by(slug=slug, draft=want_drafts).first() or abort(404)
|
|
|
|
return render_template('post.html', post=e, view='exposures')
|
|
|
|
|
|
@app.route('/experiments')
|
|
def get_experiments(filter=None):
|
|
|
|
f = {'draft': False}
|
|
if filter:
|
|
f['style'] = filter
|
|
|
|
q = Experiment.query.filter_by(**f).order_by(Experiment.id.desc()).all()
|
|
|
|
return render_template('list.html', items=q, view='experiments', filter=filter)
|
|
|
|
@app.route('/admin')
|
|
@requires_auth
|
|
def get_admin():
|
|
|
|
return get_admin_view()
|
|
|
|
@app.route('/admin/<view>', methods=['GET', 'POST'])
|
|
@requires_auth
|
|
def get_admin_view(view='pages'):
|
|
|
|
if view not in VIEWS:
|
|
abort(404)
|
|
|
|
if view == 'pages':
|
|
model = Page
|
|
elif view == 'expressions':
|
|
model = Expression
|
|
elif view == 'exposures':
|
|
model = Exposure
|
|
elif view == 'experiments':
|
|
model = Experiment
|
|
|
|
if request.method == 'POST':
|
|
p = model()
|
|
p.title = request.form['post[title]']
|
|
p.slug = slugify(p.title)
|
|
|
|
p.save()
|
|
return redirect(url_for('get_admin_edit', view=view, slug=p.slug))
|
|
|
|
drafts = model.query.filter_by(draft=True).order_by(model.id.desc()).all()
|
|
published = model.query.filter_by(draft=False).order_by(model.id.desc()).all()
|
|
|
|
return render_template('admin.html', drafts=drafts, published=published, view=view)
|
|
|
|
|
|
@app.route('/admin/<view>/<slug>', methods=['GET', 'POST'])
|
|
@requires_auth
|
|
def get_admin_edit(view, slug):
|
|
|
|
if view not in VIEWS:
|
|
abort(404)
|
|
|
|
if view == 'pages':
|
|
model = Page
|
|
elif view == 'expressions':
|
|
model = Expression
|
|
elif view == 'exposures':
|
|
model = Exposure
|
|
elif view == 'experiments':
|
|
model = Experiment
|
|
|
|
if request.method == 'POST':
|
|
e = model.query.filter_by(slug=slug).first() or abort(406)
|
|
e.title = request.form['post[title]']
|
|
e.slug = request.form['post[slug]']
|
|
e.content = request.form['post[content]']
|
|
|
|
if not view == 'pages':
|
|
e.style = request.form['post[style]']
|
|
|
|
e.draft = 'post[draft]' in request.form
|
|
e.save()
|
|
|
|
return redirect(url_for('get_admin_edit', view=view, slug=e.slug))
|
|
|
|
e = model.query.filter_by(slug=slug).first() or abort(404)
|
|
|
|
return render_template('admin-edit.html', post=e, view=view)
|
|
|
|
|
|
@app.route('/experiments/<slug>')
|
|
def get_experiment(slug):
|
|
|
|
want_drafts = ('preview' in request.args)
|
|
|
|
if slug in STYLES:
|
|
return get_experiments(filter=slug)
|
|
|
|
e = Experiment.query.filter_by(slug=slug, draft=want_drafts).first() or abort(404)
|
|
|
|
return render_template('post.html', post=e, view='experiments')
|
|
|
|
|
|
@app.route('/expressions')
|
|
def get_expressions(filter=None):
|
|
|
|
f = {'draft': False}
|
|
if filter:
|
|
f['style'] = filter
|
|
|
|
q = Expression.query.filter_by(**f).order_by(Expression.id.desc()).all()
|
|
|
|
return render_template('list.html', items=q, view='expressions', filter=filter)
|
|
|
|
@app.route('/expressions/<slug>')
|
|
def get_expression(slug):
|
|
|
|
want_drafts = ('preview' in request.args)
|
|
|
|
if slug in STYLES:
|
|
return get_expressions(filter=slug)
|
|
|
|
e = Expression.query.filter_by(slug=slug, draft=want_drafts).first() or abort(404)
|
|
|
|
return render_template('post.html', post=e, view='expressions')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run() |