Sergey Antropoff
48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime, timezone
|
|
|
|
from sqlalchemy import and_, or_, select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.models import Wrap, WrapStatus
|
|
from app.services.storage import storage
|
|
|
|
|
|
async def purge_expired_wraps(db: AsyncSession) -> int:
|
|
now = datetime.now(timezone.utc)
|
|
result = await db.execute(
|
|
select(Wrap).where(
|
|
and_(
|
|
Wrap.status == WrapStatus.pending,
|
|
Wrap.expires_at < now,
|
|
)
|
|
)
|
|
)
|
|
wraps = list(result.scalars().all())
|
|
count = 0
|
|
for wrap in wraps:
|
|
try:
|
|
await storage.delete(wrap.object_key)
|
|
except Exception:
|
|
pass
|
|
wrap.status = WrapStatus.expired
|
|
count += 1
|
|
if count:
|
|
await db.commit()
|
|
return count
|
|
|
|
|
|
async def cleanup_consumed_orphans(db: AsyncSession) -> int:
|
|
"""Best-effort: mark very old pending with missing logic already handled elsewhere."""
|
|
result = await db.execute(
|
|
select(Wrap).where(
|
|
or_(
|
|
Wrap.status == WrapStatus.consumed,
|
|
Wrap.status == WrapStatus.expired,
|
|
)
|
|
).limit(0)
|
|
)
|
|
_ = result
|
|
return 0
|