78 lines
3.1 KiB
Python
78 lines
3.1 KiB
Python
"""
|
||
用户相关ORM模型
|
||
"""
|
||
from sqlalchemy import Column, Integer, String, Boolean, TIMESTAMP, ForeignKey, UniqueConstraint, JSON, Index
|
||
from sqlalchemy.sql import func
|
||
from app.database import Base
|
||
|
||
|
||
class User(Base):
|
||
"""用户表"""
|
||
__tablename__ = "users"
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
openid = Column(String(100), unique=True, nullable=False)
|
||
nickname = Column(String(100), default="")
|
||
avatar_url = Column(String(255), default="")
|
||
gender = Column(Integer, default=0)
|
||
total_play_count = Column(Integer, default=0)
|
||
total_endings = Column(Integer, default=0)
|
||
created_at = Column(TIMESTAMP, server_default=func.now())
|
||
updated_at = Column(TIMESTAMP, server_default=func.now(), onupdate=func.now())
|
||
|
||
|
||
class UserProgress(Base):
|
||
"""用户进度表"""
|
||
__tablename__ = "user_progress"
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
|
||
story_id = Column(Integer, ForeignKey("stories.id", ondelete="CASCADE"), nullable=False)
|
||
current_node_key = Column(String(50), default="start")
|
||
is_completed = Column(Boolean, default=False)
|
||
ending_reached = Column(String(100), default="")
|
||
is_liked = Column(Boolean, default=False)
|
||
is_collected = Column(Boolean, default=False)
|
||
play_count = Column(Integer, default=1)
|
||
created_at = Column(TIMESTAMP, server_default=func.now())
|
||
updated_at = Column(TIMESTAMP, server_default=func.now(), onupdate=func.now())
|
||
|
||
__table_args__ = (
|
||
UniqueConstraint('user_id', 'story_id', name='uk_user_story'),
|
||
)
|
||
|
||
|
||
class UserEnding(Base):
|
||
"""用户结局收集表"""
|
||
__tablename__ = "user_endings"
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
|
||
story_id = Column(Integer, ForeignKey("stories.id", ondelete="CASCADE"), nullable=False)
|
||
ending_name = Column(String(100), nullable=False)
|
||
ending_score = Column(Integer, default=0)
|
||
unlocked_at = Column(TIMESTAMP, server_default=func.now())
|
||
|
||
__table_args__ = (
|
||
UniqueConstraint('user_id', 'story_id', 'ending_name', name='uk_user_ending'),
|
||
)
|
||
|
||
|
||
class PlayRecord(Base):
|
||
"""游玩记录表 - 保存每次游玩的完整路径"""
|
||
__tablename__ = "play_records"
|
||
|
||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
user_id = Column(Integer, ForeignKey("users.id", ondelete="CASCADE"), nullable=False)
|
||
story_id = Column(Integer, ForeignKey("stories.id", ondelete="CASCADE"), nullable=False)
|
||
draft_id = Column(Integer, default=None) # AI草稿ID，原故事为空
|
||
ending_name = Column(String(100), nullable=False) # 结局名称
|
||
ending_type = Column(String(20), default="") # 结局类型 (good/bad/hidden/rewrite)
|
||
path_history = Column(JSON, nullable=False) # 完整的选择路径
|
||
play_duration = Column(Integer, default=0) # 游玩时长(秒)
|
||
created_at = Column(TIMESTAMP, server_default=func.now())
|
||
|
||
__table_args__ = (
|
||
Index('idx_user_story', 'user_id', 'story_id'),
|
||
)
|