diff --git a/CTFd/__init__.py b/CTFd/__init__.py index ae42dbc..0005486 100644 --- a/CTFd/__init__.py +++ b/CTFd/__init__.py @@ -181,6 +181,17 @@ def create_app(config="CTFd.config.Config"): if url.drivername.startswith("sqlite"): db.create_all() stamp_latest_revision() + + # Enable foreign keys for SQLite + from sqlalchemy.engine import Engine + from sqlalchemy import event + + @event.listens_for(Engine, "connect") + def set_sqlite_pragma(dbapi_connection, connection_record): + cursor = dbapi_connection.cursor() + cursor.execute("PRAGMA foreign_keys=ON") + cursor.close() + else: # This creates tables instead of db.create_all() # Allows migrations to happen properly diff --git a/populate.py b/populate.py index 033d69f..07cd639 100644 --- a/populate.py +++ b/populate.py @@ -7,6 +7,7 @@ import random import argparse from CTFd import create_app +from CTFd.cache import clear_config, clear_standings, clear_pages from CTFd.models import ( Users, Teams, @@ -338,3 +339,7 @@ if __name__ == "__main__": db.session.commit() db.session.close() + + clear_config() + clear_standings() + clear_pages()