mirror of
https://github.com/aljazceru/CTFd.git
synced 2025-12-18 06:24:23 +01:00
140 lines
5.0 KiB
Python
140 lines
5.0 KiB
Python
import sys
|
|
import os
|
|
|
|
from distutils.version import StrictVersion
|
|
from flask import Flask
|
|
from jinja2 import FileSystemLoader
|
|
from sqlalchemy.engine.url import make_url
|
|
from sqlalchemy.exc import OperationalError, ProgrammingError
|
|
from sqlalchemy_utils import database_exists, create_database
|
|
from six.moves import input
|
|
|
|
from CTFd.utils import cache, migrate, migrate_upgrade, migrate_stamp
|
|
from CTFd import utils
|
|
|
|
# Hack to support Unicode in Python 2 properly
|
|
if sys.version_info[0] < 3:
|
|
reload(sys)
|
|
sys.setdefaultencoding("utf-8")
|
|
|
|
__version__ = '1.0.2'
|
|
|
|
|
|
class ThemeLoader(FileSystemLoader):
|
|
def __init__(self, searchpath, encoding='utf-8', followlinks=False):
|
|
super(ThemeLoader, self).__init__(searchpath, encoding, followlinks)
|
|
self.overriden_templates = {}
|
|
|
|
def get_source(self, environment, template):
|
|
# Check if the template has been overriden
|
|
if template in self.overriden_templates:
|
|
return self.overriden_templates[template], template, True
|
|
|
|
# Check if the template requested is for the admin panel
|
|
if template.startswith('admin/'):
|
|
template = template.lstrip('admin/')
|
|
template = "/".join(['admin', 'templates', template])
|
|
return super(ThemeLoader, self).get_source(environment, template)
|
|
|
|
# Load regular theme data
|
|
theme = utils.get_config('ctf_theme')
|
|
template = "/".join([theme, 'templates', template])
|
|
return super(ThemeLoader, self).get_source(environment, template)
|
|
|
|
|
|
def create_app(config='CTFd.config.Config'):
|
|
app = Flask(__name__)
|
|
with app.app_context():
|
|
app.config.from_object(config)
|
|
app.jinja_loader = ThemeLoader(os.path.join(app.root_path, 'themes'), followlinks=True)
|
|
|
|
from CTFd.models import db, Teams, Solves, Challenges, WrongKeys, Keys, Tags, Files, Tracking
|
|
|
|
url = make_url(app.config['SQLALCHEMY_DATABASE_URI'])
|
|
if url.drivername == 'postgres':
|
|
url.drivername = 'postgresql'
|
|
|
|
if url.drivername.startswith('mysql'):
|
|
url.query['charset'] = 'utf8mb4'
|
|
|
|
# Creates database if the database database does not exist
|
|
if not database_exists(url):
|
|
if url.drivername.startswith('mysql'):
|
|
create_database(url, encoding='utf8mb4')
|
|
else:
|
|
create_database(url)
|
|
|
|
# This allows any changes to the SQLALCHEMY_DATABASE_URI to get pushed back in
|
|
# This is mostly so we can force MySQL's charset
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = str(url)
|
|
|
|
# Register database
|
|
db.init_app(app)
|
|
|
|
# Register Flask-Migrate
|
|
migrate.init_app(app, db)
|
|
|
|
# This creates tables instead of db.create_all()
|
|
# Allows migrations to happen properly
|
|
migrate_upgrade()
|
|
|
|
# Alembic sqlite support is lacking so we should just create_all anyway
|
|
if url.drivername.startswith('sqlite'):
|
|
db.create_all()
|
|
|
|
app.db = db
|
|
|
|
cache.init_app(app)
|
|
app.cache = cache
|
|
|
|
version = utils.get_config('ctf_version')
|
|
|
|
if not version: # Upgrading from an unversioned CTFd
|
|
utils.set_config('ctf_version', __version__)
|
|
|
|
if version and (StrictVersion(version) < StrictVersion(__version__)): # Upgrading from an older version of CTFd
|
|
print("/*\\ CTFd has updated and must update the database! /*\\")
|
|
print("/*\\ Please backup your database before proceeding! /*\\")
|
|
print("/*\\ CTFd maintainers are not responsible for any data loss! /*\\")
|
|
if input('Run database migrations (Y/N)').lower().strip() == 'y':
|
|
migrate_stamp()
|
|
migrate_upgrade()
|
|
utils.set_config('ctf_version', __version__)
|
|
else:
|
|
print('/*\\ Ignored database migrations... /*\\')
|
|
exit()
|
|
|
|
if not utils.get_config('ctf_theme'):
|
|
utils.set_config('ctf_theme', 'original')
|
|
|
|
from CTFd.views import views
|
|
from CTFd.challenges import challenges
|
|
from CTFd.scoreboard import scoreboard
|
|
from CTFd.auth import auth
|
|
from CTFd.admin import admin, admin_statistics, admin_challenges, admin_pages, admin_scoreboard, admin_containers, admin_keys, admin_teams
|
|
from CTFd.utils import init_utils, init_errors, init_logs
|
|
|
|
init_utils(app)
|
|
init_errors(app)
|
|
init_logs(app)
|
|
|
|
app.register_blueprint(views)
|
|
app.register_blueprint(challenges)
|
|
app.register_blueprint(scoreboard)
|
|
app.register_blueprint(auth)
|
|
|
|
app.register_blueprint(admin)
|
|
app.register_blueprint(admin_statistics)
|
|
app.register_blueprint(admin_challenges)
|
|
app.register_blueprint(admin_teams)
|
|
app.register_blueprint(admin_scoreboard)
|
|
app.register_blueprint(admin_keys)
|
|
app.register_blueprint(admin_containers)
|
|
app.register_blueprint(admin_pages)
|
|
|
|
from CTFd.plugins import init_plugins
|
|
|
|
init_plugins(app)
|
|
|
|
return app
|