Cheng Zhou
25 lines
845 B
Python
25 lines
845 B
Python
import asyncio
|
|
from datetime import datetime, timedelta
|
|
from zoneinfo import ZoneInfo
|
|
|
|
from app.crud import visit as visit_crud
|
|
from app.db.session import SessionLocal
|
|
|
|
|
|
async def _run_mark_lost_once() -> None:
|
|
async with SessionLocal() as session:
|
|
await visit_crud.mark_overdue_as_lost_global(session)
|
|
|
|
|
|
async def run_daily_lost_visit_job(stop_event: asyncio.Event) -> None:
|
|
tz = ZoneInfo("Asia/Shanghai")
|
|
while not stop_event.is_set():
|
|
now = datetime.now(tz)
|
|
next_midnight = (now + timedelta(days=1)).replace(hour=0, minute=0, second=0, microsecond=0)
|
|
sleep_seconds = max(1, (next_midnight - now).total_seconds())
|
|
try:
|
|
await asyncio.wait_for(stop_event.wait(), timeout=sleep_seconds)
|
|
break
|
|
except asyncio.TimeoutError:
|
|
await _run_mark_lost_once()
|