File size: 4,041 Bytes
be6ee20 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 | # IDENTITY: backend/models.py
# VERSION: 1.6.0
# ⚙️ GEAR 1.2: Database Models (Entities) - PostgreSQL Optimized
from sqlalchemy import Column, Integer, String, Float, Boolean, ForeignKey, JSON, Text
from sqlalchemy.orm import relationship
try:
from database import Base
except ImportError:
from .database import Base
class User(Base):
__tablename__ = "users"
id = Column(Integer, primary_key=True, index=True)
username = Column(String(100), unique=True, index=True)
role = Column(String(50), default="Student")
sensory_mode = Column(String(50), default="Standard")
difficulty = Column(String(50), default="Medium (Standard)")
ai_persona = Column(String(100), default="Standard Edu_AI")
semester_status = Column(String(100), default="Year 4 - Redemption Arc")
interests = Column(JSON, default=list)
# Relationships
units = relationship("Unit", back_populates="owner", cascade="all, delete-orphan")
quiz_history = relationship("QuizHistory", back_populates="owner", cascade="all, delete-orphan")
chat_messages = relationship("ChatMessage", back_populates="owner", cascade="all, delete-orphan")
chat_sessions = relationship("ChatSession", back_populates="owner", cascade="all, delete-orphan")
performance_logs = relationship("PerformanceLog", back_populates="owner", cascade="all, delete-orphan")
timetables = relationship("Timetable", back_populates="owner", cascade="all, delete-orphan")
@property
def active_units_list(self):
return [u.name for u in self.units if u.is_active]
class Unit(Base):
__tablename__ = "units"
id = Column(Integer, primary_key=True, index=True)
name = Column(String(200), index=True)
is_active = Column(Boolean, default=True)
category = Column(String(100), default="General")
owner_id = Column(Integer, ForeignKey("users.id"))
owner = relationship("User", back_populates="units")
class QuizHistory(Base):
__tablename__ = "quiz_history"
id = Column(Integer, primary_key=True, index=True)
unit_name = Column(String(200))
score = Column(Integer)
total = Column(Integer)
pnl = Column(Float)
timestamp = Column(String(100))
owner_id = Column(Integer, ForeignKey("users.id"))
owner = relationship("User", back_populates="quiz_history")
class ChatSession(Base):
__tablename__ = "chat_sessions"
id = Column(Integer, primary_key=True, index=True)
title = Column(String(200), default="New Consultation")
description = Column(Text, nullable=True)
timestamp = Column(Float)
is_archived = Column(Boolean, default=False)
owner_id = Column(Integer, ForeignKey("users.id"))
owner = relationship("User", back_populates="chat_sessions")
messages = relationship("ChatMessage", back_populates="session", cascade="all, delete-orphan")
class ChatMessage(Base):
__tablename__ = "chat_messages"
id = Column(Integer, primary_key=True, index=True)
role = Column(String(20)) # "user" or "model"
content = Column(Text)
timestamp = Column(String(100))
owner_id = Column(Integer, ForeignKey("users.id"))
owner = relationship("User", back_populates="chat_messages")
session_id = Column(Integer, ForeignKey("chat_sessions.id"), nullable=True)
session = relationship("ChatSession", back_populates="messages")
class PerformanceLog(Base):
__tablename__ = "performance_logs"
id = Column(Integer, primary_key=True, index=True)
subject = Column(String(200))
grade = Column(Float)
timestamp = Column(String(100))
owner_id = Column(Integer, ForeignKey("users.id"))
owner = relationship("User", back_populates="performance_logs")
class Timetable(Base):
__tablename__ = "timetables"
id = Column(Integer, primary_key=True, index=True)
weekly_plan_json = Column(JSON)
ai_brief = Column(Text)
timestamp = Column(Float) # Time of generation
owner_id = Column(Integer, ForeignKey("users.id"))
owner = relationship("User", back_populates="timetables")
|