Add output cleanup policy
This commit is contained in:
@@ -5,6 +5,7 @@ from werkzeug.exceptions import RequestEntityTooLarge
|
||||
|
||||
from app.config import Config
|
||||
from app.routes import main_blueprint
|
||||
from app.services.storage import cleanup_expired_outputs
|
||||
|
||||
|
||||
def _format_size_limit(size_limit_bytes: int) -> str:
|
||||
@@ -27,6 +28,12 @@ def create_app(config_class: type[Config] = Config) -> Flask:
|
||||
app.config["OUTPUT_DIRECTORY"] = output_dir
|
||||
output_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
if app.config.get("CLEANUP_ON_STARTUP", False):
|
||||
cleanup_expired_outputs(
|
||||
output_dir=output_dir,
|
||||
retention_hours=app.config.get("OUTPUT_RETENTION_HOURS", 24),
|
||||
)
|
||||
|
||||
app.register_blueprint(main_blueprint)
|
||||
|
||||
@app.errorhandler(RequestEntityTooLarge)
|
||||
|
||||
Reference in New Issue
Block a user