Update routes/stylist.py

routes/stylist.py

@@ -303,7 +303,60 @@ def get_total_subscribers(user_id: str) -> int:
         subscribers = response.json()
         return len(subscribers)
     return 0
+
+def get_courtesy_consultations(user_id: str) -> Dict[str, Any]:
+    ny_timezone = pytz.timezone('America/New_York')
+    now_ny = datetime.now(ny_timezone)
+    monthly_data = {}
+    
+    # Inicializar dados para os últimos 6 meses
+    for i in range(6):
+        target_date = now_ny - relativedelta(months=i)
+        month_num = target_date.month
+        month_name = target_date.strftime('%b')
+        year = target_date.year
+        month_key = f"{year}-{month_num}"
+        monthly_data[month_key] = {
+            "month": month_num,
+            "name": month_name,
+            "current": (month_num == now_ny.month and year == now_ny.year),
+            "courtesy_count": 0
+        }
     
+    # Calcular data de início (6 meses atrás)
+    start_date = (now_ny - relativedelta(months=6)).strftime('%Y-%m-%d')
+    
+    try:
+        # Consultar agendamentos de cortesia
+        url = f"{SUPABASE_URL}/rest/v1/schedules?stylist_id=eq.{user_id}&courtesy=eq.true&date=gte.{start_date}"
+        response = requests.get(url, headers=SUPABASE_HEADERS)
+        
+        if response.status_code == 200:
+            schedules = response.json()
+            
+            # Contar consultas por mês
+            for schedule in schedules:
+                # Converter a data para o fuso horário de NY
+                schedule_date_str = schedule.get("date")
+                schedule_date = datetime.fromisoformat(schedule_date_str.replace('Z', '+00:00')).astimezone(ny_timezone)
+                
+                month_key = f"{schedule_date.year}-{schedule_date.month}"
+                if month_key in monthly_data:
+                    monthly_data[month_key]["courtesy_count"] += 1
+        
+        # Convertendo para lista e ordenando por mês (do mais recente para o mais antigo)
+        result = list(monthly_data.values())
+        result.sort(key=lambda x: (now_ny.year * 12 + now_ny.month - (x["month"])) % 12)
+        
+        return {
+            "monthly_courtesy_consultations": result
+        }
+    except Exception as e:
+        logger.error(f"❌ Error getting courtesy consultations: {str(e)}")
+        return {
+            "monthly_courtesy_consultations": list(monthly_data.values())
+        }
+
 @router.post("/bank_account_dashboard_link")
 async def generate_bank_account_dashboard_link(data: UserIDRequest):
     try:
@@ -351,6 +404,9 @@ def get_dashboard(user_token: str = Header(None, alias="User-key"), page: int =
         # Obter dados de receita mensal e total dos últimos 6 meses
         revenue_data = get_monthly_revenue(stripe_id)
         
+        # Obter dados de consultas gratuitas dos últimos 6 meses
+        courtesy_data = get_courtesy_consultations(user_id)
+        
         # Obter detalhes da conta conectada
         account_details = get_connected_account_details(stripe_id)
         
@@ -362,6 +418,7 @@ def get_dashboard(user_token: str = Header(None, alias="User-key"), page: int =
             "available_balance": get_account_balance(stripe_id),
             "monthly_revenue": revenue_data["monthly_data"],
             "total_revenue_last_6_months": revenue_data["total_last_6_months"],
+            "monthly_courtesy_consultations": courtesy_data["monthly_courtesy_consultations"],
             "total_followers": get_total_followers(user_id),
             "total_subscribers": get_total_subscribers(user_id),
             "account_details": account_details,