56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
from sqlalchemy import Column, Integer, String, Text, Boolean, Date, DateTime, ForeignKey, UniqueConstraint
|
|
from sqlalchemy.orm import relationship
|
|
from app.database import Base
|
|
from app.utils.datetime import utcnow
|
|
|
|
|
|
class HabitGroup(Base):
|
|
"""习惯分组模型"""
|
|
__tablename__ = "habit_groups"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
name = Column(String(100), nullable=False)
|
|
color = Column(String(20), default="#FFB7C5")
|
|
icon = Column(String(50), default="flag")
|
|
sort_order = Column(Integer, default=0)
|
|
|
|
# 关联关系
|
|
habits = relationship("Habit", back_populates="group", order_by="Habit.created_at")
|
|
|
|
|
|
class Habit(Base):
|
|
"""习惯模型"""
|
|
__tablename__ = "habits"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
name = Column(String(200), nullable=False)
|
|
description = Column(Text, nullable=True)
|
|
group_id = Column(Integer, ForeignKey("habit_groups.id"), nullable=True)
|
|
target_count = Column(Integer, default=1)
|
|
frequency = Column(String(20), default="daily") # daily / weekly
|
|
active_days = Column(String(100), nullable=True) # JSON 数组, 如 [0,2,4] 表示周一三五
|
|
is_archived = Column(Boolean, default=False)
|
|
created_at = Column(DateTime, default=utcnow)
|
|
updated_at = Column(DateTime, default=utcnow, onupdate=utcnow)
|
|
|
|
# 关联关系
|
|
group = relationship("HabitGroup", back_populates="habits")
|
|
checkins = relationship("HabitCheckin", back_populates="habit", cascade="all, delete-orphan")
|
|
|
|
|
|
class HabitCheckin(Base):
|
|
"""习惯打卡记录模型"""
|
|
__tablename__ = "habit_checkins"
|
|
__table_args__ = (
|
|
UniqueConstraint("habit_id", "checkin_date", name="uq_habit_checkin_date"),
|
|
)
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
habit_id = Column(Integer, ForeignKey("habits.id"), nullable=False)
|
|
checkin_date = Column(Date, nullable=False)
|
|
count = Column(Integer, default=0)
|
|
created_at = Column(DateTime, default=utcnow)
|
|
|
|
# 关联关系
|
|
habit = relationship("Habit", back_populates="checkins")
|