Initial commit: clean project structure
- Backend: FastAPI + SQLAlchemy + Celery (Python 3.11+) - Frontend: Vue 3 + TypeScript + Pinia + Tailwind - Admin Frontend: separate Vue 3 app for management - Docker Compose: 9 services orchestration - Specs: design prototypes, memory system PRD, product roadmap Cleanup performed: - Removed temporary debug scripts from backend root - Removed deprecated admin_app.py (embedded UI) - Removed duplicate docs from admin-frontend - Updated .gitignore for Vite cache and egg-info
This commit is contained in:
zhangtuo
232
backend/app/db/models.py
Normal file
232
backend/app/db/models.py
Normal file
@@ -0,0 +1,232 @@
|
||||
from datetime import date, datetime, time
|
||||
from uuid import uuid4
|
||||
|
||||
from sqlalchemy import (
|
||||
JSON,
|
||||
Boolean,
|
||||
Date,
|
||||
DateTime,
|
||||
Float,
|
||||
ForeignKey,
|
||||
Integer,
|
||||
String,
|
||||
Text,
|
||||
Time,
|
||||
UniqueConstraint,
|
||||
func,
|
||||
)
|
||||
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship
|
||||
|
||||
|
||||
class Base(DeclarativeBase):
|
||||
"""Declarative base."""
|
||||
|
||||
|
||||
class User(Base):
|
||||
"""User entity."""
|
||||
|
||||
__tablename__ = "users"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(255), primary_key=True) # OAuth provider user ID
|
||||
name: Mapped[str] = mapped_column(String(255), nullable=False)
|
||||
avatar_url: Mapped[str | None] = mapped_column(String(500))
|
||||
provider: Mapped[str] = mapped_column(String(50), nullable=False) # github / google
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now()
|
||||
)
|
||||
|
||||
stories: Mapped[list["Story"]] = relationship(
|
||||
"Story", back_populates="user", cascade="all, delete-orphan"
|
||||
)
|
||||
child_profiles: Mapped[list["ChildProfile"]] = relationship(
|
||||
"ChildProfile", back_populates="user", cascade="all, delete-orphan"
|
||||
)
|
||||
|
||||
|
||||
class Story(Base):
|
||||
"""Story entity."""
|
||||
|
||||
__tablename__ = "stories"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
||||
user_id: Mapped[str] = mapped_column(
|
||||
String(255), ForeignKey("users.id", ondelete="CASCADE"), nullable=False
|
||||
)
|
||||
child_profile_id: Mapped[str | None] = mapped_column(
|
||||
String(36), ForeignKey("child_profiles.id", ondelete="SET NULL"), nullable=True
|
||||
)
|
||||
universe_id: Mapped[str | None] = mapped_column(
|
||||
String(36), ForeignKey("story_universes.id", ondelete="SET NULL"), nullable=True
|
||||
)
|
||||
title: Mapped[str] = mapped_column(String(255), nullable=False)
|
||||
story_text: Mapped[str] = mapped_column(Text, nullable=True) # 允许为空（绘本模式下）
|
||||
pages: Mapped[list[dict] | None] = mapped_column(JSON, default=list) # 绘本分页数据
|
||||
cover_prompt: Mapped[str | None] = mapped_column(Text)
|
||||
image_url: Mapped[str | None] = mapped_column(String(500))
|
||||
mode: Mapped[str] = mapped_column(String(20), nullable=False, default="generated")
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now()
|
||||
)
|
||||
|
||||
user: Mapped["User"] = relationship("User", back_populates="stories")
|
||||
child_profile: Mapped["ChildProfile | None"] = relationship("ChildProfile")
|
||||
story_universe: Mapped["StoryUniverse | None"] = relationship("StoryUniverse")
|
||||
|
||||
|
||||
def _uuid() -> str:
|
||||
return str(uuid4())
|
||||
|
||||
|
||||
class ChildProfile(Base):
|
||||
"""Child profile entity."""
|
||||
|
||||
__tablename__ = "child_profiles"
|
||||
__table_args__ = (UniqueConstraint("user_id", "name", name="uq_child_profile_user_name"),)
|
||||
|
||||
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=_uuid)
|
||||
user_id: Mapped[str] = mapped_column(
|
||||
String(255), ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
name: Mapped[str] = mapped_column(String(50), nullable=False)
|
||||
avatar_url: Mapped[str | None] = mapped_column(String(500))
|
||||
birth_date: Mapped[date | None] = mapped_column(Date)
|
||||
gender: Mapped[str | None] = mapped_column(String(10))
|
||||
|
||||
interests: Mapped[list[str]] = mapped_column(JSON, default=list)
|
||||
growth_themes: Mapped[list[str]] = mapped_column(JSON, default=list)
|
||||
reading_preferences: Mapped[dict] = mapped_column(JSON, default=dict)
|
||||
|
||||
stories_count: Mapped[int] = mapped_column(Integer, default=0)
|
||||
total_reading_time: Mapped[int] = mapped_column(Integer, default=0)
|
||||
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now()
|
||||
)
|
||||
updated_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
|
||||
)
|
||||
|
||||
user: Mapped["User"] = relationship("User", back_populates="child_profiles")
|
||||
story_universes: Mapped[list["StoryUniverse"]] = relationship(
|
||||
"StoryUniverse", back_populates="child_profile", cascade="all, delete-orphan"
|
||||
)
|
||||
|
||||
@property
|
||||
def age(self) -> int | None:
|
||||
if not self.birth_date:
|
||||
return None
|
||||
today = date.today()
|
||||
return today.year - self.birth_date.year - (
|
||||
(today.month, today.day) < (self.birth_date.month, self.birth_date.day)
|
||||
)
|
||||
|
||||
|
||||
class StoryUniverse(Base):
|
||||
"""Story universe entity."""
|
||||
__tablename__ = "story_universes"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=_uuid)
|
||||
child_profile_id: Mapped[str] = mapped_column(
|
||||
String(36), ForeignKey("child_profiles.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
name: Mapped[str] = mapped_column(String(100), nullable=False)
|
||||
protagonist: Mapped[dict] = mapped_column(JSON, nullable=False)
|
||||
recurring_characters: Mapped[list] = mapped_column(JSON, default=list)
|
||||
world_settings: Mapped[dict] = mapped_column(JSON, default=dict)
|
||||
achievements: Mapped[list] = mapped_column(JSON, default=list)
|
||||
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now()
|
||||
)
|
||||
updated_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
|
||||
)
|
||||
|
||||
child_profile: Mapped["ChildProfile"] = relationship("ChildProfile", back_populates="story_universes")
|
||||
|
||||
|
||||
class ReadingEvent(Base):
|
||||
"""Reading event entity."""
|
||||
|
||||
__tablename__ = "reading_events"
|
||||
|
||||
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
||||
child_profile_id: Mapped[str] = mapped_column(
|
||||
String(36), ForeignKey("child_profiles.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
story_id: Mapped[int | None] = mapped_column(
|
||||
Integer, ForeignKey("stories.id", ondelete="SET NULL"), nullable=True, index=True
|
||||
)
|
||||
event_type: Mapped[str] = mapped_column(String(20), nullable=False)
|
||||
reading_time: Mapped[int] = mapped_column(Integer, default=0)
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now(), index=True
|
||||
)
|
||||
|
||||
class PushConfig(Base):
|
||||
"""Push configuration entity."""
|
||||
|
||||
__tablename__ = "push_configs"
|
||||
__table_args__ = (
|
||||
UniqueConstraint("child_profile_id", name="uq_push_config_child"),
|
||||
)
|
||||
|
||||
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=_uuid)
|
||||
user_id: Mapped[str] = mapped_column(
|
||||
String(255), ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
child_profile_id: Mapped[str] = mapped_column(
|
||||
String(36), ForeignKey("child_profiles.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
push_time: Mapped[time | None] = mapped_column(Time)
|
||||
push_days: Mapped[list[int]] = mapped_column(JSON, default=list)
|
||||
enabled: Mapped[bool] = mapped_column(Boolean, default=True)
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now()
|
||||
)
|
||||
updated_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now(), onupdate=func.now()
|
||||
)
|
||||
|
||||
|
||||
class PushEvent(Base):
|
||||
"""Push event entity."""
|
||||
|
||||
__tablename__ = "push_events"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=_uuid)
|
||||
user_id: Mapped[str] = mapped_column(
|
||||
String(255), ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
child_profile_id: Mapped[str] = mapped_column(
|
||||
String(36), ForeignKey("child_profiles.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
trigger_type: Mapped[str] = mapped_column(String(20), nullable=False)
|
||||
status: Mapped[str] = mapped_column(String(20), nullable=False)
|
||||
reason: Mapped[str | None] = mapped_column(Text)
|
||||
sent_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False, index=True)
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now()
|
||||
)
|
||||
|
||||
|
||||
class MemoryItem(Base):
|
||||
"""Memory item entity with time decay metadata."""
|
||||
|
||||
__tablename__ = "memory_items"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(36), primary_key=True, default=_uuid)
|
||||
child_profile_id: Mapped[str] = mapped_column(
|
||||
String(36), ForeignKey("child_profiles.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
universe_id: Mapped[str | None] = mapped_column(
|
||||
String(36), ForeignKey("story_universes.id", ondelete="SET NULL"), nullable=True, index=True
|
||||
)
|
||||
type: Mapped[str] = mapped_column(String(50), nullable=False)
|
||||
value: Mapped[dict] = mapped_column(JSON, nullable=False)
|
||||
base_weight: Mapped[float] = mapped_column(Float, default=1.0)
|
||||
last_used_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=True), server_default=func.now()
|
||||
)
|
||||
ttl_days: Mapped[int | None] = mapped_column(Integer)
|
||||
Reference in New Issue
Block a user