Update routes/dashboard_home.py

routes/dashboard_home.py

@@ -404,32 +404,91 @@ async def get_monthly_revenue_data(target_date) -> Dict[str, Any]:
         "transfer_count": transfer_data["transfer_count"]
     }
 
+# Função auxiliar para lidar com os períodos específicos
+def get_period_timestamps(period: str) -> Tuple[Optional[int], Optional[int]]:
+    """Retorna timestamps inicial e final baseados no período especificado"""
+    ny_timezone = pytz.timezone('America/New_York')
+    now_ny = datetime.now(ny_timezone)
+    current_year = now_ny.year
+    
+    # Inicializar timestamps
+    start_timestamp = None
+    end_timestamp = int(now_ny.timestamp())
+    
+    # Lidar com os meses específicos
+    months = {
+        'January': 1, 'February': 2, 'March': 3, 'April': 4,
+        'May': 5, 'June': 6, 'July': 7, 'August': 8,
+        'September': 9, 'October': 10, 'November': 11, 'December': 12
+    }
+    
+    if period in months:
+        # Período para um mês específico do ano atual
+        month_num = months[period]
+        
+        # Se o mês solicitado ainda não chegou neste ano, use o ano anterior
+        if month_num > now_ny.month:
+            year = current_year - 1
+        else:
+            year = current_year
+            
+        # Definir início e fim do mês
+        month_start = datetime(year, month_num, 1, tzinfo=ny_timezone)
+        if month_num == 12:
+            month_end = datetime(year + 1, 1, 1, tzinfo=ny_timezone) - timedelta(seconds=1)
+        else:
+            month_end = datetime(year, month_num + 1, 1, tzinfo=ny_timezone) - timedelta(seconds=1)
+        
+        start_timestamp = int(month_start.timestamp())
+        end_timestamp = int(month_end.timestamp())
+    
+    elif period == "All Year":
+        # Período para o ano atual inteiro
+        year_start = datetime(current_year, 1, 1, tzinfo=ny_timezone)
+        year_end = datetime(current_year + 1, 1, 1, tzinfo=ny_timezone) - timedelta(seconds=1)
+        
+        start_timestamp = int(year_start.timestamp())
+        end_timestamp = int(year_end.timestamp())
+        
+        # Se estamos no início do ano, pode ser útil mostrar dados do ano anterior também
+        if now_ny.month <= 2:
+            year_start = datetime(current_year - 1, 1, 1, tzinfo=ny_timezone)
+            start_timestamp = int(year_start.timestamp())
+    
+    # Períodos legados (manter para compatibilidade)
+    elif period == "last_month":
+        start_date = now_ny - relativedelta(months=1)
+        start_timestamp = int(start_date.timestamp())
+    elif period == "last_year":
+        start_date = now_ny - relativedelta(years=1)
+        start_timestamp = int(start_date.timestamp())
+    
+    return start_timestamp, end_timestamp
+
+# Helper function to identify if we need specific month data
+def is_specific_month(period: str) -> bool:
+    """Verifica se o período solicitado é um mês específico"""
+    months = [
+        'January', 'February', 'March', 'April', 'May', 'June',
+        'July', 'August', 'September', 'October', 'November', 'December'
+    ]
+    return period in months
+
 @router.get("/admin/dashboard")
 async def get_admin_dashboard(
     user_token: str = Header(None, alias="User-key"),
-    period: str = Query("all_time", description="Período: all_time, last_month, last_year")
+    period: str = Query("All Year", description="Período: All Year, January, February, etc.")
 ):
     """
     Endpoint para dashboard administrativo com métricas de faturamento
-    e divisão entre app e estilistas - versão otimizada
+    e divisão entre app e estilistas - versão otimizada com seleção de meses
     """
     try:
         # Verificar se é um administrador
         user_id = await verify_admin_token(user_token)
         
-        # Definir intervalo de datas baseado no período solicitado
-        ny_timezone = pytz.timezone('America/New_York')
-        now_ny = datetime.now(ny_timezone)
-        
-        start_timestamp = None
-        end_timestamp = int(now_ny.timestamp())
-        
-        if period == "last_month":
-            start_date = now_ny - relativedelta(months=1)
-            start_timestamp = int(start_date.timestamp())
-        elif period == "last_year":
-            start_date = now_ny - relativedelta(years=1)
-            start_timestamp = int(start_date.timestamp())
+        # Obter timestamps baseados no período
+        start_timestamp, end_timestamp = get_period_timestamps(period)
         
         # Executar todas as chamadas em paralelo
         tasks = [
@@ -439,10 +498,31 @@ async def get_admin_dashboard(
             get_top_stylists(start_timestamp, end_timestamp, 10)
         ]
         
-        # Adicionar task para dados mensais se necessário
+        # Adicionar task para dados mensais se for um mês específico
         monthly_task = None
-        if period == "last_month":
-            target_date = now_ny - relativedelta(months=1)
+        monthly_data = None
+        
+        if is_specific_month(period):
+            # Obter dados mensais para o mês solicitado
+            ny_timezone = pytz.timezone('America/New_York')
+            now_ny = datetime.now(ny_timezone)
+            current_year = now_ny.year
+            
+            # Obter número do mês
+            months = {
+                'January': 1, 'February': 2, 'March': 3, 'April': 4,
+                'May': 5, 'June': 6, 'July': 7, 'August': 8,
+                'September': 9, 'October': 10, 'November': 11, 'December': 12
+            }
+            month_num = months[period]
+            
+            # Se o mês ainda não chegou neste ano, use o ano anterior
+            if month_num > now_ny.month:
+                year = current_year - 1
+            else:
+                year = current_year
+                
+            target_date = datetime(year, month_num, 1, tzinfo=ny_timezone)
             monthly_task = get_monthly_revenue_data(target_date)
             tasks.append(monthly_task)
         
@@ -450,11 +530,10 @@ async def get_admin_dashboard(
         results = await asyncio.gather(*tasks)
         
         # Extrair resultados
-        if period == "last_month":
+        if is_specific_month(period):
             revenue_data, transfer_data, user_data, top_stylists, monthly_data = results
         else:
             revenue_data, transfer_data, user_data, top_stylists = results
-            monthly_data = None
         
         # Calcular divisão de receita
         revenue_share = get_app_revenue_share(
@@ -478,9 +557,27 @@ async def get_admin_dashboard(
             "top_stylists": top_stylists
         }
         
-        # Adicionar dados específicos de mês apenas para período last_month
-        if period == "last_month" and monthly_data:
+        # Adicionar dados específicos de mês
+        if is_specific_month(period) and monthly_data:
             response["monthly_data"] = monthly_data
+            
+        # Para "All Year", adicionar dados mensais para visualização
+        if period == "All Year":
+            # Obter dados para cada mês do ano atual
+            ny_timezone = pytz.timezone('America/New_York')
+            now_ny = datetime.now(ny_timezone)
+            current_year = now_ny.year
+            
+            monthly_tasks = []
+            for month in range(1, 13):
+                # Só incluir meses que já passaram ou o mês atual
+                if month <= now_ny.month:
+                    target_date = datetime(current_year, month, 1, tzinfo=ny_timezone)
+                    monthly_tasks.append(get_monthly_revenue_data(target_date))
+            
+            if monthly_tasks:
+                monthly_results = await asyncio.gather(*monthly_tasks)
+                response["year_data"] = monthly_results
         
         return response