from datetime import datetime

from tortoise import Model, fields

from app.schemas.spider_schema import TypeEnum


class SpiderModel(Model):
    id = fields.IntField(pk=True)
    time = fields.CharField(max_length=10, description="数据时间 yyyy-MM-dd 格式的字符", regex=r'^\d{4}-\d{2}-\d{2}$')
    targetType = fields.IntEnumField(TypeEnum, description="数据类型", source_field='target_type')
    platform = fields.CharField(max_length=255, description="平台名字")
    sourceName = fields.CharField(max_length=255, description="媒体资源名字", source_field='source_name')
    score = fields.FloatField(description="热度得分")
    createTime = fields.DatetimeField(default=datetime.now, description="创建时间", source_field='create_time')
    updateTime = fields.DatetimeField(default=datetime.now, description="更新时间", source_field='update_time')
    class Meta:
        table = "spider_data"