from sqlalchemy import Boolean, Column, Integer, String, ForeignKey
from sqlalchemy.orm import relationship

from database import Base
from .CustomBase import CustomBase

class Notifications(Base, CustomBase):
    __tablename__ = "notifications"
    
    id = Column(Integer, primary_key=True, index=True)
    title = Column(String)
    message = Column(String)
    is_read = Column(Boolean, default=False)

    sender_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
    sender = relationship("Users", foreign_keys=[sender_id], back_populates="sent_notifications")

    receiver_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
    receiver = relationship("Users", foreign_keys=[receiver_id], back_populates="received_notifications")