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


class Notification(Model, Timestamp):
    id = fields.IntField(pk=True)
    title = fields.CharField(max_length=255)
    message = fields.TextField()
    seen = fields.BooleanField(default=False)
    user = fields.ForeignKeyField(
        "models.User", related_name="notifications", on_delete=fields.CASCADE
    )

    class Meta:
        table = "notifications"