diff --git a/app/config.py b/app/config.py index 6b9d2e8..f675d33 100644 --- a/app/config.py +++ b/app/config.py @@ -26,6 +26,7 @@ def env_bool(name: str, default: bool = False) -> bool: class Config: + DEBUG = env_bool("DEBUG", False) SECRET_KEY = os.environ.get("SECRET_KEY") or secrets.token_hex(32) SECRET_KEY_GENERATED = not bool(os.environ.get("SECRET_KEY")) SQLALCHEMY_DATABASE_URI = os.environ.get( diff --git a/docker-compose.yml b/docker-compose.yml index a392106..77f1262 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -10,6 +10,7 @@ services: ADMIN_USERNAME: ${ADMIN_USERNAME:-admin} ADMIN_PASSWORD: ${ADMIN_PASSWORD:?Set ADMIN_PASSWORD in .env} DATABASE_URL: ${DATABASE_URL:-sqlite:////app/data/pipe_survival_0331.db} + DEBUG: ${DEBUG:-false} APP_TIMEZONE: ${APP_TIMEZONE:-Asia/Shanghai} MAX_UPLOAD_BYTES: ${MAX_UPLOAD_BYTES:-16777216} MODEL_PATH: ${MODEL_PATH:-/app/my_survival_forest_model_quxi-10-0331.joblib} diff --git a/main.py b/main.py index c502874..d95f1b1 100644 --- a/main.py +++ b/main.py @@ -1,7 +1,8 @@ from app import create_app +from app.config import Config app = create_app() if __name__ == "__main__": - app.run(host="0.0.0.0", port=5005, debug=True, use_reloader=False) + app.run(host="0.0.0.0", port=5005, debug=Config.DEBUG, use_reloader=False)