flask/flask_website/__init__.py
2010-05-16 01:06:37 +02:00

40 lines
1.2 KiB
Python

from flask import Flask, session, g, render_template
from flaskext.openid import OpenID
import websiteconfig as config
app = Flask(__name__)
app.debug = config.DEBUG
app.secret_key = config.SECRET_KEY
from flask_website.openid_auth import DatabaseOpenIDStore
oid = OpenID(store_factory=DatabaseOpenIDStore)
@app.errorhandler(404)
def not_found(error):
return render_template('404.html'), 404
@app.before_request
def load_currrent_user():
g.user = User.query.filter_by(openid=session['openid']).first() \
if 'openid' in session else None
@app.after_request
def remove_db_session(response):
db_session.remove()
return response
from flask_website.views.general import general
from flask_website.views.mailinglist import mailinglist
from flask_website.views.snippets import snippets
from flask_website.views.extensions import extensions
app.register_module(general)
app.register_module(mailinglist)
app.register_module(snippets)
app.register_module(extensions)
from flask_website.database import User, db_session
from flask_website import utils
app.jinja_env.filters['datetimeformat'] = utils.format_datetime
app.jinja_env.filters['timedeltaformat'] = utils.format_timedelta