from datetime import datetime, timezone
from typing import List, Optional
from beanie import Document
from scheduler.constants import ScheduleJobLocker


class ScheduleJobLockerDoc(Document):
    name: ScheduleJobLocker
    locked: bool = False
    expire_time: datetime = datetime.now(timezone.utc)
    lock_time: Optional[datetime] = None
    unlock_time: Optional[datetime] = None

    class Settings:
        name = "schedule_job_locker"