fin-infra 0.1.67__py3-none-any.whl → 0.1.68__py3-none-any.whl

fin_infra/analytics/add.py

@@ -208,12 +208,10 @@ def add_analytics(
         user_id: str,
         accounts: Optional[list[str]] = None,
         with_holdings: bool = Query(
-            False,
-            description="Use real holdings data from investment provider for accurate P/L"
+            False, description="Use real holdings data from investment provider for accurate P/L"
         ),
         access_token: Optional[str] = Query(
-            None,
-            description="Investment provider access token (required if with_holdings=true)"
+            None, description="Investment provider access token (required if with_holdings=true)"
         ),
     ) -> PortfolioMetrics:
         """
@@ -249,31 +247,31 @@ def add_analytics(
         if with_holdings:
             # Check if investment provider is available on app state
             investment_provider = getattr(app.state, "investment_provider", None)
-            
+
             if investment_provider and access_token:
                 try:
                     # Fetch real holdings from investment provider
                     from fin_infra.analytics.portfolio import portfolio_metrics_with_holdings
-                    
+
                     holdings = await investment_provider.get_holdings(
                         access_token=access_token,
                         account_ids=accounts,
                     )
-                    
+
                     # Calculate metrics from real holdings
                     return portfolio_metrics_with_holdings(holdings)
-                    
+
                 except Exception as e:
                     # Fall back to balance-only calculation on error
                     # Log error but don't fail the request
                     import logging
+
                     logging.warning(f"Failed to fetch holdings, falling back to balance-only: {e}")
             elif with_holdings and not access_token:
                 raise HTTPException(
-                    status_code=400,
-                    detail="access_token required when with_holdings=true"
+                    status_code=400, detail="access_token required when with_holdings=true"
                 )
-        
+
         # Default: Use balance-only calculation (existing behavior)
         return await provider.portfolio_metrics(
             user_id,

fin_infra/analytics/portfolio.py

@@ -567,22 +567,16 @@ def portfolio_metrics_with_holdings(holdings: list) -> PortfolioMetrics:
     # Import here to avoid circular dependency
 
     # Calculate total portfolio value and cost basis
-    total_value = float(sum(
-        holding.institution_value
-        for holding in holdings
-    ))
+    total_value = float(sum(holding.institution_value for holding in holdings))
 
-    total_cost_basis = float(sum(
-        holding.cost_basis if holding.cost_basis is not None else 0
-        for holding in holdings
-    ))
+    total_cost_basis = float(
+        sum(holding.cost_basis if holding.cost_basis is not None else 0 for holding in holdings)
+    )
 
     # Calculate total return (P/L)
     total_return_dollars = total_value - total_cost_basis
     total_return_percent = (
-        (total_return_dollars / total_cost_basis * 100.0)
-        if total_cost_basis > 0
-        else 0.0
+        (total_return_dollars / total_cost_basis * 100.0) if total_cost_basis > 0 else 0.0
     )
 
     # Calculate asset allocation from real security types
@@ -684,9 +678,7 @@ def calculate_day_change_with_snapshot(
     # Calculate day change
     day_change_dollars = current_total - previous_total
     day_change_percent = (
-        (day_change_dollars / previous_total * 100.0)
-        if previous_total > 0
-        else 0.0
+        (day_change_dollars / previous_total * 100.0) if previous_total > 0 else 0.0
     )
 
     return {
@@ -740,16 +732,18 @@ def _calculate_allocation_from_holdings(
     # Sum values by asset class
     allocation_values: dict[str, float] = defaultdict(float)
     for holding in holdings:
-        security_type = holding.security.type.value if hasattr(holding.security.type, 'value') else holding.security.type
+        security_type = (
+            holding.security.type.value
+            if hasattr(holding.security.type, "value")
+            else holding.security.type
+        )
         asset_class = type_to_class.get(security_type, "Other")
         allocation_values[asset_class] += float(holding.institution_value)
 
     # Convert to list of AssetAllocation objects
     allocation_list = [
         AssetAllocation(
-            asset_class=asset_class,
-            value=value,
-            percentage=round((value / total_value) * 100.0, 2)
+            asset_class=asset_class, value=value, percentage=round((value / total_value) * 100.0, 2)
         )
         for asset_class, value in allocation_values.items()
     ]

fin_infra/analytics/rebalancing.py

@@ -329,13 +329,11 @@ def _generate_trade_reasoning(
 
     if action == "buy":
         return (
-            f"Buy {symbol} to increase {asset_class} allocation "
-            f"by {diff_pct:.1f}% towards target"
+            f"Buy {symbol} to increase {asset_class} allocation by {diff_pct:.1f}% towards target"
         )
     else:
         return (
-            f"Sell {symbol} to decrease {asset_class} allocation "
-            f"by {diff_pct:.1f}% towards target"
+            f"Sell {symbol} to decrease {asset_class} allocation by {diff_pct:.1f}% towards target"
         )
 
 

fin_infra/analytics/savings.py

@@ -77,7 +77,7 @@ async def calculate_savings_rate(
         period_enum = Period(period)
     except ValueError:
         raise ValueError(
-            f"Invalid period '{period}'. Must be one of: " f"{', '.join([p.value for p in Period])}"
+            f"Invalid period '{period}'. Must be one of: {', '.join([p.value for p in Period])}"
         )
 
     try:

fin_infra/analytics/spending.py

@@ -344,7 +344,9 @@ async def _detect_spending_anomalies(
         average_amount = current_amount * Decimal("0.8")
 
         deviation_percent: float = (
-            float((current_amount - average_amount) / average_amount) * 100 if average_amount > 0 else 0.0
+            float((current_amount - average_amount) / average_amount) * 100
+            if average_amount > 0
+            else 0.0
         )
 
         # Detect anomalies based on deviation

fin_infra/banking/history.py

@@ -68,10 +68,10 @@ def _check_in_memory_warning() -> None:
     global _production_warning_logged
     if _production_warning_logged:
         return
-    
+
     env = os.getenv("ENV", "development").lower()
     storage_backend = os.getenv("FIN_INFRA_STORAGE_BACKEND", "memory").lower()
-    
+
     if env in ("production", "staging") and storage_backend == "memory":
         _logger.warning(
             "⚠️ CRITICAL: Balance history using IN-MEMORY storage in %s environment! "
@@ -135,7 +135,7 @@ def record_balance_snapshot(
     """
     # Check if in-memory storage is being used in production
     _check_in_memory_warning()
