# @Time    : 2023/8/13 20:43
# @Author  : Lan
# @File    : models.py
# @Software: PyCharm
from datetime import datetime
from typing import Optional

from tortoise import fields
from tortoise.models import Model
from tortoise.contrib.pydantic import pydantic_model_creator

from core.utils import get_now


class FileCodes(Model):
    id: Optional[int] = fields.IntField(pk=True)
    code: Optional[int] = fields.CharField(description='分享码', max_length=255, index=True, unique=True)
    prefix: Optional[str] = fields.CharField(max_length=255, description='前缀', default='')
    suffix: Optional[str] = fields.CharField(max_length=255, description='后缀', default='')
    uuid_file_name: Optional[str] = fields.CharField(max_length=255, description='uuid文件名', null=True)
    file_path: Optional[str] = fields.CharField(max_length=255, description='文件路径', null=True)
    size: Optional[int] = fields.IntField(description='文件大小', default=0)
    text: Optional[str] = fields.TextField(description='文本内容', null=True)
    expired_at: Optional[datetime] = fields.DatetimeField(null=True, description='过期时间')
    expired_count: Optional[int] = fields.IntField(description='可用次数', default=0)
    used_count: Optional[int] = fields.IntField(description='已用次数', default=0)
    created_at: Optional[datetime] = fields.DatetimeField(auto_now_add=True, description='创建时间')

    async def is_expired(self):
        # 按时间
        if self.expired_at is None:
            return False
        if self.expired_at and self.expired_count < 0:
            return self.expired_at < await get_now()
        # 按次数
        else:
            return self.expired_count <= 0

    async def get_file_path(self):
        return f"{self.file_path}/{self.uuid_file_name}"


class KeyValue(Model):
    id: Optional[int] = fields.IntField(pk=True)
    key: Optional[str] = fields.CharField(max_length=255, description='键', index=True, unique=True)
    value: Optional[str] = fields.JSONField(description='值', null=True)
    created_at: Optional[datetime] = fields.DatetimeField(auto_now_add=True, description='创建时间')


file_codes_pydantic = pydantic_model_creator(FileCodes, name='FileCodes')