47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
|
"""
|
||
|
|
Notification models for signal alerts.
|
||
|
|
"""
|
||
|
|
import enum
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
from sqlalchemy import (
|
||
|
|
Column, Integer, String, DateTime, Boolean, Text,
|
||
|
|
ForeignKey, Enum as SQLEnum,
|
||
|
|
)
|
||
|
|
|
||
|
|
from app.core.database import Base
|
||
|
|
|
||
|
|
|
||
|
|
class ChannelType(str, enum.Enum):
|
||
|
|
DISCORD = "discord"
|
||
|
|
TELEGRAM = "telegram"
|
||
|
|
|
||
|
|
|
||
|
|
class NotificationStatus(str, enum.Enum):
|
||
|
|
SENT = "sent"
|
||
|
|
FAILED = "failed"
|
||
|
|
|
||
|
|
|
||
|
|
class NotificationSetting(Base):
|
||
|
|
__tablename__ = "notification_settings"
|
||
|
|
|
||
|
|
id = Column(Integer, primary_key=True, index=True)
|
||
|
|
user_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
|
||
|
|
channel_type = Column(SQLEnum(ChannelType), nullable=False)
|
||
|
|
webhook_url = Column(String(500), nullable=False)
|
||
|
|
enabled = Column(Boolean, default=True)
|
||
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
||
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
||
|
|
|
||
|
|
|
||
|
|
class NotificationHistory(Base):
|
||
|
|
__tablename__ = "notification_history"
|
||
|
|
|
||
|
|
id = Column(Integer, primary_key=True, index=True)
|
||
|
|
signal_id = Column(Integer, ForeignKey("signals.id"), nullable=False, index=True)
|
||
|
|
channel_type = Column(SQLEnum(ChannelType), nullable=False)
|
||
|
|
sent_at = Column(DateTime, default=datetime.utcnow)
|
||
|
|
status = Column(SQLEnum(NotificationStatus), nullable=False)
|
||
|
|
message = Column(Text)
|
||
|
|
error_message = Column(Text, nullable=True)
|