58 lines
2.4 KiB
Python
58 lines
2.4 KiB
Python
|
import os
|
|||
|
|
|
|||
|
|
from django.apps import AppConfig
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AppConfig(AppConfig):
|
|||
|
|
default_auto_field = 'django.db.models.BigAutoField'
|
|||
|
|
name = 'app'
|
|||
|
|
|
|||
|
|
def ready(self):
|
|||
|
|
# SQLite 优化：每个新连接执行 PRAGMA
|
|||
|
|
# - journal_mode=DELETE：传统 rollback journal 模式，只产生一个 monitor.sqlite3 文件
|
|||
|
|
# - busy_timeout=5000：写冲突等 5s 而非立即报错
|
|||
|
|
# - cache_size=-65536：64MB 页缓存
|
|||
|
|
# - temp_store=MEMORY：临时表用内存
|
|||
|
|
from django.db.backends.signals import connection_created
|
|||
|
|
|
|||
|
|
def _setup_sqlite_pragma(sender, connection, **kwargs):
|
|||
|
|
if connection.vendor != 'sqlite':
|
|||
|
|
return
|
|||
|
|
with connection.cursor() as cur:
|
|||
|
|
cur.execute('PRAGMA journal_mode=DELETE;')
|
|||
|
|
cur.execute('PRAGMA busy_timeout=5000;')
|
|||
|
|
cur.execute('PRAGMA cache_size=-65536;')
|
|||
|
|
cur.execute('PRAGMA temp_store=MEMORY;')
|
|||
|
|
|
|||
|
|
connection_created.connect(_setup_sqlite_pragma)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
from app.utils.schema_upgrade import ensure_biz_algorithm_line_count_columns
|
|||
|
|
ensure_biz_algorithm_line_count_columns()
|
|||
|
|
except Exception as e:
|
|||
|
|
import logging
|
|||
|
|
logging.getLogger("app.bootstrap").warning("schema upgrade: %s", e)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
from django.contrib.auth.models import Group
|
|||
|
|
for role in ("system_admin", "algorithm_admin", "operator", "viewer"):
|
|||
|
|
Group.objects.get_or_create(name=role)
|
|||
|
|
from app.utils.Credentials import migrate_existing_credentials
|
|||
|
|
changed = migrate_existing_credentials()
|
|||
|
|
if changed:
|
|||
|
|
import logging
|
|||
|
|
logging.getLogger("app.bootstrap").info("encrypted credentials in %d database rows", changed)
|
|||
|
|
except Exception as e:
|
|||
|
|
import logging
|
|||
|
|
logging.getLogger("app.bootstrap").warning("security data upgrade: %s", e)
|
|||
|
|
|
|||
|
|
mode = os.environ.get("MONITOR_SERVICE_MODE", "disabled").strip().lower()
|
|||
|
|
# Backward-compatible opt-in, still protected by the cross-process lock.
|
|||
|
|
if mode == "disabled" and os.environ.get("MONITOR_BOOTSTRAP_SERVICES", "").lower() in (
|
|||
|
|
"1", "true", "yes", "on"
|
|||
|
|
):
|
|||
|
|
mode = "embedded"
|
|||
|
|
if mode == "embedded":
|
|||
|
|
from app.services.lifecycle import get_service_manager
|
|||
|
|
get_service_manager().start()
|