from datetime import datetime

from pydantic import Field

from cbh.api.account.models import AccountShorten
from cbh.api.security.dto import VerificationCodeStatus, VerificationCodeType
from cbh.core.database import MongoBaseModel


class VerificationCodeModel(MongoBaseModel):
    account: AccountShorten
    expiresAt: datetime
    type: VerificationCodeType
    status: VerificationCodeStatus = VerificationCodeStatus.PENDING
    datetimeInserted: datetime = Field(default_factory=datetime.now)