Update routes/stylist.py

routes/stylist.py

@@ -2,8 +2,10 @@ import os
 import stripe
 import requests
 import logging
+import pytz
 from fastapi import APIRouter, HTTPException, Header
 from datetime import datetime, timedelta
+from dateutil.relativedelta import relativedelta
 from typing import List, Dict, Any
 
 router = APIRouter()
@@ -47,14 +49,43 @@ def verify_token(user_token: str) -> str:
     else:
         raise HTTPException(status_code=401, detail="Invalid or expired token")
 
-def get_weekly_revenue(account_id: str) -> List[Dict[str, Any]]:
+def get_monthly_revenue(account_id: str) -> List[Dict[str, Any]]:
     """
-    Busca os valores recebidos nas últimas 6 semanas para o account_id.
-    Retorna uma lista com número da semana, data e valor.
+    Busca os valores recebidos nos últimos 6 meses para o account_id.
+    Retorna uma lista com número do mês, nome do mês, indicador se é o mês atual e valor.
     """
-    # Calcular o timestamp para 6 semanas atrás
-    end_date = datetime.now()
-    start_date = end_date - timedelta(weeks=6)
+    # Obter data atual no fuso horário de Nova York
+    ny_timezone = pytz.timezone('America/New_York')
+    now_ny = datetime.now(ny_timezone)
+    current_month = now_ny.month
+    current_year = now_ny.year
+    
+    # Preparar dicionário com os últimos 6 meses (incluindo o atual)
+    monthly_data = {}
+    
+    # Criar entradas para os últimos 6 meses, mesmo sem dados
+    for i in range(6):
+        # Calcular o mês e ano para i meses atrás
+        target_date = now_ny - relativedelta(months=i)
+        month_num = target_date.month
+        month_name = target_date.strftime('%b')  # Formato abreviado do mês (Jan, Feb, etc)
+        year = target_date.year
+        
+        # Chave composta para garantir unicidade (ano-mês)
+        month_key = f"{year}-{month_num}"
+        
+        # Verificar se é o mês atual
+        is_current = (month_num == current_month and year == current_year)
+        
+        monthly_data[month_key] = {
+            "month": month_num,
+            "name": month_name,
+            "current": is_current,
+            "amount": 0
+        }
+    
+    # Calcular o timestamp para 6 meses atrás
+    start_date = now_ny - relativedelta(months=6)
     start_timestamp = int(start_date.timestamp())
     
     # Buscar transferências para a conta
@@ -65,36 +96,28 @@ def get_weekly_revenue(account_id: str) -> List[Dict[str, Any]]:
             limit=100  # Limitando a 100 transferências mais recentes
         )
         
-        # Preparar dicionário para agrupar por semana
-        weekly_data = {}
-        
         # Processar as transferências
         for transfer in transfers.data:
-            transfer_date = datetime.fromtimestamp(transfer.created)
-            # Número da semana no ano (1-53)
-            week_number = transfer_date.isocalendar()[1]
-            week_start = transfer_date - timedelta(days=transfer_date.weekday())
-            week_start_str = week_start.strftime("%Y-%m-%d")
-            
-            if week_number not in weekly_data:
-                weekly_data[week_number] = {
-                    "week": week_number,
-                    "date": week_start_str,
-                    "amount": 0
-                }
+            transfer_date = datetime.fromtimestamp(transfer.created, ny_timezone)
+            month_num = transfer_date.month
+            year = transfer_date.year
+            month_key = f"{year}-{month_num}"
             
-            # Adicionar o valor da transferência (em centavos)
-            weekly_data[week_number]["amount"] += transfer.amount
+            # Verificar se o mês está dentro dos últimos 6 meses
+            if month_key in monthly_data:
+                # Adicionar o valor da transferência (em centavos)
+                monthly_data[month_key]["amount"] += transfer.amount
         
-        # Converter para lista e ordenar por semana
-        result = list(weekly_data.values())
-        result.sort(key=lambda x: x["week"])
+        # Converter para lista e ordenar por data (mais antigo primeiro)
+        result = list(monthly_data.values())
+        result.sort(key=lambda x: (current_month - x["month"]) % 12)
         
         return result
         
     except Exception as e:
-        logger.error(f"❌ Error getting weekly revenue: {str(e)}")
-        return []
+        logger.error(f"❌ Error getting monthly revenue: {str(e)}")
+        # Retornar os meses mesmo sem transferências
+        return list(monthly_data.values())
 
 def get_account_balance(account_id: str) -> Dict[str, Any]:
     """
@@ -168,21 +191,21 @@ def get_dashboard(user_token: str = Header(None, alias="User-key")):
                 "available_balance": 0,
                 "pending_balance": 0,
                 "currency": "BRL",
-                "weekly_revenue": []
+                "monthly_revenue": []
             }
         
         # Buscar saldo da conta
         balance_info = get_account_balance(stripe_id)
         
-        # Buscar valores recebidos nas últimas 6 semanas
-        weekly_revenue = get_weekly_revenue(stripe_id)
+        # Buscar valores recebidos nos últimos 6 meses
+        monthly_revenue = get_monthly_revenue(stripe_id)
         
         return {
             "stripe_id": stripe_id,
             "available_balance": balance_info["available_balance"],
             "pending_balance": balance_info["pending_balance"],
             "currency": balance_info["currency"],
-            "weekly_revenue": weekly_revenue
+            "monthly_revenue": monthly_revenue
         }
 
     except HTTPException as http_err: