from tortoise import Model, fields


class Review(Model):
    id = fields.IntField(pk=True)
    external_id = fields.TextField()
    platform = fields.TextField()  # "ozon" or "wb"
    rating = fields.IntField()
    is_good = fields.BooleanField()
    text = fields.TextField()
    published_at = fields.DatetimeField()

    product_id = fields.TextField()
    product_name = fields.TextField()
    product_category = fields.TextField()
    product_subcategory = fields.TextField()

    tone = fields.TextField(null=True)  # positive, neutral, negative
    tone_reason = fields.TextField(null=True)
    is_critical = fields.BooleanField(null=True)

    response_text = fields.TextField(null=True)
    response_id = fields.TextField(null=True)
    responded_at = fields.DatetimeField(null=True)

    chat_id = fields.IntField(null=True)
    message_id = fields.IntField(null=True)
    images = fields.JSONField(null=True)  # Store image URLs as JSON array
    # WB: теги отзыва из API (bables), например ["цена"]
    bables = fields.JSONField(null=True)