check_env.py

import os
from pathlib import Path
from dotenv import load_dotenv
import sys

BASE_DIR = Path(__file__).resolve().parent


env_name = os.getenv("DJANGO_ENV", ".env.sample")
env_file = BASE_DIR / f".env.{env_name}"

if not env_file.exists():
    print(f"❌ Missing env file: {env_file}")
    sys.exit(1)

print(f"🔍 Loading environment: {env_file}")
load_dotenv(env_file)

# 必填项（按环境可扩展）
required_vars = [
    "DJANGO_SECRET_KEY",
    "DEBUG",
    "DB_NAME",
    "DB_USER",
    "DB_PASSWORD",
    "DB_HOST",
    "DB_PORT",
    "ALLOWED_HOSTS",
    "CSRF_TRUSTED_ORIGINS",
]

missing = [var for var in required_vars if not os.getenv(var)]

if missing:
    print(f"❌ Missing required variables: {', '.join(missing)}")
    sys.exit(1)

print("✅ All required env variables are set.")