mirror of
https://github.com/sb745/NyaaV3.git
synced 2025-03-12 22:06:55 +02:00
87dd95f1e0
* Replace all `from nyaa import app` imports with `app = flask.current_app` (or `from flask import current_app as app` where possible) * Add a separate config object for top-level and class statements as `nyaa.extensions.config` Required because those codes don't have app context at the time of evaluation/execution. * Remove `routes.py` file and register all blueprints in `nyaa/__init__.py` * Refactor `nyaa/__init__.py` into an app factory * Update tools * Update tests (temporary, will be replaced)
84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
import logging
|
|
import os
|
|
|
|
import flask
|
|
from flask_assets import Bundle # noqa F401
|
|
|
|
from nyaa.api_handler import api_blueprint
|
|
from nyaa.extensions import assets, db, fix_paginate, toolbar
|
|
from nyaa.template_utils import bp as template_utils_bp
|
|
from nyaa.views import register_views
|
|
|
|
|
|
def create_app(config):
|
|
""" Nyaa app factory """
|
|
app = flask.Flask(__name__)
|
|
app.config.from_object(config)
|
|
|
|
# Don't refresh cookie each request
|
|
app.config['SESSION_REFRESH_EACH_REQUEST'] = False
|
|
|
|
# Debugging
|
|
if app.config['DEBUG']:
|
|
app.config['DEBUG_TB_INTERCEPT_REDIRECTS'] = False
|
|
toolbar.init_app(app)
|
|
app.logger.setLevel(logging.DEBUG)
|
|
|
|
# Forbid caching
|
|
@app.after_request
|
|
def forbid_cache(request):
|
|
request.headers['Cache-Control'] = 'no-cache, no-store, must-revalidate, max-age=0'
|
|
request.headers['Pragma'] = 'no-cache'
|
|
request.headers['Expires'] = '0'
|
|
return request
|
|
|
|
else:
|
|
app.logger.setLevel(logging.WARNING)
|
|
|
|
# Logging
|
|
if 'LOG_FILE' in app.config:
|
|
from logging.handlers import RotatingFileHandler
|
|
app.log_handler = RotatingFileHandler(
|
|
app.config['LOG_FILE'], maxBytes=10000, backupCount=1)
|
|
app.logger.addHandler(app.log_handler)
|
|
|
|
# Log errors and display a message to the user in production mdode
|
|
if not app.config['DEBUG']:
|
|
@app.errorhandler(500)
|
|
def internal_error(exception):
|
|
app.logger.error(exception)
|
|
flask.flash(flask.Markup(
|
|
'<strong>An error occurred!</strong> Debug information has been logged.'), 'danger')
|
|
return flask.redirect('/')
|
|
|
|
# Get git commit hash
|
|
app.config['COMMIT_HASH'] = None
|
|
master_head = os.path.abspath(os.path.join(os.path.dirname(__file__),
|
|
'../.git/refs/heads/master'))
|
|
if os.path.isfile(master_head):
|
|
with open(master_head, 'r') as head:
|
|
app.config['COMMIT_HASH'] = head.readline().strip()
|
|
|
|
# Enable the jinja2 do extension.
|
|
app.jinja_env.add_extension('jinja2.ext.do')
|
|
app.jinja_env.lstrip_blocks = True
|
|
app.jinja_env.trim_blocks = True
|
|
|
|
# Database
|
|
fix_paginate() # This has to be before the database is initialized
|
|
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
|
app.config['MYSQL_DATABASE_CHARSET'] = 'utf8mb4'
|
|
db.init_app(app)
|
|
|
|
# Assets
|
|
assets.init_app(app)
|
|
# css = Bundle('style.scss', filters='libsass',
|
|
# output='style.css', depends='**/*.scss')
|
|
# assets.register('style_all', css)
|
|
|
|
# Blueprints
|
|
app.register_blueprint(template_utils_bp)
|
|
app.register_blueprint(api_blueprint, url_prefix='/api')
|
|
register_views(app)
|
|
|
|
return app
|