-    
+
     snapshot = BalanceSnapshot(
         account_id=account_id,
         balance=balance,

fin_infra/banking/utils.py

@@ -17,12 +17,14 @@ from ..providers.base import BankingProvider
 
 class BankingConnectionInfo(BaseModel):
     """Information about a banking provider connection."""
-    
+
     model_config = ConfigDict()
-    
+
     provider: Literal["plaid", "teller", "mx"]
     connected: bool
-    access_token: Optional[str] = Field(None, description="Token (only for internal use, never expose)")
+    access_token: Optional[str] = Field(
+        None, description="Token (only for internal use, never expose)"
+    )
     item_id: Optional[str] = None
     enrollment_id: Optional[str] = None
     connected_at: Optional[datetime] = None
@@ -33,12 +35,12 @@ class BankingConnectionInfo(BaseModel):
 
 class BankingConnectionStatus(BaseModel):
     """Status of all banking connections for a user."""
-    
+
     plaid: Optional[BankingConnectionInfo] = None
     teller: Optional[BankingConnectionInfo] = None
     mx: Optional[BankingConnectionInfo] = None
     has_any_connection: bool = False
-    
+
     @property
     def connected_providers(self) -> list[str]:
         """List of connected provider names."""
@@ -50,13 +52,13 @@ class BankingConnectionStatus(BaseModel):
         if self.mx and self.mx.connected:
             providers.append("mx")
         return providers
-    
+
     @property
     def primary_provider(self) -> Optional[str]:
         """Primary provider (first connected, or most recently synced)."""
         if not self.has_any_connection:
             return None
-        
+
         # Preference order: plaid > teller > mx
         if self.plaid and self.plaid.connected:
             return "plaid"
@@ -70,17 +72,17 @@ class BankingConnectionStatus(BaseModel):
 def validate_plaid_token(access_token: str) -> bool:
     """
     Validate Plaid access token format.
-    
+
     Args:
         access_token: Plaid access token to validate
-        
+
     Returns:
         True if token format is valid
-        
+
     Note:
         This only validates format, not that the token is active/unexpired.
         Use provider's API to verify token health.
-    
+
     Example:
         >>> validate_plaid_token("access-sandbox-abc123")
         True
@@ -89,26 +91,26 @@ def validate_plaid_token(access_token: str) -> bool:
     """
     if not access_token:
         return False
-    
+
     # Plaid tokens typically start with "access-{environment}-"
-    pattern = r'^access-(sandbox|development|production)-[a-zA-Z0-9-_]+$'
+    pattern = r"^access-(sandbox|development|production)-[a-zA-Z0-9-_]+$"
     return bool(re.match(pattern, access_token))
 
 
 def validate_teller_token(access_token: str) -> bool:
     """
     Validate Teller access token format.
-    
+
     Args:
         access_token: Teller access token to validate
-        
+
     Returns:
         True if token format is valid
-        
+
     Note:
         This only validates format, not that the token is active/unexpired.
         Use provider's API to verify token health.
-    
+
     Example:
         >>> validate_teller_token("test_token_abc123")
         True
@@ -117,46 +119,46 @@ def validate_teller_token(access_token: str) -> bool:
     """
     if not access_token:
         return False
-    
+
     # Teller tokens are typically alphanumeric with underscores
     # Sandbox tokens often start with "test_"
-    pattern = r'^[a-zA-Z0-9_-]{10,}$'
+    pattern = r"^[a-zA-Z0-9_-]{10,}$"
     return bool(re.match(pattern, access_token))
 
 
 def validate_mx_token(access_token: str) -> bool:
     """
     Validate MX access token format.
-    
+
     Args:
         access_token: MX access token to validate
-        
+
     Returns:
         True if token format is valid
-    
+
     Example:
         >>> validate_mx_token("USR-abc123")
         True
     """
     if not access_token:
         return False
-    
+
     # MX tokens typically have a prefix like "USR-"
-    pattern = r'^[A-Z]+-[a-zA-Z0-9-_]+$'
+    pattern = r"^[A-Z]+-[a-zA-Z0-9-_]+$"
     return bool(re.match(pattern, access_token))
 
 
 def validate_provider_token(provider: str, access_token: str) -> bool:
     """
     Validate token format for any provider.
-    
+
     Args:
         provider: Provider name ("plaid", "teller", "mx")
         access_token: Token to validate
-        
+
     Returns:
         True if token format is valid for the provider
-        
+
     Example:
         >>> validate_provider_token("plaid", "access-sandbox-abc")
         True
@@ -168,29 +170,29 @@ def validate_provider_token(provider: str, access_token: str) -> bool:
         "teller": validate_teller_token,
         "mx": validate_mx_token,
     }
-    
+
     validator = validators.get(provider.lower())
     if not validator:
         # Unknown provider - do basic validation
         return bool(access_token and len(access_token) > 10)
-    
+
     return validator(access_token)
 
 
 def parse_banking_providers(banking_providers: Dict[str, Any]) -> BankingConnectionStatus:
     """
     Parse banking_providers JSON field into structured status.
-    
+
     Args:
         banking_providers: Dictionary from User.banking_providers field
             Structure: {
                 "plaid": {"access_token": "...", "item_id": "...", "connected_at": "..."},
                 "teller": {"access_token": "...", "enrollment_id": "..."}
             }
-            
+
     Returns:
         Structured status with connection info for all providers
-        
+
     Example:
         >>> status = parse_banking_providers(user.banking_providers)
         >>> if status.has_any_connection:
@@ -199,10 +201,10 @@ def parse_banking_providers(banking_providers: Dict[str, Any]) -> BankingConnect
         ...         print(f"Connected: {provider}")
     """
     status = BankingConnectionStatus()
-    
+
     if not banking_providers:
         return status
-    
+
     # Parse Plaid
     if "plaid" in banking_providers:
         plaid_data = banking_providers["plaid"]
@@ -216,7 +218,7 @@ def parse_banking_providers(banking_providers: Dict[str, Any]) -> BankingConnect
             is_healthy=plaid_data.get("is_healthy", True),
             error_message=plaid_data.get("error_message"),
         )
-    
+
     # Parse Teller
     if "teller" in banking_providers:
         teller_data = banking_providers["teller"]
@@ -230,7 +232,7 @@ def parse_banking_providers(banking_providers: Dict[str, Any]) -> BankingConnect
             is_healthy=teller_data.get("is_healthy", True),
             error_message=teller_data.get("error_message"),
         )
-    
+
     # Parse MX
     if "mx" in banking_providers:
         mx_data = banking_providers["mx"]
@@ -243,26 +245,28 @@ def parse_banking_providers(banking_providers: Dict[str, Any]) -> BankingConnect
             is_healthy=mx_data.get("is_healthy", True),
             error_message=mx_data.get("error_message"),
         )
-    
-    status.has_any_connection = any([
-        status.plaid and status.plaid.connected,
-        status.teller and status.teller.connected,
-        status.mx and status.mx.connected,
-    ])
-    
+
+    status.has_any_connection = any(
+        [
+            status.plaid and status.plaid.connected,
+            status.teller and status.teller.connected,
+            status.mx and status.mx.connected,
+        ]
+    )
+
     return status
 
 
 def sanitize_connection_status(status: BankingConnectionStatus) -> Dict[str, Any]:
     """
     Sanitize connection status for API responses (removes access tokens).
-    
+
     Args:
         status: Connection status with tokens
-        
+
     Returns:
         Dictionary safe for API responses (no tokens)
-        
+
     Example:
         >>> status = parse_banking_providers(user.banking_providers)
         >>> safe_data = sanitize_connection_status(status)
@@ -274,7 +278,7 @@ def sanitize_connection_status(status: BankingConnectionStatus) -> Dict[str, Any
         "primary_provider": status.primary_provider,
         "providers": {},
     }
-    
+
     for provider_name in ["plaid", "teller", "mx"]:
         info = getattr(status, provider_name)
         if info:
@@ -289,7 +293,7 @@ def sanitize_connection_status(status: BankingConnectionStatus) -> Dict[str, Any
                 "error_message": info.error_message,
                 # NO access_token - this is sanitized
             }
-    
+
     return result
 
 
@@ -300,15 +304,15 @@ def mark_connection_unhealthy(
 ) -> Dict[str, Any]:
     """
     Mark a provider connection as unhealthy (for error handling).
-    
+
     Args:
         banking_providers: Current banking_providers dict
         provider: Provider name ("plaid", "teller", "mx")
         error_message: Error description
-        
+
     Returns:
         Updated banking_providers dict
-        
+
     Example:
         >>> try:
         ...     accounts = await banking.get_accounts(access_token)
@@ -322,11 +326,11 @@ def mark_connection_unhealthy(
     """
     if provider not in banking_providers:
         return banking_providers
-    
+
     banking_providers[provider]["is_healthy"] = False
     banking_providers[provider]["error_message"] = error_message
     banking_providers[provider]["error_at"] = datetime.now(timezone.utc).isoformat()
-    
+
     return banking_providers
 
 
@@ -336,14 +340,14 @@ def mark_connection_healthy(
 ) -> Dict[str, Any]:
     """
     Mark a provider connection as healthy (after successful sync).
-    
+
     Args:
         banking_providers: Current banking_providers dict
         provider: Provider name
-        
+
     Returns:
         Updated banking_providers dict
-        
+
     Example:
         >>> accounts = await banking.get_accounts(access_token)
         >>> user.banking_providers = mark_connection_healthy(
@@ -355,26 +359,28 @@ def mark_connection_healthy(
     """
     if provider not in banking_providers:
         return banking_providers
-    
+
     banking_providers[provider]["is_healthy"] = True
     banking_providers[provider]["error_message"] = None
     banking_providers[provider]["last_synced_at"] = datetime.now(timezone.utc).isoformat()
-    
+
     return banking_providers
 
 
-def get_primary_access_token(banking_providers: Dict[str, Any]) -> tuple[Optional[str], Optional[str]]:
+def get_primary_access_token(
+    banking_providers: Dict[str, Any],
+) -> tuple[Optional[str], Optional[str]]:
     """
     Get the primary access token and provider name.
-    
+
     Returns the first healthy, connected provider in priority order: plaid > teller > mx.
-    
+
     Args:
         banking_providers: Dictionary from User.banking_providers
-        
+
     Returns:
         Tuple of (access_token, provider_name) or (None, None)
-        
+
     Example:
         >>> access_token, provider = get_primary_access_token(user.banking_providers)
         >>> if access_token:
@@ -382,13 +388,13 @@ def get_primary_access_token(banking_providers: Dict[str, Any]) -> tuple[Optiona
         ...     accounts = await banking.get_accounts(access_token)
     """
     status = parse_banking_providers(banking_providers)
-    
+
     # Priority order: plaid > teller > mx
     for provider_name in ["plaid", "teller", "mx"]:
         info = getattr(status, provider_name)
         if info and info.connected and info.is_healthy and info.access_token:
             return info.access_token, provider_name
-    
+
     return None, None
 
 
@@ -398,14 +404,14 @@ async def test_connection_health(
 ) -> tuple[bool, Optional[str]]:
     """
     Test if a banking connection is healthy by making a lightweight API call.
-    
+
     Args:
         provider: Banking provider instance (from easy_banking())
         access_token: Access token to test
-        
+
     Returns:
         Tuple of (is_healthy, error_message)
-        
+
     Example:
         >>> banking = easy_banking(provider="plaid")
         >>> is_healthy, error = await test_connection_health(banking, access_token)
@@ -415,13 +421,13 @@ async def test_connection_health(
     try:
         # Try to fetch accounts (lightweight call)
         provider.accounts(access_token)
-        
+
         # If we got here, connection is healthy
         return True, None
-        
+
     except Exception as e:
         error_msg = str(e)
-        
+
         # Check for common error patterns
         if "unauthorized" in error_msg.lower() or "invalid" in error_msg.lower():
             return False, "Token invalid or expired"
@@ -434,14 +440,14 @@ async def test_connection_health(
 def should_refresh_token(banking_providers: Dict[str, Any], provider: str) -> bool:
     """
     Check if a provider token should be refreshed.
-    
+
     Args:
         banking_providers: Current banking_providers dict
         provider: Provider name
-        
+
     Returns:
         True if token should be refreshed
-        
+
     Example:
         >>> if should_refresh_token(user.banking_providers, "plaid"):
         ...     # Trigger token refresh flow
@@ -449,13 +455,13 @@ def should_refresh_token(banking_providers: Dict[str, Any], provider: str) -> bo
     """
     if provider not in banking_providers:
         return False
-    
+
     provider_data = banking_providers[provider]
-    
+
     # Check if marked unhealthy
     if not provider_data.get("is_healthy", True):
         return True
-    
+
     # Check last sync time
     last_synced_str = provider_data.get("last_synced_at")
     if last_synced_str:
@@ -465,7 +471,7 @@ def should_refresh_token(banking_providers: Dict[str, Any], provider: str) -> bo
             days_since_sync = (datetime.now(timezone.utc) - last_synced).days
             if days_since_sync > 30:
                 return True
-    
+
     return False
 
 
@@ -473,15 +479,15 @@ def _parse_datetime(value: Any) -> Optional[datetime]:
     """Parse datetime from various formats."""
     if not value:
         return None
-    
+
     if isinstance(value, datetime):
         return value
-    
+
     if isinstance(value, str):
         try:
             # Try ISO format
-            return datetime.fromisoformat(value.replace('Z', '+00:00'))
+            return datetime.fromisoformat(value.replace("Z", "+00:00"))
         except (ValueError, AttributeError):
             pass
-    
+
     return None

fin_infra/brokerage/__init__.py

@@ -123,7 +123,7 @@ def easy_brokerage(
         )
 
     else:
-        raise ValueError(f"Unknown brokerage provider: {provider_name}. " f"Supported: alpaca")
+        raise ValueError(f"Unknown brokerage provider: {provider_name}. Supported: alpaca")
 
 
 def add_brokerage(

fin_infra/budgets/tracker.py

@@ -155,7 +155,7 @@ class BudgetTracker:
             BudgetType(type)
         except ValueError:
             raise ValueError(
-                f"Invalid budget type: {type}. " f"Valid types: {[t.value for t in BudgetType]}"
+                f"Invalid budget type: {type}. Valid types: {[t.value for t in BudgetType]}"
             )
 
         # Validate budget period
@@ -163,8 +163,7 @@ class BudgetTracker:
             BudgetPeriod(period)
         except ValueError:
             raise ValueError(
-                f"Invalid budget period: {period}. "
-                f"Valid periods: {[p.value for p in BudgetPeriod]}"
+                f"Invalid budget period: {period}. Valid periods: {[p.value for p in BudgetPeriod]}"
             )
 
         # Validate categories