init commit

.history/app/db/models/family_20251210202255.py

@@ -0,0 +1,98 @@
+"""Family and Family-related models"""
+
+from sqlalchemy import Column, Integer, String, DateTime, Boolean, ForeignKey, Enum
+from sqlalchemy.orm import relationship
+from datetime import datetime
+from enum import Enum as PyEnum
+from app.db.database import Base
+
+
+class FamilyRole(str, PyEnum):
+    """Roles in family"""
+    OWNER = "owner"
+    MEMBER = "member"
+    RESTRICTED = "restricted"
+
+
+class Family(Base):
+    """Family model - represents a family group"""
+
+    __tablename__ = "families"
+
+    id = Column(Integer, primary_key=True)
+    owner_id = Column(Integer, ForeignKey("users.id"), nullable=False)
+    name = Column(String(255), nullable=False)
+    description = Column(String(500), nullable=True)
+    currency = Column(String(3), default="RUB")  # ISO 4217 code
+    invite_code = Column(String(20), unique=True, nullable=False, index=True)
+    
+    # Settings
+    notification_level = Column(String(50), default="all")  # all, important, none
+    accounting_period = Column(String(20), default="month")  # week, month, year
+    
+    # Status
+    is_active = Column(Boolean, default=True)
+    
+    # Timestamps
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    
+    # Relationships
+    members = relationship("FamilyMember", back_populates="family", cascade="all, delete-orphan")
+    invites = relationship("FamilyInvite", back_populates="family", cascade="all, delete-orphan")
+    accounts = relationship("Account", back_populates="family", cascade="all, delete-orphan")
+    categories = relationship("Category", back_populates="family", cascade="all, delete-orphan")
+    budgets = relationship("Budget", back_populates="family", cascade="all, delete-orphan")
+    goals = relationship("Goal", back_populates="family", cascade="all, delete-orphan")
+
+    def __repr__(self) -> str:
+        return f"<Family(id={self.id}, name={self.name}, currency={self.currency})>"
+
+
+class FamilyMember(Base):
+    """Family member model - user membership in family"""
+
+    __tablename__ = "family_members"
+
+    id = Column(Integer, primary_key=True)
+    family_id = Column(Integer, ForeignKey("families.id"), nullable=False, index=True)
+    user_id = Column(Integer, ForeignKey("users.id"), nullable=False, index=True)
+    role = Column(Enum(FamilyRole), default=FamilyRole.MEMBER)
+    
+    # Permissions
+    can_edit_budget = Column(Boolean, default=True)
+    can_manage_members = Column(Boolean, default=False)
+    
+    # Timestamps
+    joined_at = Column(DateTime, default=datetime.utcnow, nullable=False)
+    
+    # Relationships
+    family = relationship("Family", back_populates="members")
+    user = relationship("User", back_populates="family_members")
+
+    def __repr__(self) -> str:
+        return f"<FamilyMember(family_id={self.family_id}, user_id={self.user_id}, role={self.role})>"
+
+
+class FamilyInvite(Base):
+    """Family invite model - pending invitations"""
+
+    __tablename__ = "family_invites"
+
+    id = Column(Integer, primary_key=True)
+    family_id = Column(Integer, ForeignKey("families.id"), nullable=False, index=True)
+    invite_code = Column(String(20), unique=True, nullable=False, index=True)
+    created_by_id = Column(Integer, ForeignKey("users.id"), nullable=False)
+    
+    # Invite validity
+    is_active = Column(Boolean, default=True)
+    expires_at = Column(DateTime, nullable=True)
+    
+    # Timestamps
+    created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
+    
+    # Relationships
+    family = relationship("Family", back_populates="invites")
+
+    def __repr__(self) -> str:
+        return f"<FamilyInvite(id={self.id}, family_id={self.family_id})>"