Merge branch 'refactor/backend-architecture' into development
This commit is contained in:
131
api/app/features/memoir/models.py
Normal file
131
api/app/features/memoir/models.py
Normal file
@@ -0,0 +1,131 @@
|
||||
from sqlalchemy import (
|
||||
JSON,
|
||||
Boolean,
|
||||
Column,
|
||||
DateTime,
|
||||
ForeignKey,
|
||||
Integer,
|
||||
String,
|
||||
Text,
|
||||
)
|
||||
from sqlalchemy.orm import relationship
|
||||
|
||||
from app.core.db import Base, utc_now
|
||||
|
||||
|
||||
class Chapter(Base):
|
||||
__tablename__ = "chapters"
|
||||
|
||||
id = Column(String, primary_key=True)
|
||||
user_id = Column(String, ForeignKey("users.id"), nullable=False)
|
||||
title = Column(String, nullable=False)
|
||||
order_index = Column(Integer, nullable=False)
|
||||
status = Column(String, default="draft")
|
||||
cover_image = Column(JSON, nullable=True)
|
||||
updated_at = Column(DateTime(timezone=True), default=utc_now, onupdate=utc_now)
|
||||
category = Column(String, nullable=True)
|
||||
is_new = Column(Boolean, default=True)
|
||||
is_active = Column(Boolean, default=True)
|
||||
source_segments = Column(JSON, nullable=True)
|
||||
|
||||
user = relationship("User", back_populates="chapters")
|
||||
sections = relationship(
|
||||
"ChapterSection",
|
||||
back_populates="chapter",
|
||||
order_by="ChapterSection.order_index",
|
||||
cascade="all, delete-orphan",
|
||||
)
|
||||
images = relationship(
|
||||
"MemoirImage",
|
||||
back_populates="chapter",
|
||||
foreign_keys="MemoirImage.chapter_id",
|
||||
cascade="all, delete-orphan",
|
||||
)
|
||||
|
||||
|
||||
class ChapterSection(Base):
|
||||
__tablename__ = "chapter_sections"
|
||||
|
||||
id = Column(String, primary_key=True)
|
||||
chapter_id = Column(
|
||||
String, ForeignKey("chapters.id", ondelete="CASCADE"), nullable=False
|
||||
)
|
||||
order_index = Column(Integer, nullable=False)
|
||||
content = Column(Text, nullable=False)
|
||||
image_id = Column(
|
||||
String, ForeignKey("memoir_images.id", ondelete="SET NULL"), nullable=True
|
||||
)
|
||||
updated_at = Column(DateTime(timezone=True), default=utc_now, onupdate=utc_now)
|
||||
|
||||
chapter = relationship("Chapter", back_populates="sections")
|
||||
image_record = relationship(
|
||||
"MemoirImage",
|
||||
back_populates="section",
|
||||
uselist=False,
|
||||
foreign_keys="ChapterSection.image_id",
|
||||
cascade="all, delete-orphan",
|
||||
single_parent=True,
|
||||
)
|
||||
|
||||
|
||||
class MemoirImage(Base):
|
||||
__tablename__ = "memoir_images"
|
||||
|
||||
id = Column(String, primary_key=True)
|
||||
chapter_id = Column(
|
||||
String, ForeignKey("chapters.id", ondelete="CASCADE"), nullable=False
|
||||
)
|
||||
section_id = Column(
|
||||
String, ForeignKey("chapter_sections.id", ondelete="CASCADE"), nullable=True
|
||||
)
|
||||
order_index = Column(Integer, nullable=False, default=0)
|
||||
placeholder = Column(Text, nullable=True)
|
||||
description = Column(Text, nullable=True)
|
||||
status = Column(String, nullable=False, default="pending")
|
||||
prompt = Column(Text, nullable=True)
|
||||
url = Column(Text, nullable=True)
|
||||
storage_key = Column(Text, nullable=True)
|
||||
provider = Column(String, nullable=True)
|
||||
style = Column(String, nullable=True)
|
||||
size = Column(String, nullable=True)
|
||||
error = Column(Text, nullable=True)
|
||||
retryable = Column(Boolean, nullable=True)
|
||||
created_at = Column(DateTime(timezone=True), nullable=True)
|
||||
updated_at = Column(DateTime(timezone=True), default=utc_now, onupdate=utc_now)
|
||||
|
||||
chapter = relationship("Chapter", back_populates="images")
|
||||
section = relationship(
|
||||
"ChapterSection",
|
||||
back_populates="image_record",
|
||||
foreign_keys="ChapterSection.image_id",
|
||||
)
|
||||
|
||||
|
||||
class Book(Base):
|
||||
__tablename__ = "books"
|
||||
|
||||
id = Column(String, primary_key=True)
|
||||
user_id = Column(String, ForeignKey("users.id"), nullable=False)
|
||||
title = Column(String, nullable=False)
|
||||
total_pages = Column(Integer, default=0)
|
||||
total_words = Column(Integer, default=0)
|
||||
cover_image_url = Column(String, nullable=True)
|
||||
updated_at = Column(DateTime(timezone=True), default=utc_now, onupdate=utc_now)
|
||||
has_update = Column(Boolean, default=False)
|
||||
last_update_chapter_id = Column(String, nullable=True)
|
||||
|
||||
user = relationship("User", back_populates="books")
|
||||
|
||||
|
||||
class MemoirState(Base):
|
||||
__tablename__ = "memoir_states"
|
||||
|
||||
id = Column(String, primary_key=True)
|
||||
user_id = Column(String, ForeignKey("users.id"), unique=True, nullable=False)
|
||||
stage_order = Column(JSON, default=list)
|
||||
current_stage = Column(String, default="childhood")
|
||||
covered_stages = Column(JSON, default=list)
|
||||
slots = Column(JSON, nullable=False)
|
||||
updated_at = Column(DateTime(timezone=True), default=utc_now, onupdate=utc_now)
|
||||
|
||||
user = relationship("User", back_populates="memoir_state")
|
||||
Reference in New Issue
Block a user