flask/flask_website/__init__.py
2010-06-21 19:53:31 +02:00

54 lines
1.7 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(app, store_factory=DatabaseOpenIDStore)
@app.errorhandler(404)
def not_found(error):
return render_template('404.html'), 404
@app.before_request
def load_current_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
app.add_url_rule('/docs/', endpoint='docs.index', build_only=True)
app.add_url_rule('/docs/<path:page>/', endpoint='docs.show',
build_only=True)
app.add_url_rule('/docs/flask-docs.pdf', endpoint='docs.pdf',
build_only=True)
app.add_url_rule('/docs/flask-docs.zip', endpoint='docs.zip',
build_only=True)
from flask_website.views.general import general
from flask_website.views.community import community
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(community)
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
app.jinja_env.filters['displayopenid'] = utils.display_openid