67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
|
"""
|
||
|
|
故事相关ORM模型
|
||
|
|
"""
|
||
|
|
from sqlalchemy import Column, Integer, String, Text, Boolean, TIMESTAMP, ForeignKey
|
||
|
|
from sqlalchemy.orm import relationship
|
||
|
|
from sqlalchemy.sql import func
|
||
|
|
from app.database import Base
|
||
|
|
|
||
|
|
|
||
|
|
class Story(Base):
|
||
|
|
"""故事主表"""
|
||
|
|
__tablename__ = "stories"
|
||
|
|
|
||
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
|
|
title = Column(String(100), nullable=False)
|
||
|
|
cover_url = Column(String(255), default="")
|
||
|
|
description = Column(Text)
|
||
|
|
author_id = Column(Integer, default=0)
|
||
|
|
category = Column(String(50), nullable=False)
|
||
|
|
play_count = Column(Integer, default=0)
|
||
|
|
like_count = Column(Integer, default=0)
|
||
|
|
is_featured = Column(Boolean, default=False)
|
||
|
|
status = Column(Integer, default=1)
|
||
|
|
created_at = Column(TIMESTAMP, server_default=func.now())
|
||
|
|
updated_at = Column(TIMESTAMP, server_default=func.now(), onupdate=func.now())
|
||
|
|
|
||
|
|
nodes = relationship("StoryNode", back_populates="story", cascade="all, delete-orphan")
|
||
|
|
|
||
|
|
|
||
|
|
class StoryNode(Base):
|
||
|
|
"""故事节点表"""
|
||
|
|
__tablename__ = "story_nodes"
|
||
|
|
|
||
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
|
|
story_id = Column(Integer, ForeignKey("stories.id", ondelete="CASCADE"), nullable=False)
|
||
|
|
node_key = Column(String(50), nullable=False)
|
||
|
|
content = Column(Text, nullable=False)
|
||
|
|
speaker = Column(String(50), default="")
|
||
|
|
background_image = Column(String(255), default="")
|
||
|
|
character_image = Column(String(255), default="")
|
||
|
|
bgm = Column(String(255), default="")
|
||
|
|
is_ending = Column(Boolean, default=False)
|
||
|
|
ending_name = Column(String(100), default="")
|
||
|
|
ending_score = Column(Integer, default=0)
|
||
|
|
ending_type = Column(String(20), default="")
|
||
|
|
sort_order = Column(Integer, default=0)
|
||
|
|
created_at = Column(TIMESTAMP, server_default=func.now())
|
||
|
|
|
||
|
|
story = relationship("Story", back_populates="nodes")
|
||
|
|
choices = relationship("StoryChoice", back_populates="node", cascade="all, delete-orphan")
|
||
|
|
|
||
|
|
|
||
|
|
class StoryChoice(Base):
|
||
|
|
"""故事选项表"""
|
||
|
|
__tablename__ = "story_choices"
|
||
|
|
|
||
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
||
|
|
node_id = Column(Integer, ForeignKey("story_nodes.id", ondelete="CASCADE"), nullable=False)
|
||
|
|
story_id = Column(Integer, nullable=False)
|
||
|
|
text = Column(String(200), nullable=False)
|
||
|
|
next_node_key = Column(String(50), nullable=False)
|
||
|
|
sort_order = Column(Integer, default=0)
|
||
|
|
is_locked = Column(Boolean, default=False)
|
||
|
|
created_at = Column(TIMESTAMP, server_default=func.now())
|
||
|
|
|
||
|
|
node = relationship("StoryNode", back_populates="choices")
|