core/orm/notification.py

from enum import Enum as Enumeration

from sqlalchemy import Boolean, Column, DateTime, Enum, ForeignKey, Integer, func
from sqlalchemy.dialects.postgresql import JSONB

from base.orm import Base


class NotificationType(Enumeration):
    NEW_COMMENT = 1
    NEW_REPLY = 2


class Notification(Base):
    __tablename__ = "notification"

    shout: Column = Column(ForeignKey("shout.id"), index=True)
    reaction: Column = Column(ForeignKey("reaction.id"), index=True)
    user: Column = Column(ForeignKey("user.id"), index=True)
    createdAt = Column(
        DateTime(timezone=True), nullable=False, server_default=func.now(), index=True
    )
    seen = Column(Boolean, nullable=False, default=False, index=True)
    type = Column(Enum(NotificationType), nullable=False)
    data = Column(JSONB, nullable=True)
    occurrences = Column(Integer, default=1)