| |
| """ |
| Manual log cleanup script for Attestation Store. |
| |
| Usage: |
| python scripts/cleanup_logs.py # Use default retention from LOG_RETENTION_DAYS |
| python scripts/cleanup_logs.py --days 30 # Delete logs older than 30 days |
| python scripts/cleanup_logs.py --dry-run # Preview what would be deleted |
| """ |
|
|
| import argparse |
| import sys |
| from datetime import datetime, timedelta |
|
|
| |
| import os |
| sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) |
|
|
| from app import app |
| from models import SystemLog |
|
|
|
|
| def cleanup_logs(retention_days=None, dry_run=False): |
| """ |
| Delete logs older than the specified retention period. |
| |
| Args: |
| retention_days (int, optional): Number of days to retain logs. |
| If not provided, uses LOG_RETENTION_DAYS from config. |
| dry_run (bool): If True, only show what would be deleted without deleting. |
| |
| Returns: |
| int: Number of log entries that were (or would be) deleted |
| """ |
| with app.app_context(): |
| from logging_utils import cleanup_old_logs |
| from flask import current_app |
| |
| |
| if retention_days is None: |
| config = current_app.config.get('STORAGE_CONFIG') |
| retention_days = config.LOG_RETENTION_DAYS if config else 10 |
| |
| |
| if retention_days <= 0: |
| print("Log cleanup disabled (retention_days <= 0)") |
| return 0 |
| |
| |
| cutoff_date = datetime.utcnow() - timedelta(days=retention_days) |
| |
| |
| count = SystemLog.query.filter(SystemLog.timestamp < cutoff_date).count() |
| |
| if dry_run: |
| print(f"DRY RUN: Would delete {count} log entries older than {retention_days} days (before {cutoff_date.isoformat()})") |
| return count |
| |
| |
| deleted = cleanup_old_logs(retention_days=retention_days) |
| print(f"Deleted {deleted} log entries older than {retention_days} days (before {cutoff_date.isoformat()})") |
| return deleted |
|
|
|
|
| def main(): |
| parser = argparse.ArgumentParser( |
| description='Clean up old logs from the Attestation Store database', |
| formatter_class=argparse.RawDescriptionHelpFormatter, |
| epilog=""" |
| Examples: |
| # Use default retention period from LOG_RETENTION_DAYS environment variable |
| python scripts/cleanup_logs.py |
| |
| # Delete logs older than 30 days |
| python scripts/cleanup_logs.py --days 30 |
| |
| # Preview what would be deleted without actually deleting |
| python scripts/cleanup_logs.py --dry-run |
| |
| # Preview deletion with custom retention period |
| python scripts/cleanup_logs.py --days 60 --dry-run |
| """ |
| ) |
| |
| parser.add_argument( |
| '--days', |
| type=int, |
| metavar='N', |
| help='Number of days to retain logs (logs older than N days will be deleted)' |
| ) |
| |
| parser.add_argument( |
| '--dry-run', |
| action='store_true', |
| help='Show what would be deleted without actually deleting' |
| ) |
| |
| args = parser.parse_args() |
| |
| try: |
| deleted = cleanup_logs(retention_days=args.days, dry_run=args.dry_run) |
| |
| if deleted == 0 and not args.dry_run: |
| print("No logs were deleted.") |
| |
| sys.exit(0) |
| |
| except Exception as e: |
| print(f"Error: {e}", file=sys.stderr) |
| sys.exit(1) |
|
|
|
|
| if __name__ == '__main__': |
| main() |
|
|
