contentdb/app/__init__.py

117 lines
3.3 KiB
Python
Raw Normal View History

2020-07-12 17:34:25 +02:00
# ContentDB
# Copyright (C) 2018-21 rubenwardy
2018-05-17 16:18:20 +02:00
#
# This program is free software: you can redistribute it and/or modify
2021-01-30 17:59:42 +01:00
# it under the terms of the GNU Affero General Public License as published by
2018-05-17 16:18:20 +02:00
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
2021-01-30 17:59:42 +01:00
# GNU Affero General Public License for more details.
2018-05-17 16:18:20 +02:00
#
2021-01-30 17:59:42 +01:00
# You should have received a copy of the GNU Affero General Public License
2018-05-17 16:18:20 +02:00
# along with this program. If not, see <https://www.gnu.org/licenses/>.
2018-03-18 18:43:30 +01:00
from flask import *
2018-12-22 21:13:43 +01:00
from flask_gravatar import Gravatar
2018-03-18 19:05:53 +01:00
import flask_menu as menu
2018-05-14 01:40:34 +02:00
from flask_mail import Mail
2018-03-18 19:05:53 +01:00
from flask_github import GitHub
2020-01-23 01:19:04 +01:00
from flask_wtf.csrf import CSRFProtect
2021-02-02 21:05:24 +01:00
from flask_flatpages import FlatPages
2019-07-29 22:44:39 +02:00
from flask_babel import Babel
2020-12-04 23:29:10 +01:00
from flask_login import logout_user, current_user, LoginManager
2019-11-18 22:42:56 +01:00
import os, redis
2018-03-18 18:43:30 +01:00
2018-05-25 16:23:55 +02:00
app = Flask(__name__, static_folder="public/static")
2018-05-14 15:46:41 +02:00
app.config["FLATPAGES_ROOT"] = "flatpages"
app.config["FLATPAGES_EXTENSION"] = ".md"
2021-02-02 21:05:24 +01:00
app.config["FLATPAGES_MARKDOWN_EXTENSIONS"] = ["fenced_code", "tables", "codehilite", 'toc']
app.config["FLATPAGES_EXTENSION_CONFIG"] = {
"fenced_code": {},
"tables": {},
"codehilite": {
"linenums": "True"
}
}
2018-03-24 03:36:14 +01:00
app.config.from_pyfile(os.environ["FLASK_CONFIG"])
2018-03-18 18:43:30 +01:00
2019-11-18 22:42:56 +01:00
r = redis.Redis.from_url(app.config["REDIS_URL"])
2018-03-18 19:05:53 +01:00
menu.Menu(app=app)
github = GitHub(app)
2020-01-23 01:19:04 +01:00
csrf = CSRFProtect(app)
2018-05-14 01:40:34 +02:00
mail = Mail(app)
2018-05-14 15:46:41 +02:00
pages = FlatPages(app)
2019-07-29 22:44:39 +02:00
babel = Babel(app)
2018-12-22 21:13:43 +01:00
gravatar = Gravatar(app,
size=58,
rating="g",
default="mp",
2018-12-22 21:13:43 +01:00
force_default=False,
force_lower=False,
use_ssl=True,
base_url=None)
2018-03-18 19:05:53 +01:00
2020-12-04 23:29:10 +01:00
login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = "users.login"
from .sass import sass
sass(app)
if not app.debug and app.config["MAIL_UTILS_ERROR_SEND_TO"]:
2020-12-07 19:06:34 +01:00
from .maillogger import build_handler
app.logger.addHandler(build_handler(app))
2018-07-06 23:52:19 +02:00
2019-07-29 22:44:39 +02:00
2021-02-02 21:05:24 +01:00
from app.utils.markdown import init_app
2020-01-22 23:10:02 +01:00
init_app(app)
2020-01-22 23:11:01 +01:00
# @babel.localeselector
# def get_locale():
# return request.accept_languages.best_match(app.config["LANGUAGES"].keys())
2020-12-07 19:06:34 +01:00
from . import models, template_filters
2020-12-04 23:29:10 +01:00
@login_manager.user_loader
def load_user(user_id):
return models.User.query.filter_by(username=user_id).first()
from .blueprints import create_blueprints
create_blueprints(app)
@app.route("/uploads/<path:path>")
def send_upload(path):
return send_from_directory(app.config["UPLOAD_DIR"], path)
2019-07-29 22:44:39 +02:00
@menu.register_menu(app, ".help", "Help", order=19, endpoint_arguments_constructor=lambda: { "path": "help" })
@app.route("/<path:path>/")
def flatpage(path):
2020-12-04 03:23:04 +01:00
page = pages.get_or_404(path)
template = page.meta.get("template", "flatpage.html")
2020-12-04 03:23:04 +01:00
return render_template(template, page=page)
2019-07-29 22:44:39 +02:00
@app.before_request
def check_for_ban():
if current_user.is_authenticated:
if current_user.rank == models.UserRank.BANNED:
2020-01-24 19:15:09 +01:00
flash("You have been banned.", "danger")
logout_user()
return redirect(url_for("users.login"))
elif current_user.rank == models.UserRank.NOT_JOINED:
current_user.rank = models.UserRank.MEMBER
models.db.session.commit()
from .utils import clearNotifications
@app.before_request
def check_for_notifications():
if current_user.is_authenticated:
clearNotifications(request.path)