Add the categories for the own users

app/constants/categories.py

@@ -1,14 +1,17 @@
 CATEGORIES = [
-    {"name": "Groceries", "icon": "🛒", "budget": 0, "spent": 0, "remaining": 0},
-    {"name": "Food", "icon": "🍽", "budget": 0, "spent": 0, "remaining": 0},
-    {"name": "Transport", "icon": "🚌", "budget": 0, "spent": 0, "remaining": 0},
-    {"name": "Health", "icon": "💊", "budget": 0, "spent": 0, "remaining": 0},
-    {"name": "Gifts", "icon": "🎁", "budget": 0, "spent": 0, "remaining": 0},
-    {"name": "Rent", "icon": "🏠", "budget": 0, "spent": 0, "remaining": 0},
-    {"name": "Utilities", "icon": "⚡", "budget": 0, "spent": 0, "remaining": 0},
-    {"name": "Entertainment", "icon": "🎉", "budget": 0, "spent": 0, "remaining": 0},
-    {"name": "Education", "icon": "📚", "budget": 0, "spent": 0, "remaining": 0},
-    {"name": "Insurance", "icon": "🛡", "budget": 0, "spent": 0, "remaining": 0},
-]
-
+    # ---------- HEAD CATEGORIES ----------
+    {"name": "Groceries", "icon": "🛒", "scope": "head"},
+    {"name": "Food", "icon": "🍽", "scope": "head"},
+    {"name": "Transport", "icon": "🚌", "scope": "head"},
+    {"name": "Health", "icon": "💊", "scope": "head"},
+    {"name": "Gifts", "icon": "🎁", "scope": "head"},
+    {"name": "Rent", "icon": "🏠", "scope": "head"},
+    {"name": "Utilities", "icon": "⚡", "scope": "head"},
+    {"name": "Entertainment", "icon": "🎉", "scope": "head"},
+    {"name": "Education", "icon": "📚", "scope": "head"},
+    {"name": "Insurance", "icon": "🛡", "scope": "head"},
 
+    # ---------- MEMBER CATEGORIES ----------
+    {"name": "Student Groceries", "icon": "🛒", "scope": "member"},
+    {"name": "Food", "icon": "🍽", "scope": "member"},
+]

app/db/categories_budget.py

@@ -1,17 +1,27 @@
-from sqlalchemy import Column, Integer, String, Float, DateTime, ForeignKey
+from sqlalchemy import Column, Integer, String, Float, DateTime, ForeignKey, UniqueConstraint
 from sqlalchemy.orm import relationship
 from sqlalchemy.sql import func
 from app.db.database import Base
 
-
 class CategoryBudget(Base):
     __tablename__ = "category_budgets"
 
     id = Column(Integer, primary_key=True, index=True)
-    family_code = Column(String(10), index=True, nullable=False)
 
+    family_code = Column(String(10), index=True, nullable=False)
     category_name = Column(String(50), nullable=False)
+
+    scope = Column(String(10), nullable=False)   # "family" | "member"
+    owner_id = Column(Integer, nullable=True)    # NULL for family, member_id for member
+
     budget = Column(Float, default=0.0)
     spent = Column(Float, default=0.0)
 
-    created_at = Column(DateTime(timezone=True), server_default=func.now())
+    created_at = Column(DateTime(timezone=True), server_default=func.now())
+
+    __table_args__ = (
+        UniqueConstraint(
+            "family_code", "scope", "owner_id", "category_name",
+            name="uq_category_budget_scope"
+        ),
+    )

app/routers/expense.py

@@ -13,7 +13,7 @@ from app.db.categories_budget import CategoryBudget
 
 router = APIRouter(prefix="/expense", tags=["expense"])
 
-CATEGORIES = [
+HEAD_CATEGORIES = [
     {"name": "Groceries", "icon": "🛒", "budget": 0, "spent": 0, "remaining": 0},
     {"name": "Food", "icon": "🍽", "budget": 0, "spent": 0, "remaining": 0},
     {"name": "Transport", "icon": "🚌", "budget": 0, "spent": 0, "remaining": 0},
@@ -26,6 +26,14 @@ CATEGORIES = [
     {"name": "Insurance", "icon": "🛡", "budget": 0, "spent": 0, "remaining": 0},
 ]
 
+MEMBER_CATEGORIES = [
+    {"name": "Food", "icon": "🍽", "budget": 0, "spent": 0, "remaining": 0},
+    {"name": "Transport", "icon": "🚌", "budget": 0, "spent": 0, "remaining": 0},
+    {"name": "Entertainment", "icon": "🎉", "budget": 0, "spent": 0, "remaining": 0},
+    {"name": "Education", "icon": "📚", "budget": 0, "spent": 0, "remaining": 0},
+    {"name": "Gifts", "icon": "🎁", "budget": 0, "spent": 0, "remaining": 0},
+]
+
 
 @router.get("/categories")
 def get_categories(
@@ -34,45 +42,60 @@ def get_categories(
 ):
     family_code = current_user.family_code
 
-    # Fetch budget rows from DB
+    # ---------------- ROLE CHECK ----------------
+    if current_user.role == "member":
+        fm = db.query(FamilyMember).filter(
+            FamilyMember.family_code == family_code,
+            FamilyMember.user_id == current_user.id
+        ).first()
+
+        if not fm:
+            raise HTTPException(400, "Member record not found")
+
+        scope = "member"
+        owner_id = fm.id
+        base_categories = MEMBER_CATEGORIES
+
+    else:  # head
+        scope = "family"
+        owner_id = None
+        base_categories = HEAD_CATEGORIES
+
+    # ---------------- FETCH BUDGETS ----------------
     rows = db.query(CategoryBudget).filter(
-        CategoryBudget.family_code == family_code
+        CategoryBudget.family_code == family_code,
+        CategoryBudget.scope == scope,
+        CategoryBudget.owner_id == owner_id
     ).all()
 
-    # Map for quick lookup
+    # Safe lookup
     db_map = {row.category_name: row for row in rows}
 
+    # ---------------- BUILD RESPONSE ----------------
     result = []
 
-    for cat in CATEGORIES:
+    for cat in base_categories:
         name = cat["name"]
         icon = cat["icon"]
 
-        # Read values from DB if available
-        if name in db_map:
-            row = db_map[name]
-            budget = row.budget
-            spent = row.spent
-            remaining = budget - spent
-        else:
-            budget = 0
-            spent = 0
-            remaining = 0
+        row = db_map.get(name)
+        budget = row.budget if row else 0
+        spent = row.spent if row else 0
 
         result.append({
             "name": name,
             "icon": icon,
             "budget": budget,
             "spent": spent,
-            "remaining": remaining,
+            "remaining": max(budget - spent, 0),
         })
 
     return {
         "status": True,
+        "role": current_user.role,
         "categories": result
     }
 
-
 @router.post("/add")
 def add_expense(
     payload: AddExpenseRequest,