init commit

app/db/repositories/family.py

@@ -0,0 +1,69 @@
+"""Family repository"""
+
+from typing import Optional, List
+from sqlalchemy.orm import Session
+from app.db.models import Family, FamilyMember, FamilyInvite
+from app.db.repositories.base import BaseRepository
+
+
+class FamilyRepository(BaseRepository[Family]):
+    """Family data access operations"""
+
+    def __init__(self, session: Session):
+        super().__init__(session, Family)
+
+    def get_by_invite_code(self, invite_code: str) -> Optional[Family]:
+        """Get family by invite code"""
+        return self.session.query(Family).filter(Family.invite_code == invite_code).first()
+
+    def get_user_families(self, user_id: int) -> List[Family]:
+        """Get all families for a user"""
+        return (
+            self.session.query(Family)
+            .join(FamilyMember)
+            .filter(FamilyMember.user_id == user_id)
+            .all()
+        )
+
+    def is_member(self, family_id: int, user_id: int) -> bool:
+        """Check if user is member of family"""
+        return (
+            self.session.query(FamilyMember)
+            .filter(
+                FamilyMember.family_id == family_id,
+                FamilyMember.user_id == user_id
+            )
+            .first() is not None
+        )
+
+    def add_member(self, family_id: int, user_id: int, role: str = "member") -> FamilyMember:
+        """Add user to family"""
+        member = FamilyMember(family_id=family_id, user_id=user_id, role=role)
+        self.session.add(member)
+        self.session.commit()
+        self.session.refresh(member)
+        return member
+
+    def remove_member(self, family_id: int, user_id: int) -> bool:
+        """Remove user from family"""
+        member = (
+            self.session.query(FamilyMember)
+            .filter(
+                FamilyMember.family_id == family_id,
+                FamilyMember.user_id == user_id
+            )
+            .first()
+        )
+        if member:
+            self.session.delete(member)
+            self.session.commit()
+            return True
+        return False
+
+    def get_invite(self, invite_code: str) -> Optional[FamilyInvite]:
+        """Get invite by code"""
+        return (
+            self.session.query(FamilyInvite)
+            .filter(FamilyInvite.invite_code == invite_code)
+            .first()
+        )