69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
from sqlalchemy import Column, String, Boolean, Text, DateTime, Integer
|
|
from datetime import datetime
|
|
from db import Base
|
|
|
|
class User(Base):
|
|
__tablename__ = "users"
|
|
id = Column(String, primary_key=True)
|
|
email = Column(String, unique=True)
|
|
name = Column(String)
|
|
twofa_enabled = Column(Boolean, default=False)
|
|
twofa_method = Column(String, default="totp")
|
|
twofa_secret = Column(String)
|
|
twofa_sensitive_actions = Column(Boolean, default=False)
|
|
|
|
class Preferences(Base):
|
|
__tablename__ = "preferences"
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
user_id = Column(String)
|
|
preferred_mode = Column(String, default="conversar")
|
|
auto_search = Column(Boolean, default=True)
|
|
auto_monitoring = Column(Boolean, default=False)
|
|
preferred_export = Column(String, default="pdfa")
|
|
likes_suggestions = Column(Boolean, default=True)
|
|
likes_quick_actions = Column(Boolean, default=True)
|
|
|
|
class Conversation(Base):
|
|
__tablename__ = "conversations"
|
|
id = Column(String, primary_key=True)
|
|
user_id = Column(String)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
mode = Column(String)
|
|
context_summary = Column(Text)
|
|
|
|
class Message(Base):
|
|
__tablename__ = "messages"
|
|
id = Column(String, primary_key=True)
|
|
conversation_id = Column(String)
|
|
role = Column(String)
|
|
content = Column(Text)
|
|
metadata = Column(Text)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
class MonitoringTopic(Base):
|
|
__tablename__ = "monitoring_topics"
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
user_id = Column(String)
|
|
label = Column(String)
|
|
sources = Column(Text)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
active = Column(Boolean, default=True)
|
|
|
|
class Alert(Base):
|
|
__tablename__ = "alerts"
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
user_id = Column(String)
|
|
monitoring_topic_id = Column(Integer)
|
|
title = Column(String)
|
|
description = Column(Text)
|
|
source = Column(String)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
class AuditTrail(Base):
|
|
__tablename__ = "audit_trail"
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
user_email = Column(String)
|
|
action = Column(String)
|
|
details = Column(Text)
|
|
timestamp = Column(DateTime, default=datetime.utcnow)
|