init commit

app/services/finance/budget_service.py

@@ -0,0 +1,67 @@
+"""Budget service"""
+
+from typing import Optional, List
+from datetime import datetime
+from sqlalchemy.orm import Session
+from app.db.repositories import BudgetRepository, TransactionRepository, CategoryRepository
+from app.db.models import Budget, TransactionType
+from app.schemas import BudgetCreateSchema
+
+
+class BudgetService:
+    """Service for budget operations"""
+
+    def __init__(self, session: Session):
+        self.session = session
+        self.budget_repo = BudgetRepository(session)
+        self.transaction_repo = TransactionRepository(session)
+        self.category_repo = CategoryRepository(session)
+
+    def create_budget(self, family_id: int, data: BudgetCreateSchema) -> Budget:
+        """Create new budget"""
+        return self.budget_repo.create(
+            family_id=family_id,
+            name=data.name,
+            limit_amount=data.limit_amount,
+            period=data.period,
+            alert_threshold=data.alert_threshold,
+            category_id=data.category_id,
+            start_date=data.start_date,
+        )
+
+    def get_budget_status(self, budget_id: int) -> dict:
+        """Get budget status with spent amount and percentage"""
+        budget = self.budget_repo.get_by_id(budget_id)
+        if not budget:
+            return {}
+
+        spent_percent = (budget.spent_amount / budget.limit_amount * 100) if budget.limit_amount > 0 else 0
+        remaining = budget.limit_amount - budget.spent_amount
+        is_exceeded = spent_percent > 100
+        is_warning = spent_percent >= budget.alert_threshold
+
+        return {
+            "budget_id": budget.id,
+            "name": budget.name,
+            "limit": budget.limit_amount,
+            "spent": budget.spent_amount,
+            "remaining": remaining,
+            "spent_percent": spent_percent,
+            "is_exceeded": is_exceeded,
+            "is_warning": is_warning,
+            "alert_threshold": budget.alert_threshold,
+        }
+
+    def get_family_budget_status(self, family_id: int) -> List[dict]:
+        """Get status of all budgets in family"""
+        budgets = self.budget_repo.get_family_budgets(family_id)
+        return [self.get_budget_status(budget.id) for budget in budgets]
+
+    def check_budget_exceeded(self, budget_id: int) -> bool:
+        """Check if budget limit exceeded"""
+        status = self.get_budget_status(budget_id)
+        return status.get("is_exceeded", False)
+
+    def reset_budget(self, budget_id: int) -> Optional[Budget]:
+        """Reset budget spent amount for new period"""
+        return self.budget_repo.update(budget_id, spent_amount=0.0)