77 lines
3.1 KiB
Python
77 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)
|
|
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'),
|
|
)
|