service/service/__init__.py

from functools import wraps
import json

from werkzeug.utils import redirect

import APIFactory
from service import user
import config
from models import UserEncoder
import providers
from service.utils import secured


__author__ = 'hanl'

from flask import Flask, render_template, request, session, url_for, flash, current_app, Blueprint
from flask_email import LocmemMail
from flask_babel import Babel
from flask_login import LoginManager, current_user

app = Flask(__name__)

lm = LoginManager(app)
babel = Babel(app)
# admin = Admin(service)
mail = LocmemMail(app)

#router = Blueprint('app', __name__,
#                  template_folder='templates')


@app.route('/lang/<language>')
def set_locale(lang=None):
    setattr(session, 'lang', lang)
    return render_template(url_for('.index'), lang_code=lang)


@babel.localeselector
def get_locale():
    browser_locale = request.accept_languages.best_match(app.config.get('LANGUAGES'), default="de")
    lang = session.get('lang', browser_locale)
    setattr(session, 'lang', lang)
    return lang


@lm.user_loader
def load_user(username):
    # either from session of via rest; cache if possible
    return providers.PROVIDER.get_user(session)


# used to override the cookie encryption behaviour
# @lm.token_loader
def load_token(token):
    return providers.PROVIDER.get_user()


@app.route('/')
def index():
    return render_template('index.html', lang_code=get_locale())


@app.route('/search', methods=['GET'])
@secured
def search():
    if request.method == "GET":
        data = request.args
        print "has query %s" % str(data)
        if 'q' in data:
            # auth=APIFactory.Oauth2Auth(session['access_token'])
            response = APIFactory.get("search", params=data)
            if response is None:
                pass
            elif providers.NotificationHandler.isError(response):
                providers.NotificationHandler.notify(response.json(), flash)
            else:
                raw_json = response.json()
                return render_template('search.html', lang=app.config.get('QUERY_LANGUAGES'),
                                       q=data['q'], ql=data['ql'], result=raw_json,
                                       result_string=json.dumps(raw_json, indent=4))
    return render_template('search.html', lang=app.config.get('QUERY_LANGUAGES'),
                           q=None, ql=None,
                           result=None)


@app.route('/serialize', methods=['GET', 'POST'])
@secured
def serialize():
    return render_template('search.html')


@app.errorhandler(404)
def page_not_found(e):
    return render_template('404.html'), 404


if __name__ == '__main__':
    import admin

    app.config.from_object(config)
    app.json_encoder = UserEncoder
    app.debug = config.DEBUG
    app.register_blueprint(user.router, url_prefix='/user')
    app.register_blueprint(admin.router, url_prefix='/admin')
    providers.init_app(app.config)
    app.run()