"""Run hourly: python -m app.answer_cleanup (same server-only configuration). Removes unreferenced answer images
and worksheets that no assignment uses."""
from pathlib import Path

from dotenv import load_dotenv

from .answer_scans import AnswerScans, ScanError
from .assignments import Assignments, AssignmentsUnavailable
from .auth import load_auth
from .worksheets import Worksheets, WorksheetsUnavailable, cleanup_unused


def main():
    load_dotenv(Path(__file__).resolve().parent.parent / '.env')
    auth = load_auth()
    failed = []
    try:
        # Rollback disables student access, not deletion of orphaned private data.
        removed = AnswerScans(auth, enabled=True).cleanup()
        print(f'Removed {removed} unreferenced answer images.')
    except ScanError as exc:
        failed.append('answers: ' + exc.code)
    try:   # worksheets uploaded a day ago that no assignment uses (app/worksheets.py)
        removed = cleanup_unused(Worksheets(auth), Assignments(auth))
        print(f'Removed {removed} unused worksheets.')
    except (WorksheetsUnavailable, AssignmentsUnavailable) as exc:
        failed.append('worksheets: ' + str(exc))
    if failed:
        raise SystemExit('Cleanup failed: ' + '; '.join(failed))


if __name__ == '__main__':
    main()