from django.db import models

from common.models.baseModels import BaseModel
from common.models.choiceModels import CertificateCodeUseType, InviteCodeUseType
from common.utils.codeManger import set_expire


class CertificationCode(BaseModel):
    use_type = models.CharField(choices=CertificateCodeUseType.choices, max_length=5)
    code = models.CharField(max_length=6)
    expire_at = models.DateTimeField(default=set_expire(minutes=5))
    is_used = models.BooleanField(default=False)
    identifier = models.CharField(max_length=40)

class InviteCode(BaseModel):
    use_type = models.CharField(choices=InviteCodeUseType.choices, max_length=5)
    code = models.CharField(max_length=10)
    expire_at = models.DateTimeField(default=set_expire(minutes=10080)) # 일주일은 10080분
    identifier = models.CharField(max_length=40)