from tortoise import fields
from tortoise.models import Model
from .timestamp import Timestamp


class Code(Model, Timestamp):
    id = fields.IntField(pk=True)
    usage_count = fields.IntField(default=0)
    code = fields.CharField(max_length=6, unique=True)
    message = fields.CharField(max_length=255, null=True)

    class Meta:
        table = "code"