feat(compliance): data classification fields and retention policies job [FASE-3.5]
Some checks failed
Aegis CI / lint-and-test (push) Has been cancelled
Some checks failed
Aegis CI / lint-and-test (push) Has been cancelled
This commit is contained in:
53
backend/app/jobs/retention_job.py
Normal file
53
backend/app/jobs/retention_job.py
Normal file
@@ -0,0 +1,53 @@
|
||||
"""Data retention policies — scheduled cleanup of aged records."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
from datetime import datetime, timedelta, timezone
|
||||
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from app.database import SessionLocal
|
||||
from app.models.audit import AuditLog
|
||||
from app.services.notification_service import cleanup_old_notifications
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
AUDIT_LOG_RETENTION_DAYS = 730
|
||||
|
||||
|
||||
def apply_retention_policies(db: Session) -> dict[str, int]:
|
||||
"""Apply retention rules. Commits the session before returning."""
|
||||
cutoff = datetime.now(timezone.utc) - timedelta(days=AUDIT_LOG_RETENTION_DAYS)
|
||||
deleted_audit = (
|
||||
db.query(AuditLog)
|
||||
.filter(AuditLog.timestamp < cutoff)
|
||||
.delete(synchronize_session=False)
|
||||
)
|
||||
if deleted_audit:
|
||||
logger.info(
|
||||
"Retention: deleted %d audit logs older than %d days",
|
||||
deleted_audit,
|
||||
AUDIT_LOG_RETENTION_DAYS,
|
||||
)
|
||||
|
||||
deleted_notifications = cleanup_old_notifications(db, days=90)
|
||||
db.commit()
|
||||
return {
|
||||
"audit_logs_deleted": deleted_audit,
|
||||
"notifications_deleted": deleted_notifications,
|
||||
}
|
||||
|
||||
|
||||
def run_retention_job() -> None:
|
||||
"""Entry point for the daily retention scheduler job."""
|
||||
logger.info("Scheduled retention job starting...")
|
||||
db = SessionLocal()
|
||||
try:
|
||||
summary = apply_retention_policies(db)
|
||||
logger.info("Retention job finished — %s", summary)
|
||||
except Exception:
|
||||
logger.exception("Retention job failed")
|
||||
db.rollback()
|
||||
finally:
|
||||
db.close()
|
||||
Reference in New Issue
Block a user