config.py

import os
from pathlib import Path


BASE_DIR = Path(__file__).resolve().parent
ENV_FILE = BASE_DIR / ".env"


def load_local_env() -> None:
    if not ENV_FILE.exists():
        return

    for raw_line in ENV_FILE.read_text(encoding="utf-8").splitlines():
        line = raw_line.strip()
        if not line or line.startswith("#") or "=" not in line:
            continue

        key, value = line.split("=", 1)
        key = key.strip()
        value = value.strip()
        if not key:
            continue
        if value[:1] == value[-1:] and value[:1] in {"'", '"'}:
            value = value[1:-1]
        os.environ.setdefault(key, value)


load_local_env()


def default_data_dir() -> Path:
    configured = os.getenv("COURSE_SYSTEM_DATA_DIR")
    if configured:
        return Path(configured).expanduser()
    legacy_database = BASE_DIR / "instance" / "attainment_system.db"
    if legacy_database.exists():
        return BASE_DIR
    return BASE_DIR / "var"


class BaseConfig:
    DATA_DIR = str(default_data_dir())
    SECRET_KEY = os.getenv("SECRET_KEY", "")
    SQLALCHEMY_DATABASE_URI = os.getenv(
        "DATABASE_URL",
        f"sqlite:///{Path(DATA_DIR) / 'instance' / 'attainment_system.db'}",
    )
    SQLALCHEMY_TRACK_MODIFICATIONS = False
    SESSION_COOKIE_HTTPONLY = True
    SESSION_COOKIE_SAMESITE = "Lax"
    SESSION_COOKIE_SECURE = os.getenv("SESSION_COOKIE_SECURE", "false").lower() in {"1", "true", "yes"}
    UPLOAD_FOLDER = str(Path(DATA_DIR) / "uploads")
    EXPORT_FOLDER = str(Path(DATA_DIR) / "exports")
    REPORT_FOLDER = str(Path(DATA_DIR) / "exports" / "reports")
    BACKUP_FOLDER = str(Path(DATA_DIR) / "exports" / "backups")
    REPORT_TEMPLATE_DOCX = os.getenv("REPORT_TEMPLATE_DOCX", "")
    SAMPLE_DATA_FOLDER = str(BASE_DIR / "sample_data")
    MAX_CONTENT_LENGTH = 16 * 1024 * 1024
    ALLOWED_EXTENSIONS = {"xls", "xlsx", "xlsm", "csv", "docx"}
    DEFAULT_EXPECTED_VALUE = 0.65
    DEFAULT_PAGE_SIZE = 20
    LOGIN_DISABLED = os.getenv("LOGIN_DISABLED", "false").lower() in {"1", "true", "yes"}
    DEFAULT_ADMIN_USERNAME = os.getenv("DEFAULT_ADMIN_USERNAME", "admin")
    DEFAULT_ADMIN_PASSWORD = os.getenv("DEFAULT_ADMIN_PASSWORD", "admin123")
    DEFAULT_ADMIN_DISPLAY_NAME = os.getenv("DEFAULT_ADMIN_DISPLAY_NAME", "系统管理员")
    LLM_API_BASE = os.getenv("LLM_API_BASE", os.getenv("DEEPSEEK_API_BASE", "https://api.deepseek.com"))
    LLM_API_KEY = os.getenv("LLM_API_KEY", os.getenv("DEEPSEEK_API_KEY", ""))
    LLM_MODEL = os.getenv("LLM_MODEL", os.getenv("DEEPSEEK_MODEL", "deepseek-v4-flash"))
    LLM_TIMEOUT = int(os.getenv("LLM_TIMEOUT", "60"))
    LLM_VERIFY_SSL = os.getenv("LLM_VERIFY_SSL", "true").lower() not in {"0", "false", "no"}


class DevelopmentConfig(BaseConfig):
    DEBUG = True


class TestingConfig(BaseConfig):
    TESTING = True
    SECRET_KEY = "course-attainment-test-secret"  # nosec B105
    WTF_CSRF_ENABLED = False
    LOGIN_DISABLED = True
    SQLALCHEMY_DATABASE_URI = "sqlite:///:memory:"


class ProductionConfig(BaseConfig):
    DEBUG = False