File size: 3,166 Bytes
b3d29a7 | 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 | from sqlalchemy import Column, Integer, String, Text, DateTime, ForeignKey, UniqueConstraint
from sqlalchemy.orm import relationship
from sqlalchemy.sql import func
from db import Base
class Student(Base):
__tablename__ = "students"
id = Column(Integer, primary_key=True, index=True)
student_id = Column(String(100), unique=True, index=True, nullable=False)
name = Column(String(200), nullable=True)
email = Column(String(200), nullable=True)
created_at = Column(DateTime(timezone=True), server_default=func.now())
class HomeworkAssignment(Base):
__tablename__ = "homework_assignments"
id = Column(Integer, primary_key=True, index=True)
homework_id = Column(String(100), unique=True, index=True, nullable=False)
title = Column(String(255), nullable=True)
description = Column(Text, nullable=True)
created_at = Column(DateTime(timezone=True), server_default=func.now())
class Submission(Base):
__tablename__ = "submissions"
id = Column(Integer, primary_key=True, index=True)
student_id = Column(String(100), index=True, nullable=False)
homework_id = Column(String(100), index=True, nullable=False)
student_ref_id = Column(Integer, ForeignKey("students.id"), nullable=True)
homework_ref_id = Column(Integer, ForeignKey("homework_assignments.id"), nullable=True)
status = Column(String(50), default="processed")
created_at = Column(DateTime(timezone=True), server_default=func.now())
student = relationship("Student", lazy="joined")
homework = relationship("HomeworkAssignment", lazy="joined")
class HomeworkImage(Base):
__tablename__ = "homework_images"
id = Column(Integer, primary_key=True, index=True)
submission_id = Column(Integer, ForeignKey("submissions.id"), nullable=True)
homework_id = Column(String(100), index=True, nullable=False)
role = Column(String(20), nullable=False) # teacher/student
filename = Column(String(255), nullable=False)
content_type = Column(String(100), nullable=True)
disk_path = Column(String(500), nullable=True)
ocr_text = Column(Text, nullable=True)
created_at = Column(DateTime(timezone=True), server_default=func.now())
class Result(Base):
__tablename__ = "results"
id = Column(Integer, primary_key=True, index=True)
submission_id = Column(Integer, ForeignKey("submissions.id"), nullable=False)
filename = Column(String(255), nullable=False)
extracted_text = Column(Text, nullable=True)
segmented_answers_json = Column(Text, nullable=True)
validation_json = Column(Text, nullable=True)
created_at = Column(DateTime(timezone=True), server_default=func.now())
__table_args__ = (
UniqueConstraint("submission_id", "filename", name="uq_result_submission_filename"),
)
class AuditLog(Base):
__tablename__ = "audit_logs"
id = Column(Integer, primary_key=True, index=True)
submission_id = Column(Integer, ForeignKey("submissions.id"), nullable=True)
level = Column(String(20), default="INFO")
message = Column(Text, nullable=False)
created_at = Column(DateTime(timezone=True), server_default=func.now())
|