mirror of
https://gitlab.com/SIGBUS/nyaa.git
synced 2024-12-22 08:49:59 +00:00
bd419c5d39
Using Flask-Caching, we can memoize the magnet_uri method. Here, a timeout of 1 hour is chosen, though that value can be fiddled with. The cache is defined in extensions.py, but gets initialised in __init__.py.
109 lines
3.8 KiB
Python
109 lines
3.8 KiB
Python
import logging
|
|
import os
|
|
import string
|
|
|
|
import flask
|
|
from flask_assets import Bundle # noqa F401
|
|
|
|
from nyaa.api_handler import api_blueprint
|
|
from nyaa.extensions import assets, cache, db, fix_paginate, toolbar
|
|
from nyaa.template_utils import bp as template_utils_bp
|
|
from nyaa.utils import random_string
|
|
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):
|
|
random_id = random_string(8, string.ascii_uppercase + string.digits)
|
|
# Pst. Not actually unique, but don't tell anyone!
|
|
app.logger.error('Exception occurred! Unique ID: %s', random_id, exc_info=exception)
|
|
markup_source = ' '.join([
|
|
'<strong>An error occurred!</strong>',
|
|
'Debug information has been logged.',
|
|
'Please pass along this ID: <kbd>{}</kbd>'.format(random_id)
|
|
])
|
|
|
|
flask.flash(flask.Markup(markup_source), 'danger')
|
|
return flask.redirect(flask.url_for('main.home'))
|
|
|
|
# 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)
|
|
assets._named_bundles = {} # Hack to fix state carrying over in tests
|
|
main_js = Bundle('js/main.js', filters='rjsmin', output='js/main.min.js')
|
|
bs_js = Bundle('js/bootstrap-select.js', filters='rjsmin',
|
|
output='js/bootstrap-select.min.js')
|
|
assets.register('main_js', main_js)
|
|
assets.register('bs_js', bs_js)
|
|
# 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)
|
|
register_views(app)
|
|
|
|
# Pregenerate some URLs to avoid repeat url_for calls
|
|
if 'SERVER_NAME' in app.config and app.config['SERVER_NAME']:
|
|
with app.app_context():
|
|
url = flask.url_for('static', filename='img/avatar/default.png', _external=True)
|
|
app.config['DEFAULT_GRAVATAR_URL'] = url
|
|
|
|
# Cache
|
|
cache.init_app(app)
|
|
|
|
return app
|