Mechanic's work place

scripts/cleanup_jobs.py

@@ -0,0 +1,67 @@
+from __future__ import annotations
+
+import asyncio
+from datetime import UTC, datetime, timedelta
+
+from sqlalchemy import delete, select
+
+from app.db.session import async_session_factory
+from app.models.car import ServiceEmployee, ServiceNotification, ServiceVisit
+
+
+async def main() -> None:
+    now = datetime.now(UTC)
+    async with async_session_factory() as session:
+        expired = (
+            await session.execute(
+                select(ServiceEmployee).where(
+                    ServiceEmployee.status == "invited",
+                    ServiceEmployee.invite_expires_at.is_not(None),
+                    ServiceEmployee.invite_expires_at <= now,
+                )
+            )
+        ).scalars()
+        expired_count = 0
+        for employee in expired:
+            employee.status = "expired"
+            employee.invite_token = None
+            expired_count += 1
+
+        abandoned_count = 0
+        abandoned = (
+            await session.execute(
+                select(ServiceNotification).where(
+                    ServiceNotification.status.in_(["failed", "retrying"]),
+                    ServiceNotification.retry_count >= 5,
+                    ServiceNotification.created_at < now - timedelta(days=1),
+                )
+            )
+        ).scalars()
+        for notification in abandoned:
+            notification.status = "abandoned"
+            abandoned_count += 1
+
+        old_notifications = await session.execute(
+            delete(ServiceNotification).where(
+                ServiceNotification.status == "abandoned",
+                ServiceNotification.created_at < now - timedelta(days=30),
+            )
+        )
+        orphan_drafts = await session.execute(
+            delete(ServiceVisit).where(
+                ServiceVisit.status == "draft",
+                ServiceVisit.created_at < now - timedelta(days=90),
+            )
+        )
+        await session.commit()
+    print(
+        "Cleanup done: "
+        f"expired_invites={expired_count}, "
+        f"abandoned_notifications={abandoned_count}, "
+        f"deleted_old_notifications={old_notifications.rowcount or 0}, "
+        f"orphan_drafts={orphan_drafts.rowcount or 0}"
+    )
+
+
+if __name__ == "__main__":
+    asyncio.run(main())