Update routes/stylist.py

routes/stylist.py

@@ -117,6 +117,24 @@ def get_monthly_revenue(account_id: str) -> List[Dict[str, Any]]:
         logger.error(f"❌ Error getting monthly revenue: {str(e)}")
         return list(monthly_data.values())
 
+def format_subscription_date(created_at_str: str) -> str:
+    """Format the subscription date to the requested format: e.g. '13th January 2025'"""
+    try:
+        created_at = datetime.fromisoformat(created_at_str.replace('Z', '+00:00'))
+        
+        # Add suffix to day
+        day = created_at.day
+        if 4 <= day <= 20 or 24 <= day <= 30:
+            suffix = "th"
+        else:
+            suffix = {1: "st", 2: "nd", 3: "rd"}.get(day % 10, "th")
+        
+        # Format the date
+        return f"{day}{suffix} {created_at.strftime('%B %Y')}"
+    except Exception as e:
+        logger.error(f"❌ Error formatting subscription date: {str(e)}")
+        return "Unknown date"
+
 def get_active_subscribers(user_id: str, page: int) -> Dict[str, Any]:
     limit = 30
     offset = page * limit
@@ -127,18 +145,17 @@ def get_active_subscribers(user_id: str, page: int) -> Dict[str, Any]:
         subscriber_list = []
         for sub in subscribers:
             customer_id = sub.get("customer_id")
-            created_at = datetime.strptime(sub.get("created_at"), "%Y-%m-%d %H:%M:%S.%f%z")
-            formatted_date = created_at.strftime("%dth %B %Y")
             user_data_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{customer_id}"
             user_response = requests.get(user_data_url, headers=SUPABASE_HEADERS)
             if user_response.status_code == 200 and user_response.json():
                 user_info = user_response.json()[0]
+                subscription_date = format_subscription_date(sub.get("created_at", ""))
                 subscriber_list.append({
                     "id": user_info.get("id"),
                     "name": user_info.get("name"),
                     "avatar": user_info.get("avatar"),
                     "blurhash": user_info.get("blurhash"),
-                    "subscription_date": formatted_date
+                    "subscription_date": subscription_date
                 })
         
         has_next_page = len(subscribers) == limit
@@ -146,6 +163,22 @@ def get_active_subscribers(user_id: str, page: int) -> Dict[str, Any]:
     
     return {"subscribers": [], "has_next_page": False}
 
+def get_total_followers(user_id: str) -> int:
+    url = f"{SUPABASE_URL}/rest/v1/followers?following_id=eq.{user_id}"
+    response = requests.get(url, headers=SUPABASE_HEADERS)
+    if response.status_code == 200:
+        followers = response.json()
+        return len(followers)
+    return 0
+
+def get_total_subscribers(user_id: str) -> int:
+    url = f"{SUPABASE_URL}/rest/v1/Subscriptions?stylist_id=eq.{user_id}&active=eq.true"
+    response = requests.get(url, headers=SUPABASE_HEADERS)
+    if response.status_code == 200:
+        subscribers = response.json()
+        return len(subscribers)
+    return 0
+
 @router.get("/dashboard")
 def get_dashboard(user_token: str = Header(None, alias="User-key"), page: int = Query(0, ge=0)):
     try: